Fix tf tests

05595b85 · Paul · b0586006 · 05595b85
Commit 05595b85 authored Aug 12, 2019 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 4 deletions

test/tf/tf_test.cpp test/tf/tf_test.cpp +2 -4

No files found.
--- a/test/tf/tf_test.cpp
+++ b/test/tf/tf_test.cpp
@@ -257,8 +257,7 @@ TEST_CASE(mean_test)
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {1, 3, 16, 16}});
    p.add_literal(l);
    p.add_literal(l);
-    migraphx::op::pooling op;
+    migraphx::op::reduce_mean op{{2, 3}};
-    op.lengths = {16, 16};
    p.add_instruction(op, l0);
    auto l3 = p.add_instruction(op, l0);
    p.add_instruction(migraphx::op::squeeze{{2, 3}}, l3);
@@ -272,8 +271,7 @@ TEST_CASE(mean_test_nhwc)
    migraphx::program p;
    migraphx::literal l{migraphx::shape{migraphx::shape::int32_type, {2}}, {1, 2}};
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {1, 3, 16, 16}});
-    migraphx::op::pooling op;
+    migraphx::op::reduce_mean op{{2, 3}};
-    op.lengths = {16, 16};
    auto l3    = p.add_instruction(op, l0);
    p.add_instruction(migraphx::op::squeeze{{2, 3}}, l3);
    auto prog = optimize_tf("mean_test_nhwc.pb", true);