Home
last modified time | relevance | path

Searched refs:is_v2 (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py436 if run_params.is_v2:
453 if run_params.is_v2:
798 if run_params.is_v2:
901 if run_params.is_v2:
969 if run_params.is_v2:
998 if run_params.is_v2:
1108 def _AddTestsFor(test_class, is_v2): argument
1110 opts = _GetTestConfigsV2() if is_v2 else _GetTestConfigsV1()
1115 test_name = "%s_%s_%s_%s_%s" % ("testTfTrtV2" if is_v2 else "testTfTrt",
1124 is_v2=is_v2)
[all …]
Dtrt_mode_test.py123 return run_params.is_v2 and trt_test.IsTensorRTVersionGreaterEqual(6) and (
160 return run_params.is_v2 and trt_test.IsTensorRTVersionGreaterEqual(6) and (
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py173 def _check_conversion_params(conversion_params, is_v2=False): argument
235 is_v2=False, argument
255 _check_conversion_params(conversion_params, is_v2=is_v2)
256 if is_v2 and is_dynamic_op is not None and not is_dynamic_op:
258 if not is_v2 and is_dynamic_op is None:
319 is_v2=False, argument
322 max_batch_size, is_v2,
949 _check_conversion_params(conversion_params, is_v2=True)
Dtrt_convert_test.py360 is_v2=False))