Searched refs:pad_broadcast_dims (Results 1 – 1 of 1) sorted by relevance
329 absl::flat_hash_set<int64> pad_broadcast_dims; in AddInstruction() local334 pad_broadcast_dims.insert(i); in AddInstruction()337 if (!pad_broadcast_dims.empty()) { in AddInstruction()338 broadcast_dims_[hlo] = std::move(pad_broadcast_dims); in AddInstruction()