Searched refs:segment_nodes (Results 1 – 2 of 2) sorted by relevance
937 std::set<const Node*, NodePtrCompare>& segment_nodes = itr.second.nodes; in SegmentGraph() local938 VLOG(1) << "Segment original size: " << segment_nodes.size(); in SegmentGraph()944 for (auto node : segment_nodes) { in SegmentGraph()948 !segment_nodes.count(edge->src())) { // 'node' is an input node. in SegmentGraph()959 !segment_nodes.count(edge->dst())) { // 'node' is an output node. in SegmentGraph()985 auto remove_nodes = [&segment_nodes](bool is_input_nodes, in SegmentGraph()994 segment_nodes.erase(node); in SegmentGraph()998 if (segment_nodes.count(in)) { in SegmentGraph()1017 VLOG(1) << "Segment new size: " << segment_nodes.size(); in SegmentGraph()1025 std::set<const Node*, NodePtrCompare> segment_nodes( in SegmentGraph() local[all …]
140 const std::set<const Node*, NodePtrCompare>& segment_nodes = segment.nodes; in GetEngineInfo() local158 if (segment_nodes.count(node) == 0) continue; in GetEngineInfo()171 if (input_node->IsSource() || segment_nodes.count(input_node)) { in GetEngineInfo()219 if (output_node->IsSink() || segment_nodes.count(output_node)) { in GetEngineInfo()