Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc844 EngineInfo curr_engine; in ConvertAfterShapes() local
845 curr_engine.engine_name = StrCat("TRTEngineOp_", t); in ConvertAfterShapes()
848 node_map, reverse_topo_order, &curr_engine); in ConvertAfterShapes()
854 curr_engine.precision_mode = params.precision_mode; in ConvertAfterShapes()
855 curr_engine.engine_type = ((params.is_dyn_op || params.use_calibration) in ConvertAfterShapes()
858 curr_engine.use_calibration = params.use_calibration; in ConvertAfterShapes()
859 curr_engine.cached_engine_batches = params.cached_engine_batches; in ConvertAfterShapes()
860 curr_engine.maximum_cached_engines = params.max_cached_engines; in ConvertAfterShapes()
863 &graph, curr_engine.segment_graph_def, curr_engine.engine_name); in ConvertAfterShapes()
871 engine_bytes_size.push_back(curr_engine.segment_graph_def.ByteSizeLong()); in ConvertAfterShapes()
[all …]