Home
last modified time | relevance | path

Searched refs:max_cached_engines (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dtrt_optimization_pass.cc103 GetNodeAttr(attr, "_tftrt_max_cached_engines", &cp.max_cached_engines)); in UpdateFunctionSpecificConversionParams()
135 params_.max_cached_engines = params.at("maximum_cached_engines").i(); in Init()
H A Dtrt_optimization_pass.h52 int max_cached_engines = 1; member
H A Dconvert_graph.cc842 curr_engine.maximum_cached_engines = params.max_cached_engines; in ConvertGraph()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/
H A Dtrt_convert_api.h58 int max_cached_engines = 1; member
H A Dtrt_convert_api.cc144 (*trt_parameter_map)["max_cached_engines"].set_i(params.max_cached_engines); in GetTrtRewriterConfig()