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

  /external/tensorflow/tensorflow/compiler/xla/tests/
convolution_dimension_numbers_test.cc 122 auto expected_conv1 = ReferenceUtil::ConvArray4D(*input_array, *weight_array,
convolution_variants_test.cc 828 std::unique_ptr<Array4D<float>> expected = ReferenceUtil::ConvArray4D(
861 std::unique_ptr<Array4D<float>> expected = ReferenceUtil::ConvArray4D(
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/
reference_util_test.cc 258 ReferenceUtil::ConvArray4D(input, weights, {1, 1}, Padding::kSame);
293 ReferenceUtil::ConvArray4D(input, weights, {1, 1}, Padding::kValid);
reference_util.cc 109 /* static */ std::unique_ptr<Array4D<float>> ReferenceUtil::ConvArray4D(
146 return ConvArray4D(input, weights, kernel_stride, padding);
reference_util.h 71 static std::unique_ptr<Array4D<float>> ConvArray4D(

Completed in 194 milliseconds