Searched refs:elements_per_feature (Results 1 – 1 of 1) sorted by relevance
231 auto elements_per_feature = in HandleBatchNormTraining() local257 auto mean = add(Mean(elements_per_feature, sum, add)); in HandleBatchNormTraining()263 auto square_mean = add(Mean(elements_per_feature, squared_sum, add)); in HandleBatchNormTraining()472 auto elements_per_feature = in HandleBatchNormGrad() local567 add(Mean(elements_per_feature, scale_times_rsqrt_var_add_epsilon, add)); in HandleBatchNormGrad()571 activation_shape, elements_per_feature, {}))); in HandleBatchNormGrad()