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.cc572 ReferenceUtil::Reduce4DTo1D(input_array, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
576 ReferenceUtil::Reduce4DTo1D(*input_squared, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
673 ReferenceUtil::Reduce4DTo1D(input_array, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
677 ReferenceUtil::Reduce4DTo1D(*input_squared, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
784 ReferenceUtil::Reduce4DTo1D(input_array, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
788 ReferenceUtil::Reduce4DTo1D(*input_squared, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
846 auto grad_scale = ReferenceUtil::Reduce4DTo1D( in XLA_TEST_P()
851 ReferenceUtil::Reduce4DTo1D(grad_output_array, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
864 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.cc539 /*static*/ std::vector<float> ReferenceUtil::Reduce4DTo1D( in Reduce4DTo1D() function in xla::ReferenceUtil