Searched refs:grad_activation (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 782 auto grad_activation = *ReferenceUtil::MapArray4D( in XLA_TEST_P() local
785 grad_activation = *ReferenceUtil::MapArray4D( in XLA_TEST_P()
786 grad_activation, I5, [](float a, float b) { return a - b; }); in XLA_TEST_P()
788 grad_activation = *ReferenceUtil::MapArray4D( in XLA_TEST_P()
789 grad_activation, scale4D, [](float a, float b) { return a * b; }); in XLA_TEST_P()
791 grad_activation = *ReferenceUtil::MapArray4D( in XLA_TEST_P()
792 grad_activation, rsqrt_var_add_epsilon, [=](float a, float b) { in XLA_TEST_P()
800 LiteralUtil::CreateR4FromArray4D<float>(grad_activation); in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batchnorm_expander.cc | 579 auto grad_activation = add_binary(activation_shape, HloOpcode::kMultiply, in HandleBatchNormGrad() local
582 HloInstruction::CreateTuple({grad_activation, grad_scale, grad_beta}); in HandleBatchNormGrad()
|