"web/app/api/git@developer.sourcefind.cn:orangecat/ollama.git" did not exist on "7bf3212c5bfdfb470b726bda3648f6049f7092d3"
Commit 4e9393c4 authored by Khalique's avatar Khalique
Browse files

formatting

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