Searched refs:scale_broadcasted (Results 1 – 1 of 1) sorted by relevance
234 auto scale_broadcasted = add( in HandleBatchNormTraining() local293 normalized, scale_broadcasted); in HandleBatchNormTraining()372 auto scale_broadcasted = add( in HandleBatchNormInference() local401 normalized, scale_broadcasted); in HandleBatchNormInference()496 auto scale_broadcasted = add(HloInstruction::CreateBroadcast( in HandleBatchNormGrad() local563 add_binary(activation_shape, HloOpcode::kMultiply, scale_broadcasted, in HandleBatchNormGrad()