Searched refs:add_reduce_computation (Results 1 – 1 of 1) sorted by relevance
240 HloComputation* add_reduce_computation = in HandleBatchNormTraining() local249 add_reduce_computation)); in HandleBatchNormTraining()254 add_reduce_computation)); in HandleBatchNormTraining()524 HloComputation* add_reduce_computation = in HandleBatchNormGrad() local531 dimensions_without_feature, add_reduce_computation)); in HandleBatchNormGrad()536 add_reduce_computation)); in HandleBatchNormGrad()