Commit ce62af17 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format.

parent 97d605b8
...@@ -653,7 +653,7 @@ struct gather ...@@ -653,7 +653,7 @@ struct gather
} }
// negative value means counting dimensions from back // negative value means counting dimensions from back
if (axis < 0) if(axis < 0)
{ {
axis += lens.size(); axis += lens.size();
} }
......
...@@ -415,8 +415,8 @@ TEST_CASE(shape_test) ...@@ -415,8 +415,8 @@ TEST_CASE(shape_test)
TEST_CASE(gather_test) TEST_CASE(gather_test)
{ {
migraphx::program p; migraphx::program p;
auto l0 = p.add_parameter("data", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}}); auto l0 = p.add_parameter("data", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}});
auto l1 = p.add_parameter("indices", migraphx::shape{migraphx::shape::int32_type, {2, 3}}); auto l1 = p.add_parameter("indices", migraphx::shape{migraphx::shape::int32_type, {2, 3}});
int axis = 1; int axis = 1;
p.add_instruction(migraphx::op::gather{axis}, l0, l1); p.add_instruction(migraphx::op::gather{axis}, l0, l1);
auto prog = migraphx::parse_onnx("gather_test.onnx"); auto prog = migraphx::parse_onnx("gather_test.onnx");
...@@ -431,7 +431,7 @@ TEST_CASE(shape_gather_test) ...@@ -431,7 +431,7 @@ TEST_CASE(shape_gather_test)
auto l1 = auto l1 =
p.add_literal(migraphx::shape{migraphx::shape::int64_type, {3}}, l0->get_shape().lens()); p.add_literal(migraphx::shape{migraphx::shape::int64_type, {3}}, l0->get_shape().lens());
migraphx::shape const_shape{migraphx::shape::int32_type, {1}}; migraphx::shape const_shape{migraphx::shape::int32_type, {1}};
auto l2 = p.add_literal(migraphx::literal{const_shape, {1}}); auto l2 = p.add_literal(migraphx::literal{const_shape, {1}});
int axis = 0; int axis = 0;
p.add_instruction(migraphx::op::gather{axis}, l1, l2); p.add_instruction(migraphx::op::gather{axis}, l1, l2);
auto prog = migraphx::parse_onnx("shape_gather.onnx"); auto prog = migraphx::parse_onnx("shape_gather.onnx");
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment