Home
last modified time | relevance | path

Searched refs:maximum_cached_engines (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py172 maximum_cached_engines=1, argument
177 minimum_segment_size, maximum_cached_engines,
335 "maximum_cached_engines"].i = conversion_params.maximum_cached_engines
427 maximum_cached_engines=1, argument
520 maximum_cached_engines=maximum_cached_engines,
843 maximum_cached_engines, argument
850 self._maximum_cached_engines = maximum_cached_engines
1284 self._conversion_params.maximum_cached_engines)
1331 maximum_cached_engines=1, argument
1400 maximum_cached_engines=maximum_cached_engines,
Dtrt_convert_test.py184 maximum_cached_engines=1, argument
212 maximum_cached_engines=maximum_cached_engines)
381 maximum_cached_engines=2, argument
389 maximum_cached_engines=maximum_cached_engines,
560 maximum_cached_engines=3)
831 maximum_cached_engines=2)
862 output_saved_model_dir=output_saved_model_dir, maximum_cached_engines=1)
Dtrt_convert_windows.py87 maximum_cached_engines=1, argument
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.experimental.tensorrt.-conversion-params.pbtxt15 name: "maximum_cached_engines"
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h97 maximum_cached_engines(0), in EngineInfo()
116 int maximum_cached_engines; member
Dconvert_graph.cc489 .Attr("max_cached_engines_count", info.maximum_cached_engines) in CreateTRTNode()
812 curr_engine.maximum_cached_engines = params.max_cached_engines; in ConvertAfterShapes()
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py286 maximum_cached_engines=1,
473 maximum_cached_engines=conversion_params.maximum_cached_engines,
485 assert conversion_params.maximum_cached_engines == 1
/external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/
Dmodel_handler.py462 maximum_cached_engines=trt_convert_params.maximum_cached_engines,