Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/
Dtrt_convert_api.h58 int max_cached_engines = 1; member
Dtrt_convert_api.cc144 (*trt_parameter_map)["max_cached_engines"].set_i(params.max_cached_engines); in GetTrtRewriterConfig()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
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()
Dtrt_optimization_pass.h52 int max_cached_engines = 1; member
Dconvert_graph.cc842 curr_engine.maximum_cached_engines = params.max_cached_engines; in ConvertGraph()