Searched refs:new_broadcast (Results 1 – 2 of 2) sorted by relevance
1038 auto new_broadcast = MakeBroadcastHlo(consumer->mutable_operand(0), in PropagateOnBroadcast() local1040 VLOG(1) << "Created broadcast " << new_broadcast->ToString(); in PropagateOnBroadcast()1043 new_broadcast = in PropagateOnBroadcast()1044 MakeReshapeHlo(new_producer->shape().dimensions(), new_broadcast) in PropagateOnBroadcast()1046 VLOG(2) << "Created reshape of broadcast " << new_broadcast->ToString(); in PropagateOnBroadcast()1053 broadcast_map_[consumer].insert(new_broadcast); in PropagateOnBroadcast()1242 HloInstruction* new_broadcast = nullptr; in Propagate() local1247 new_broadcast = previous_broadcast; in Propagate()1251 CHECK_NE(new_broadcast, nullptr); in Propagate()1253 new_consumer->ReplaceOperandWithDifferentShape(i, new_broadcast)); in Propagate()
746 HloInstruction* new_broadcast = computation_->AddInstruction( in ScalarMultiplyReduction() local751 target->shape(), HloOpcode::kMultiply, target, new_broadcast)); in ScalarMultiplyReduction()3117 HloInstruction* new_broadcast = computation_->AddInstruction( in HandleBroadcast() local3123 return user->ReplaceAllUsesWith(new_broadcast); in HandleBroadcast()3672 HloInstruction* new_broadcast = in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand() local3675 broadcast->SetupDerivedInstruction(new_broadcast); in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand()3676 VLOG(4) << " new broadcast: " << new_broadcast->ToString(); in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand()3677 TF_RETURN_IF_ERROR(user->ReplaceAllUsesWith(new_broadcast)); in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand()4271 auto new_broadcast = HloInstruction::CreateBroadcast( in HandleSlice() local4274 VLOG(3) << "New broadcast: " << new_broadcast->ToString(); in HandleSlice()[all …]