Searched refs:scaled_normalized (Results 1 – 1 of 1) sorted by relevance
292 auto scaled_normalized = add_binary(operand_shape, HloOpcode::kMultiply, in HandleBatchNormTraining() local297 scaled_normalized, offset_broadcasted); in HandleBatchNormTraining()400 auto scaled_normalized = add_binary(operand_shape, HloOpcode::kMultiply, in HandleBatchNormInference() local405 operand_shape, HloOpcode::kAdd, scaled_normalized, offset_broadcasted); in HandleBatchNormInference()