Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py470 cached_engine_batches=None, argument
550 if cached_engine_batches:
552 cached_engine_batches)
570 cached_engine_batches=None, argument
678 if cached_engine_batches:
679 if not isinstance(cached_engine_batches, list):
681 if len(cached_engine_batches) > maximum_cached_engines:
707 self._cached_engine_batches = cached_engine_batches
718 cached_engine_batches=self._cached_engine_batches,
790 cached_engine_batches=None, argument
[all …]
Dtrt_convert_test.py72 cached_engine_batches=[1, 128])
/external/tensorflow/tensorflow/contrib/tensorrt/python/
Dtrt_convert.py33 cached_engine_batches=None, argument
47 cached_engine_batches=cached_engine_batches,
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.h74 std::vector<int> cached_engine_batches; member
89 int max_cached_engines = 1, std::vector<int> cached_engine_batches = {},
Dconvert_graph.cc106 std::vector<int> cached_engine_batches, bool use_calibration) { in ConvertGraphDefToTensorRT() argument
166 if (!cached_engine_batches.empty()) { in ConvertGraphDefToTensorRT()
168 for (const int batch : cached_engine_batches) { in ConvertGraphDefToTensorRT()
550 !info.cached_engine_batches.empty()) { in CreateTRTNode()
859 curr_engine.cached_engine_batches = params.cached_engine_batches; in ConvertAfterShapes()
Dtrt_optimization_pass.cc265 cp.cached_engine_batches = batches_; in Optimize()
Dconvert_nodes.h117 std::vector<int> cached_engine_batches; member
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py184 cached_engine_batches=None,
229 conversion_params.cached_engine_batches,
311 cached_engine_batches=conversion_params.cached_engine_batches,
323 assert not conversion_params.cached_engine_batches