Searched refs:segment_nodes (Results 1 – 2 of 2) sorted by relevance
582 std::set<const Node*, NodePtrCompare>& segment_nodes = itr.second; in SegmentGraph() local583 VLOG(1) << "Segment original size: " << segment_nodes.size(); in SegmentGraph()589 for (auto node : segment_nodes) { in SegmentGraph()593 !segment_nodes.count(edge->src())) { // 'node' is an input node. in SegmentGraph()604 !segment_nodes.count(edge->dst())) { // 'node' is an output node. in SegmentGraph()630 auto remove_nodes = [&segment_nodes](bool is_input_nodes, in SegmentGraph()639 segment_nodes.erase(node); in SegmentGraph()643 if (segment_nodes.count(in)) { in SegmentGraph()662 VLOG(1) << "Segment new size: " << segment_nodes.size(); in SegmentGraph()670 std::set<const Node*> segment_nodes(itr.second.begin(), itr.second.end()); in SegmentGraph() local[all …]
197 const std::set<const Node*>& segment_nodes, in GetEngineInfo() argument215 if (segment_nodes.count(node) == 0) continue; in GetEngineInfo()252 if (input_node->IsSource() || segment_nodes.count(input_node)) { in GetEngineInfo()304 if (output_node->IsSink() || segment_nodes.count(output_node)) { in GetEngineInfo()