Searched refs:visited_count_ (Results 1 – 1 of 1) sorted by relevance
815 absl::flat_hash_map<HloInstruction*, int>& visited_count_; member in xla::conditional_opt::GroupConnectedBoundaries885 visited_count_(visited_count), in GroupConnectedBoundaries()917 visited_count_.erase(boundary.operands()[0]); in clear_recently_visited()966 if (!ContainsKey(visited_count_, op) && op != conditional_) { in ReusesBeforeBoundary()1019 } else if (ContainsKey(visited_count_, op)) { in ReusesAfterBoundary()1141 if (!ContainsKey(visited_count_, next_boundary.operands()[0])) { in IsSafeToMoveBoundary()1145 visited_count_[next_boundary.operands()[0]] = 1; in IsSafeToMoveBoundary()1152 int count = ++visited_count_[next_boundary.operands()[0]]; in IsSafeToMoveBoundary()1158 visited_count_.erase(next_boundary.operands()[0]); in IsSafeToMoveBoundary()1200 visited_count_[b.operands()[0]] = 1; in AddBoundaries()