Home
last modified time | relevance | path

Searched refs:MapArray4D (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbatch_normalization_test.cc528 ReferenceUtil::MapArray4D(input_array, [](float a) { return a * a; }); in XLA_TEST_P()
629 ReferenceUtil::MapArray4D(input_array, [](float a) { return a * a; }); in XLA_TEST_P()
740 ReferenceUtil::MapArray4D(input_array, [](float a) { return a * a; }); in XLA_TEST_P()
790 auto var_add_epsilon = *ReferenceUtil::MapArray4D( in XLA_TEST_P()
793 auto rsqrt_var_add_epsilon = *ReferenceUtil::MapArray4D( in XLA_TEST_P()
797 *ReferenceUtil::MapArray4D(grad_output_array, var_add_epsilon, in XLA_TEST_P()
800 auto activation_shifted = *ReferenceUtil::MapArray4D( in XLA_TEST_P()
804 *ReferenceUtil::MapArray4D(grad_output_array, activation_shifted, in XLA_TEST_P()
807 auto grad_scale_before_reduction = *ReferenceUtil::MapArray4D( in XLA_TEST_P()
819 auto scale_times_rsqrt_var_add_epsilon = *ReferenceUtil::MapArray4D( in XLA_TEST_P()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dreference_util_test.cc123 TEST_F(ReferenceUtilTest, MapArray4D) { in TEST_F() argument
128 auto result = ReferenceUtil::MapArray4D(*input, multiply_by_two); in TEST_F()
H A Dreference_util.cc315 *MapArray4D(input, mean, [](float a, float b) { return a - b; }); in BatchNorm4D()
316 normalized = *MapArray4D(normalized, var, [&](float a, float b) { in BatchNorm4D()
320 *MapArray4D(normalized, scale, [](float a, float b) { return a * b; }); in BatchNorm4D()
321 return MapArray4D(normalized, offset, [](float a, float b) { return a + b; }); in BatchNorm4D()
H A Dreference_util.h446 static std::unique_ptr<Array4D<float>> MapArray4D(const Array4D<float>& input, in MapArray4D() function
480 static std::unique_ptr<Array4D<float>> MapArray4D(const Array4D<float>& lhs, in MapArray4D() function