Searched defs:rsqrt_var_add_epsilon (Results 1 – 2 of 2) sorted by relevance
281 auto rsqrt_var_add_epsilon = add(Rsqrt(var_add_epsilon, add)); in HandleBatchNormTraining() local389 auto rsqrt_var_add_epsilon = add(Rsqrt(var_add_epsilon, add)); in HandleBatchNormInference() local511 auto rsqrt_var_add_epsilon = add(Rsqrt( in HandleBatchNormGrad() local
730 auto rsqrt_var_add_epsilon = *ReferenceUtil::MapArray4D( in XLA_TEST_P() local