Searched refs:broadcast_dims_ (Results 1 – 2 of 2) sorted by relevance
90 auto it = broadcast_dims_.find(hlo); in BroadcastDimsForCreatedHlo()91 if (it == broadcast_dims_.end()) { in BroadcastDimsForCreatedHlo()109 broadcast_dims_; variable
226 broadcast_dims_[hlo].insert(i); in AddInstruction()236 auto it = broadcast_dims_.find(hlo->operand(i)); in AddInstruction()237 if (it == broadcast_dims_.end()) { in AddInstruction()248 broadcast_dims_[hlo] = std::move(broadcast_dims); in AddInstruction()252 auto it = broadcast_dims_.find(hlo->operand(0)); in AddInstruction()253 if (it != broadcast_dims_.end()) { in AddInstruction()262 broadcast_dims_[hlo] = std::move(xpose_broadcast_dims); in AddInstruction()267 auto it = broadcast_dims_.find(hlo->operand(0)); in AddInstruction()268 if (it != broadcast_dims_.end()) { in AddInstruction()314 broadcast_dims_[hlo] = std::move(reshape_broadcast_dims); in AddInstruction()[all …]