Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
H A Dtrt_engine_op_test.cc67 void AddSimpleTrtOp(DataType dtype, int max_cached_engines_count = 1, in AddSimpleTrtOp() argument
145 .Attr("max_cached_engines_count", max_cached_engines_count)
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/
H A Dtrt_convert.py872 max_cached_engines_count=self._maximum_cached_engines)