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.cc465 ReferenceUtil::MapArray4D(input_array, [](float a) { return a * a; }); in XLA_TEST_P()
566 ReferenceUtil::MapArray4D(input_array, [](float a) { return a * a; }); in XLA_TEST_P()
677 ReferenceUtil::MapArray4D(input_array, [](float a) { return a * a; }); in XLA_TEST_P()
727 auto var_add_epsilon = *ReferenceUtil::MapArray4D( in XLA_TEST_P()
730 auto rsqrt_var_add_epsilon = *ReferenceUtil::MapArray4D( in XLA_TEST_P()
734 *ReferenceUtil::MapArray4D(grad_output_array, var_add_epsilon, in XLA_TEST_P()
737 auto activation_shifted = *ReferenceUtil::MapArray4D( in XLA_TEST_P()
741 *ReferenceUtil::MapArray4D(grad_output_array, activation_shifted, in XLA_TEST_P()
744 auto grad_scale_before_reduction = *ReferenceUtil::MapArray4D( in XLA_TEST_P()
756 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.cc379 *MapArray4D(input, mean, [](float a, float b) { return a - b; }); in BatchNorm4D()
380 normalized = *MapArray4D(normalized, var, [&](float a, float b) { in BatchNorm4D()
384 *MapArray4D(normalized, scale, [](float a, float b) { return a * b; }); in BatchNorm4D()
385 return MapArray4D(normalized, offset, [](float a, float b) { return a + b; }); in BatchNorm4D()
Dreference_util.h442 static std::unique_ptr<Array4D<float>> MapArray4D(const Array4D<float>& input, in MapArray4D() function
476 static std::unique_ptr<Array4D<float>> MapArray4D(const Array4D<float>& lhs, in MapArray4D() function