Searched refs:rsqrt_var_add_epsilon (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batchnorm_expander.cc | 281 auto rsqrt_var_add_epsilon = add(Rsqrt(var_add_epsilon, add)); in HandleBatchNormTraining() local
289 operand_minus_mean, rsqrt_var_add_epsilon); in HandleBatchNormTraining()
389 auto rsqrt_var_add_epsilon = add(Rsqrt(var_add_epsilon, add)); in HandleBatchNormInference() local
397 operand_minus_mean, rsqrt_var_add_epsilon); in HandleBatchNormInference()
511 auto rsqrt_var_add_epsilon = add(Rsqrt( in HandleBatchNormGrad() local
541 rsqrt_var_add_epsilon); in HandleBatchNormGrad()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 730 auto rsqrt_var_add_epsilon = *ReferenceUtil::MapArray4D( in XLA_TEST_P() local
745 activation_shifted_times_grad_output, rsqrt_var_add_epsilon, in XLA_TEST_P()
757 scale4D, rsqrt_var_add_epsilon, [](float a, float b) { return a * b; }); in XLA_TEST_P()
792 grad_activation, rsqrt_var_add_epsilon, [=](float a, float b) { in XLA_TEST_P()
|