Searched refs:ConvArray4D (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util_test.cc | 258 ReferenceUtil::ConvArray4D(input, weights, {1, 1}, Padding::kSame); in TEST_F()
293 ReferenceUtil::ConvArray4D(input, weights, {1, 1}, Padding::kValid); in TEST_F()
|
| D | reference_util.cc | 109 /* static */ std::unique_ptr<Array4D<float>> ReferenceUtil::ConvArray4D( in ConvArray4D() function in xla::ReferenceUtil
146 return ConvArray4D(input, weights, kernel_stride, padding); in SeparableConvArray4D()
|
| D | reference_util.h | 71 static std::unique_ptr<Array4D<float>> ConvArray4D(
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_dimension_numbers_test.cc | 122 auto expected_conv1 = ReferenceUtil::ConvArray4D(*input_array, *weight_array, in XLA_TEST_F()
|
| D | convolution_variants_test.cc | 828 std::unique_ptr<Array4D<float>> expected = ReferenceUtil::ConvArray4D( in XLA_TEST_F()
861 std::unique_ptr<Array4D<float>> expected = ReferenceUtil::ConvArray4D( in XLA_TEST_F()
894 std::unique_ptr<Array4D<float>> expected = ReferenceUtil::ConvArray4D( in XLA_TEST_F()
927 std::unique_ptr<Array4D<float>> expected = ReferenceUtil::ConvArray4D( in XLA_TEST_F()
961 std::unique_ptr<Array4D<float>> expected = ReferenceUtil::ConvArray4D( in XLA_TEST_F()
|