"...text-generation-inference-dcu.git" did not exist on "366dfe8247b1978f9a1cd4aee5fcf91bde836372"
Commit d205aa73 authored by Khalique's avatar Khalique
Browse files

change order of nodes

parent 85496cd9
...@@ -24,7 +24,7 @@ inline namespace MIGRAPHX_INLINE_NS { ...@@ -24,7 +24,7 @@ inline namespace MIGRAPHX_INLINE_NS {
struct tf_parser struct tf_parser
{ {
using attribute_map = std::unordered_map<std::string, tensorflow::AttrValue>; using attribute_map = std::unordered_map<std::string, tensorflow::AttrValue>;
using node_map = std::unordered_map<std::string, tensorflow::NodeDef>; using node_map = std::map<std::string, tensorflow::NodeDef>;
// using input_node_map = std::unordered_map<std::string, std::unordered_set<std::string>>; // using input_node_map = std::unordered_map<std::string, std::unordered_set<std::string>>;
using op_func = std::function<instruction_ref(attribute_map, std::vector<instruction_ref>)>; using op_func = std::function<instruction_ref(attribute_map, std::vector<instruction_ref>)>;
......
...@@ -178,9 +178,9 @@ TEST_CASE(mean_test) ...@@ -178,9 +178,9 @@ TEST_CASE(mean_test)
p.add_literal(l); p.add_literal(l);
migraphx::op::pooling op; migraphx::op::pooling op;
op.lengths = {16, 16}; op.lengths = {16, 16};
p.add_instruction(op, l0);
auto l3 = p.add_instruction(op, l0); auto l3 = p.add_instruction(op, l0);
p.add_instruction(migraphx::op::squeeze{{2, 3}}, l3); p.add_instruction(migraphx::op::squeeze{{2, 3}}, l3);
p.add_instruction(op, l0);
auto prog = migraphx::parse_tf("mean_test.pb", false); auto prog = migraphx::parse_tf("mean_test.pb", false);
EXPECT(p == prog); EXPECT(p == prog);
...@@ -195,9 +195,9 @@ TEST_CASE(mean_test_nhwc) ...@@ -195,9 +195,9 @@ TEST_CASE(mean_test_nhwc)
p.add_literal(l); p.add_literal(l);
migraphx::op::pooling op; migraphx::op::pooling op;
op.lengths = {16, 16}; op.lengths = {16, 16};
p.add_instruction(op, l0);
auto l3 = p.add_instruction(op, l0); auto l3 = p.add_instruction(op, l0);
p.add_instruction(migraphx::op::squeeze{{2, 3}}, l3); p.add_instruction(migraphx::op::squeeze{{2, 3}}, l3);
p.add_instruction(op, l0);
auto prog = migraphx::parse_tf("mean_test_nhwc.pb", true); auto prog = migraphx::parse_tf("mean_test_nhwc.pb", true);
EXPECT(p == prog); EXPECT(p == prog);
...@@ -271,8 +271,8 @@ TEST_CASE(pooling_test) ...@@ -271,8 +271,8 @@ TEST_CASE(pooling_test)
max_pool_op.stride = {2, 2}; max_pool_op.stride = {2, 2};
avg_pool_op.lengths = {2, 2}; avg_pool_op.lengths = {2, 2};
max_pool_op.lengths = {2, 2}; max_pool_op.lengths = {2, 2};
p.add_instruction(max_pool_op, l0);
p.add_instruction(avg_pool_op, l0); p.add_instruction(avg_pool_op, l0);
p.add_instruction(max_pool_op, l0);
auto prog = migraphx::parse_tf("pooling_test.pb", true); auto prog = migraphx::parse_tf("pooling_test.pb", true);
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