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.py159 maximum_cached_engines=1, argument
164 minimum_segment_size, maximum_cached_engines,
295 "maximum_cached_engines"].i = conversion_params.maximum_cached_engines
382 maximum_cached_engines=1, argument
475 maximum_cached_engines=maximum_cached_engines,
813 maximum_cached_engines, argument
821 self._maximum_cached_engines = maximum_cached_engines
1197 self._conversion_params.maximum_cached_engines)
1244 maximum_cached_engines=1, argument
1313 maximum_cached_engines=maximum_cached_engines,
Dtrt_convert_test.py182 maximum_cached_engines=1, argument
210 maximum_cached_engines=maximum_cached_engines)
383 maximum_cached_engines=2): argument
390 maximum_cached_engines=maximum_cached_engines))
508 maximum_cached_engines=3)
790 maximum_cached_engines=2)
823 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.h96 maximum_cached_engines(0), in EngineInfo()
114 int maximum_cached_engines; member
Dconvert_graph.cc491 .Attr("max_cached_engines_count", info.maximum_cached_engines) in CreateTRTNode()
780 curr_engine.maximum_cached_engines = params.max_cached_engines; in ConvertAfterShapes()
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py283 maximum_cached_engines=1,
470 maximum_cached_engines=conversion_params.maximum_cached_engines,
482 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,