Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/
H A Dtrt_convert.py174 maximum_cached_engines=1, argument
179 minimum_segment_size, maximum_cached_engines,
348 "maximum_cached_engines"].i = conversion_params.maximum_cached_engines
440 maximum_cached_engines=1, argument
533 maximum_cached_engines=maximum_cached_engines,
856 maximum_cached_engines, argument
863 self._maximum_cached_engines = maximum_cached_engines
1152 maximum_cached_engines=1, argument
1205 maximum_cached_engines=maximum_cached_engines,
1551 self._conversion_params.maximum_cached_engines)
[all …]
H A Dtrt_convert_test.py184 maximum_cached_engines=1, argument
212 maximum_cached_engines=maximum_cached_engines)
381 maximum_cached_engines=2, argument
388 maximum_cached_engines=maximum_cached_engines,
558 maximum_cached_engines=3)
829 maximum_cached_engines=2)
860 output_saved_model_dir=output_saved_model_dir, maximum_cached_engines=1)
1158 maximum_cached_engines=1)
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.h112 maximum_cached_engines(0), in EngineInfo()
132 int maximum_cached_engines; member
H A Dconvert_graph.cc498 .Attr("max_cached_engines_count", info.maximum_cached_engines) in CreateTRTNode()
842 curr_engine.maximum_cached_engines = params.max_cached_engines; in ConvertGraph()
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
H A Dtf_trt_integration_test_base.py287 maximum_cached_engines=1,
484 maximum_cached_engines=conversion_params.maximum_cached_engines,
498 assert conversion_params.maximum_cached_engines == 1, (
H A Dquantization_mnist_test.py345 maximum_cached_engines=1)
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.experimental.tensorrt.-conversion-params.pbtxt15 name: "maximum_cached_engines"
H A Dtensorflow.experimental.tensorrt.-converter.pbtxt7 …ce_size_bytes\', \'precision_mode\', \'minimum_segment_size\', \'maximum_cached_engines\', \'use_c…
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/
H A Dmodel_handler.py462 maximum_cached_engines=trt_convert_params.maximum_cached_engines,
/aosp_15_r20/external/tensorflow/
H A DRELEASE.md481 `minimum_segment_size`, `maximum_cached_engines`, `use_calibration` and