Searched refs:broadcast_map_ (Results 1 – 1 of 1) sorted by relevance
236 broadcast_map_; member in xla::__anon6418faad0111::ConvolutionVisitor1014 bool map_found = broadcast_map_.contains(consumer); in PropagateOnBroadcast()1017 for (auto previous_broadcast : broadcast_map_[consumer]) { in PropagateOnBroadcast()1051 broadcast_map_[consumer] = set_of_broadcasts; in PropagateOnBroadcast()1053 broadcast_map_[consumer].insert(new_broadcast); in PropagateOnBroadcast()1244 for (auto previous_broadcast : broadcast_map_[broadcast]) { in Propagate()