HomeSort by relevance Sort by last modified time
    Searched refs:MapArray4D (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
batch_normalization_test.cc 456 ReferenceUtil::MapArray4D(input_array, [](float a) { return a * a; });
556 ReferenceUtil::MapArray4D(input_array, [](float a) { return a * a; });
667 ReferenceUtil::MapArray4D(input_array, [](float a) { return a * a; });
717 auto var_add_epsilon = *ReferenceUtil::MapArray4D(
720 auto rsqrt_var_add_epsilon = *ReferenceUtil::MapArray4D(
724 *ReferenceUtil::MapArray4D(grad_output_array, var_add_epsilon,
727 auto activation_shifted = *ReferenceUtil::MapArray4D(
731 *ReferenceUtil::MapArray4D(grad_output_array, activation_shifted,
734 auto grad_scale_before_reduction = *ReferenceUtil::MapArray4D(
746 auto scale_times_rsqrt_var_add_epsilon = *ReferenceUtil::MapArray4D(
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/
reference_util_test.cc 114 TEST_F(ReferenceUtilTest, MapArray4D) {
119 auto result = ReferenceUtil::MapArray4D(*input, multiply_by_two);
reference_util.cc 413 *MapArray4D(input, mean, [](float a, float b) { return a - b; });
414 normalized = *MapArray4D(normalized, var, [&](float a, float b) {
418 *MapArray4D(normalized, scale, [](float a, float b) { return a * b; });
419 return MapArray4D(normalized, offset, [](float a, float b) { return a + b; });
reference_util.h 433 static std::unique_ptr<Array4D<float>> MapArray4D(const Array4D<float>& input,
467 static std::unique_ptr<Array4D<float>> MapArray4D(const Array4D<float>& lhs,

Completed in 169 milliseconds