def _TestRun(self, sess, batch_size, expect_engine_is_run):
clear_test_values("")
result = sess.run("output:0", feed_dict={"input:0": [[[1.0]]] * batch_size})
self.assertAllEqual([[[4.0]]] * batch_size, result)
execute_engine_test_value = ("done" if expect_engine_is_run else "")
execute_native_segment_test_value = ("" if expect_engine_is_run else "done")
self.assertEqual(execute_engine_test_value,
get_test_value("TRTEngineOp_0:ExecuteTrtEngine"))
self.assertEqual(execute_native_segment_test_value,
get_test_value("TRTEngineOp_0:ExecuteNativeSegment"))
def _PrepareRun(self, graph_state):
"""Set up necessary testing environment before calling sess.run()."""
# Clear test values added by TRTEngineOp.
clear_test_values("TRTEngineOp_.*:ExecuteTrtEngine")
clear_test_values("TRTEngineOp_.*:ExecuteCalibration")
clear_test_values("TRTEngineOp_.*:ExecuteNativeSegment")
def setUp(self):
"""Setup method."""
super(TfTrtIntegrationTestBase, self).setUp()
warnings.simplefilter("always")
clear_test_values("")
def _PrepareRun(self, graph_state):
"""Set up necessary testing environment before calling sess.run()."""
# Clear test values added by TRTEngineOp.
clear_test_values("TRTEngineOp_.*:ExecuteTrtEngine")
clear_test_values("TRTEngineOp_.*:ExecuteCalibration")
clear_test_values("TRTEngineOp_.*:ExecuteNativeSegment")
def setUp(self):
"""Setup method."""
super(TfTrtIntegrationTestBase, self).setUp()
warnings.simplefilter("always")
clear_test_values("")
def setUp(self):
"""Setup method."""
super(PartiallyConvertedTestB, self).setUp()
# Let it fail to build the first engine.
clear_test_values("")
add_test_value("TRTEngineOp_0:CreateTRTNode", "fail")
