Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h136 std::vector<EngineConnection>* connections, GraphDef* segment_def,
Dconvert_nodes.cc6457 std::vector<EngineConnection>* connections, GraphDef* segment_def, in ConvertSegmentToGraphDef() argument
6499 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()