Searched refs:gdef_to_verify (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 816 def _VerifyGraphDefV1(self, run_params, original_gdef, gdef_to_verify, argument
820 functions = [f.signature.name for f in gdef_to_verify.library.function]
821 for node in gdef_to_verify.node:
854 self._VerifyConnections(expected_engines, original_gdef, gdef_to_verify)
858 converted_gdef=gdef_to_verify,
862 def _VerifyGraphDefV2(self, run_params, original_gdef, gdef_to_verify, argument
867 all_op_names = [node.name for node in gdef_to_verify.node]
869 node.name for node in gdef_to_verify.node if node.op == "TRTEngineOp"
871 for func in gdef_to_verify.library.function:
898 gdef_to_verify = self._GetGraphDef(run_params,
[all …]
|