Searched refs:mean_broadcasted (Results 1 – 1 of 1) sorted by relevance
259 auto mean_broadcasted = add( in HandleBatchNormTraining() local285 operand, mean_broadcasted); in HandleBatchNormTraining()378 auto mean_broadcasted = add( in HandleBatchNormInference() local393 operand, mean_broadcasted); in HandleBatchNormInference()502 auto mean_broadcasted = add( in HandleBatchNormGrad() local517 activation_shape, HloOpcode::kSubtract, activation, mean_broadcasted); in HandleBatchNormGrad()