Commit 4e9393c4 authored by Khalique's avatar Khalique
Browse files

formatting

parent 5086b1cb
......@@ -742,7 +742,8 @@ struct tf_parser
return to_nhwc(prog.add_instruction(op::squeeze{squeeze_axes}, l0));
}
instruction_ref parse_transpose(const std::string&, const attribute_map&, std::vector<instruction_ref> args)
instruction_ref
parse_transpose(const std::string&, const attribute_map&, std::vector<instruction_ref> args)
{
auto perm = args[1]->eval().get<int32_t>().to_vector();
op::transpose op;
......
......@@ -394,8 +394,8 @@ TEST_CASE(transpose_test)
migraphx::program p;
auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {1, 3, 16, 16}});
migraphx::shape s0{migraphx::shape::int32_type, {4}};
p.add_literal(migraphx::literal{s0, {0,2,3,1}});
p.add_instruction(migraphx::op::transpose{{0,2,3,1}}, l0);
p.add_literal(migraphx::literal{s0, {0, 2, 3, 1}});
p.add_instruction(migraphx::op::transpose{{0, 2, 3, 1}}, l0);
auto prog = optimize_tf("transpose_test.pb", false);
EXPECT(p == prog);
......
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