[TENSORFLOW][TEST] add tf_testing.AddShapesToGraphDef test

* update frontend test * retrigger CI
apache · Mar 8, 2020 · 24e2b52 · 24e2b52
1 parent 277dc46
commit 24e2b52
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/tests/python/frontend/tensorflow/test_forward.py b/tests/python/frontend/tensorflow/test_forward.py
@@ -171,11 +171,7 @@ def name_without_num(name):
     with tf.Session() as sess:
         if init_global_variables:
             sess.run(variables.global_variables_initializer())
-        final_graph_def = tf.graph_util.convert_variables_to_constants(
-            sess,
-            sess.graph.as_graph_def(add_shapes=True),
-            out_node,
-        )
+        final_graph_def = tf_testing.AddShapesToGraphDef(sess, out_node)
         tf_output = run_tf_graph(sess, in_data, in_name, out_name)
 
         for device in ["llvm", "cuda"]: