HomeSort by relevance Sort by last modified time
    Searched refs:Reduce4DTo1D (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
batch_normalization_test.cc 474 ReferenceUtil::Reduce4DTo1D(input_array, /*init=*/0.0f, reduce_dims,
478 ReferenceUtil::Reduce4DTo1D(*input_squared, /*init=*/0.0f, reduce_dims,
575 ReferenceUtil::Reduce4DTo1D(input_array, /*init=*/0.0f, reduce_dims,
579 ReferenceUtil::Reduce4DTo1D(*input_squared, /*init=*/0.0f, reduce_dims,
686 ReferenceUtil::Reduce4DTo1D(input_array, /*init=*/0.0f, reduce_dims,
690 ReferenceUtil::Reduce4DTo1D(*input_squared, /*init=*/0.0f, reduce_dims,
748 auto grad_scale = ReferenceUtil::Reduce4DTo1D(
753 ReferenceUtil::Reduce4DTo1D(grad_output_array, /*init=*/0.0f, reduce_dims,
766 ReferenceUtil::Reduce4DTo1D(activation_shifted_times_grad_output,
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/
reference_util_test.cc 90 auto result = LiteralUtil::CreateR1<float>(ReferenceUtil::Reduce4DTo1D(
reference_util.h 146 static std::vector<float> Reduce4DTo1D(
reference_util.cc 604 /*static*/ std::vector<float> ReferenceUtil::Reduce4DTo1D(

Completed in 373 milliseconds