Searched refs:segment_def (Results 1 – 2 of 2) sorted by relevance
136 std::vector<EngineConnection>* connections, GraphDef* segment_def,
6457 std::vector<EngineConnection>* connections, GraphDef* segment_def, in ConvertSegmentToGraphDef() argument6499 auto seg_node = segment_def->add_node(); in ConvertSegmentToGraphDef()6520 auto seg_node = segment_def->add_node(); in ConvertSegmentToGraphDef()6539 old_to_new_id_map[node->id()] = segment_def->node_size(); in ConvertSegmentToGraphDef()6540 auto snode = segment_def->add_node(); in ConvertSegmentToGraphDef()6549 segment_def->mutable_node(old_to_new_id_map[connection.inside_id]); in ConvertSegmentToGraphDef()6563 for (int i = 0; i < segment_def->node_size(); ++i) { in ConvertSegmentToGraphDef()6564 auto snode = segment_def->mutable_node(i); in ConvertSegmentToGraphDef()