MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / GetLabeledNodes

Function GetLabeledNodes

tensorflow/compiler/tf2tensorrt/segment/segment.cc:45–59  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

43namespace {
44
45void GetLabeledNodes(gtl::FlatSet<string>* node_set, Graph* g) {
46 std::unordered_set<Node*> boundary_node_set;
47 graph_util::GetComputeGraphBoundaryNodes(g, boundary_node_set);
48
49 auto label_node_func = [node_set](Node* n) {
50 node_set->insert(n->name());
51 };
52
53 std::vector<Node* > boundary_node_vec;
54 for (const auto node : boundary_node_set) {
55 boundary_node_vec.emplace_back(node);
56 }
57 ReverseDFSFrom(*g, boundary_node_vec,
58 std::move(label_node_func), nullptr);
59}
60
61} // namespace
62

Callers 1

SegmentGraphFunction · 0.85

Calls 5

ReverseDFSFromFunction · 0.85
nameMethod · 0.65
insertMethod · 0.45
emplace_backMethod · 0.45

Tested by

no test coverage detected