Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc758 std::vector<EngineInfo> engine_segments; in ConvertAfterShapes() local
759 engine_segments.reserve(initial_segments.size()); in ConvertAfterShapes()
796 engine_segments.push_back(std::move(curr_engine)); in ConvertAfterShapes()
800 string fname = engine_segments.back().engine_name; in ConvertAfterShapes()
804 f << engine_segments.at(t).segment_graph_def.SerializeAsString(); in ConvertAfterShapes()
831 engine_nodes.resize(engine_segments.size()); in ConvertAfterShapes()
832 for (int i = 0; i < engine_segments.size(); ++i) { in ConvertAfterShapes()
833 auto& engine = engine_segments.at(i); in ConvertAfterShapes()
839 auto status = CreateTRTNode(params, engine_segments, i, in ConvertAfterShapes()