Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc730 std::vector<UnionFind<SimpleNode*>> node_segments; in SegmentGraph() local
781 AddSegmentForNode(graph_properties, &node_segments, node, *device_name, in SegmentGraph()
818 if (node_segments[node->id()].Value() == nullptr) { in SegmentGraph()
826 node_segments[node->id()].Property().BatchSize(); in SegmentGraph()
828 node_segments[node->id()].Property().DeviceName(); in SegmentGraph()
842 &node_segments[out_edge->dst()->id()]; in SegmentGraph()
891 node_segments[src->id()].Merge(&node_segments[dst->id()])); in SegmentGraph()
906 node_segments[node->id()].Property().BatchSize()) { in SegmentGraph()
911 node_segments[node->id()].Property().DeviceName()) { in SegmentGraph()
925 for (auto& u : node_segments) { in SegmentGraph()