Searched refs:var_add_epsilon (Results 1 – 2 of 2) sorted by relevance
277 auto var_add_epsilon = in HandleBatchNormTraining() local281 auto rsqrt_var_add_epsilon = add(Rsqrt(var_add_epsilon, add)); in HandleBatchNormTraining()385 auto var_add_epsilon = in HandleBatchNormInference() local389 auto rsqrt_var_add_epsilon = add(Rsqrt(var_add_epsilon, add)); in HandleBatchNormInference()
727 auto var_add_epsilon = *ReferenceUtil::MapArray4D( in XLA_TEST_P() local731 var_add_epsilon, [](float a) { return 1 / std::sqrt(a); }); in XLA_TEST_P()734 *ReferenceUtil::MapArray4D(grad_output_array, var_add_epsilon, in XLA_TEST_P()779 auto I5 = *ReferenceUtil::MapArray4D(I4, var_add_epsilon, in XLA_TEST_P()