Home
last modified time | relevance | path

Searched refs:max_cached_engines_count (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op_test.cc62 void AddSimpleTrtOp(DataType dtype, int max_cached_engines_count = 1, in AddSimpleTrtOp() argument
109 .Attr("max_cached_engines_count", max_cached_engines_count)
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py830 max_cached_engines_count=self._maximum_cached_engines)