Merge branch 'develop' of https://github.com/ROCmSoftwarePlatform/AMDMIGraphX into op_capture

test/tf/tf_test.cpp

@@ -159,6 +159,31 @@ TEST_CASE(depthwiseconv_test)
     EXPECT(p == prog);
 }
 
+TEST_CASE(expanddims_test)
+{
+    migraphx::program p;
+
+    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 3, 4}});
+    p.add_literal(0);
+    p.add_instruction(migraphx::op::reshape{{1, 2, 3, 4}}, l0);
+    auto prog = optimize_tf("expanddims_test.pb", false);
+
+    EXPECT(p == prog);
+}
+
+TEST_CASE(expanddims_test_neg_dims)
+{
+    // this check makes sure the pb parses negative dim value correctly
+    migraphx::program p;
+
+    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 3, 4}});
+    p.add_literal(-1);
+    p.add_instruction(migraphx::op::reshape{{2, 3, 4, 1}}, l0);
+    auto prog = optimize_tf("expanddims_neg_test.pb", false);
+
+    EXPECT(p == prog);
+}
+
 TEST_CASE(identity_test)
 {
     migraphx::program p;