Commit cedebc6c authored by Khalique's avatar Khalique
Browse files

formatting

parent 0d5a6cd0
...@@ -736,12 +736,11 @@ struct tf_parser ...@@ -736,12 +736,11 @@ struct tf_parser
return prog.add_instruction(op, make_contiguous(args[0])); return prog.add_instruction(op, make_contiguous(args[0]));
} }
instruction_ref parse_slice(const std::string&, instruction_ref
const attribute_map&, parse_slice(const std::string&, const attribute_map&, std::vector<instruction_ref> args)
std::vector<instruction_ref> args)
{ {
op::slice op; op::slice op;
auto starts = args[1]->eval().get<int32_t>().to_vector(); auto starts = args[1]->eval().get<int32_t>().to_vector();
auto size = args[2]->eval().get<int32_t>().to_vector(); auto size = args[2]->eval().get<int32_t>().to_vector();
auto axes = args[0]->get_shape().lens(); auto axes = args[0]->get_shape().lens();
size_t num_axes = axes.size(); size_t num_axes = axes.size();
......
...@@ -355,7 +355,7 @@ TEST_CASE(slice_test) ...@@ -355,7 +355,7 @@ TEST_CASE(slice_test)
{ {
migraphx::program p; migraphx::program p;
std::size_t num_axes = 2; std::size_t num_axes = 2;
auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {5,10}}); auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {5, 10}});
migraphx::shape s0{migraphx::shape::int32_type, {num_axes}}; migraphx::shape s0{migraphx::shape::int32_type, {num_axes}};
p.add_literal(migraphx::literal{s0, {1, 0}}); p.add_literal(migraphx::literal{s0, {1, 0}});
p.add_literal(migraphx::literal{s0, {2, -1}}); p.add_literal(migraphx::literal{s0, {2, -1}});
......
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