Searched refs:engine_segments (Results 1 – 1 of 1) sorted by relevance
834 std::vector<EngineInfo> engine_segments; in ConvertAfterShapes() local835 engine_segments.reserve(initial_segments.size()); in ConvertAfterShapes()874 engine_segments.push_back(std::move(curr_engine)); in ConvertAfterShapes()878 string fname = engine_segments.back().engine_name; in ConvertAfterShapes()882 f << engine_segments.at(t).segment_graph_def.SerializeAsString(); in ConvertAfterShapes()895 engine_nodes.resize(engine_segments.size()); in ConvertAfterShapes()896 for (int i = 0; i < engine_segments.size(); ++i) { in ConvertAfterShapes()897 auto& engine = engine_segments.at(i); in ConvertAfterShapes()920 CreateTRTNode(params, engine_segments, i, params.max_batch_size, &graph, in ConvertAfterShapes()