HomeSort by relevance Sort by last modified time
    Searched full:grad_activation (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
batch_normalization_test.cc 772 auto grad_activation = *ReferenceUtil::MapArray4D( local
775 grad_activation = *ReferenceUtil::MapArray4D(
776 grad_activation, I5, [](float a, float b) { return a - b; });
778 grad_activation = *ReferenceUtil::MapArray4D(
779 grad_activation, scale4D, [](float a, float b) { return a * b; });
781 grad_activation = *ReferenceUtil::MapArray4D(
782 grad_activation, rsqrt_var_add_epsilon, [=](float a, float b) {
790 Literal::CreateR4FromArray4D<float>(grad_activation);
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
batchnorm_expander.cc 574 auto grad_activation = local
578 HloInstruction::CreateTuple({grad_activation, grad_scale, grad_beta});

Completed in 77 milliseconds