Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dbatch_normalization_test.cc563 ReferenceUtil::MapArray4D(input_array, [](float a) { return a * a; }); in XLA_TEST_P()
664 ReferenceUtil::MapArray4D(input_array, [](float a) { return a * a; }); in XLA_TEST_P()
775 ReferenceUtil::MapArray4D(input_array, [](float a) { return a * a; }); in XLA_TEST_P()
825 auto var_add_epsilon = *ReferenceUtil::MapArray4D( in XLA_TEST_P()
828 auto rsqrt_var_add_epsilon = *ReferenceUtil::MapArray4D( in XLA_TEST_P()
832 *ReferenceUtil::MapArray4D(grad_output_array, var_add_epsilon, in XLA_TEST_P()
835 auto activation_shifted = *ReferenceUtil::MapArray4D( in XLA_TEST_P()
839 *ReferenceUtil::MapArray4D(grad_output_array, activation_shifted, in XLA_TEST_P()
842 auto grad_scale_before_reduction = *ReferenceUtil::MapArray4D( in XLA_TEST_P()
854 auto scale_times_rsqrt_var_add_epsilon = *ReferenceUtil::MapArray4D( in XLA_TEST_P()
[all …]
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util_test.cc114 TEST_F(ReferenceUtilTest, MapArray4D) { in TEST_F() argument
119 auto result = ReferenceUtil::MapArray4D(*input, multiply_by_two); in TEST_F()
Dreference_util.cc313 *MapArray4D(input, mean, [](float a, float b) { return a - b; }); in BatchNorm4D()
314 normalized = *MapArray4D(normalized, var, [&](float a, float b) { in BatchNorm4D()
318 *MapArray4D(normalized, scale, [](float a, float b) { return a * b; }); in BatchNorm4D()
319 return MapArray4D(normalized, offset, [](float a, float b) { return a + b; }); in BatchNorm4D()
Dreference_util.h434 static std::unique_ptr<Array4D<float>> MapArray4D(const Array4D<float>& input, in MapArray4D() function
468 static std::unique_ptr<Array4D<float>> MapArray4D(const Array4D<float>& lhs, in MapArray4D() function