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.cc768 EngineInfo curr_engine; in ConvertAfterShapes() local
769 curr_engine.engine_name = StrCat(engine_name_prefix, t); in ConvertAfterShapes()
771 node_map, reverse_topo_order, &curr_engine); in ConvertAfterShapes()
777 curr_engine.precision_mode = params.precision_mode; in ConvertAfterShapes()
778 curr_engine.engine_type = GetEngineType(params); in ConvertAfterShapes()
779 curr_engine.use_calibration = params.use_calibration; in ConvertAfterShapes()
780 curr_engine.maximum_cached_engines = params.max_cached_engines; in ConvertAfterShapes()
781 curr_engine.allow_build_at_runtime = params.allow_build_at_runtime; in ConvertAfterShapes()
782 if (!curr_engine.max_batch_size.has_value()) { in ConvertAfterShapes()
783 curr_engine.max_batch_size = params.max_batch_size; in ConvertAfterShapes()
[all …]