Searched refs:scale_times_rsqrt_var_add_epsilon (Results 1 – 2 of 2) sorted by relevance
562 auto scale_times_rsqrt_var_add_epsilon = in HandleBatchNormGrad() local566 scale_times_rsqrt_var_add_epsilon = in HandleBatchNormGrad()567 add(Mean(elements_per_feature, scale_times_rsqrt_var_add_epsilon, add)); in HandleBatchNormGrad()580 scale_times_rsqrt_var_add_epsilon, i6); in HandleBatchNormGrad()
756 auto scale_times_rsqrt_var_add_epsilon = *ReferenceUtil::MapArray4D( in XLA_TEST_P() local