Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dbatch_normalization_test.cc474 ReferenceUtil::Reduce4DTo1D(input_array, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
478 ReferenceUtil::Reduce4DTo1D(*input_squared, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
575 ReferenceUtil::Reduce4DTo1D(input_array, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
579 ReferenceUtil::Reduce4DTo1D(*input_squared, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
686 ReferenceUtil::Reduce4DTo1D(input_array, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
690 ReferenceUtil::Reduce4DTo1D(*input_squared, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
748 auto grad_scale = ReferenceUtil::Reduce4DTo1D( in XLA_TEST_P()
753 ReferenceUtil::Reduce4DTo1D(grad_output_array, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
766 ReferenceUtil::Reduce4DTo1D(activation_shifted_times_grad_output, in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util_test.cc90 auto result = LiteralUtil::CreateR1<float>(ReferenceUtil::Reduce4DTo1D( in TEST_F()
Dreference_util.h146 static std::vector<float> Reduce4DTo1D(
Dreference_util.cc604 /*static*/ std::vector<float> ReferenceUtil::Reduce4DTo1D( in Reduce4DTo1D() function in xla::ReferenceUtil