Searched refs:reachable_nodes (Results 1 – 2 of 2) sorted by relevance
99 const std::vector<bool>& reachable_nodes,169 std::vector<bool> reachable_nodes(scope_.graph()->num_node_ids(), false); in GetReachableNodes() local172 if (!reachable_nodes[out.node()->id()]) { in GetReachableNodes()174 reachable_nodes[out.node()->id()] = true; in GetReachableNodes()183 if (!reachable_nodes[e->src()->id()]) { in GetReachableNodes()185 reachable_nodes[e->src()->id()] = true; in GetReachableNodes()189 return reachable_nodes; in GetReachableNodes()193 const std::vector<bool>& reachable_nodes, in GetStopBackpropNodes() argument219 if (e->IsControlEdge() || !reachable_nodes[e->dst()->id()]) continue; in GetStopBackpropNodes()255 std::vector<bool> reachable_nodes = GetReachableNodes(); in Initialize() local[all …]
197 reachable_nodes{{0, id}}; in MarkRoot() local198 while (!reachable_nodes.empty()) { in MarkRoot()201 std::tie(distance, cur_node) = reachable_nodes.front(); in MarkRoot()202 reachable_nodes.pop_front(); in MarkRoot()222 reachable_nodes.emplace_back(distance + 1, child_node); in MarkRoot()