Searched refs:offset_broadcasted (Results 1 – 1 of 1) sorted by relevance
237 auto offset_broadcasted = add( in HandleBatchNormTraining() local297 scaled_normalized, offset_broadcasted); in HandleBatchNormTraining()375 auto offset_broadcasted = add( in HandleBatchNormInference() local405 operand_shape, HloOpcode::kAdd, scaled_normalized, offset_broadcasted); in HandleBatchNormInference()