Searched defs:add_reduce_computation (Results 1 – 1 of 1) sorted by relevance
240 HloComputation* add_reduce_computation = in HandleBatchNormTraining() local524 HloComputation* add_reduce_computation = in HandleBatchNormGrad() local