Searched refs:maximum_cached_engines (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensorrt/python/ |
| D | trt_convert.py | 32 maximum_cached_engines=1, argument
46 maximum_cached_engines=maximum_cached_engines,
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert.py | 469 maximum_cached_engines=1, argument
549 optimizer.parameter_map["maximum_cached_engines"].i = maximum_cached_engines
569 maximum_cached_engines=1, argument
681 if len(cached_engine_batches) > maximum_cached_engines:
706 self._maximum_cached_engines = maximum_cached_engines
717 maximum_cached_engines=self._maximum_cached_engines,
789 maximum_cached_engines=1, argument
870 maximum_cached_engines=maximum_cached_engines,
|
| D | trt_convert_test.py | 71 maximum_cached_engines=2,
176 maximum_cached_engines=1, argument
191 maximum_cached_engines=maximum_cached_engines,
383 maximum_cached_engines=2,
421 maximum_cached_engines=2, # This is noop, added just for testing.
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 183 maximum_cached_engines=1,
228 conversion_params.maximum_cached_engines,
310 maximum_cached_engines=conversion_params.maximum_cached_engines,
322 assert conversion_params.maximum_cached_engines == 1
|
| D | int32_test.py | 65 maximum_cached_engines=1,
|
| D | dynamic_input_shapes_test.py | 85 maximum_cached_engines=10,
|
| D | biasadd_matmul_test.py | 124 maximum_cached_engines=1,
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 116 int maximum_cached_engines; member
|
| D | convert_graph.cc | 565 .Attr("max_cached_engines_count", info.maximum_cached_engines) in CreateTRTNode()
860 curr_engine.maximum_cached_engines = params.max_cached_engines; in ConvertAfterShapes()
|