Searched refs:reshape_broadcast_dims (Results 1 – 1 of 1) sorted by relevance
269 absl::flat_hash_set<int64> reshape_broadcast_dims; in AddInstruction() local271 reshape_broadcast_dims.insert(i); in AddInstruction()291 reshape_broadcast_dims.erase(current_after_dim); in AddInstruction()305 reshape_broadcast_dims.erase(i); in AddInstruction()311 reshape_broadcast_dims.clear(); in AddInstruction()313 if (!reshape_broadcast_dims.empty()) { in AddInstruction()314 broadcast_dims_[hlo] = std::move(reshape_broadcast_dims); in AddInstruction()