Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.h71 int max_cached_engines = 1; member
89 int max_cached_engines = 1, std::vector<int> cached_engine_batches = {},
Dconvert_graph.cc105 int minimum_segment_size, bool is_dyn_op, int max_cached_engines, in ConvertGraphDefToTensorRT() argument
165 parameters["maximum_cached_engines"].set_i(max_cached_engines); in ConvertGraphDefToTensorRT()
860 curr_engine.maximum_cached_engines = params.max_cached_engines; in ConvertAfterShapes()
Dtrt_optimization_pass.cc266 cp.max_cached_engines = max_cached_batches_; in Optimize()