diff options
author | 2017-09-26 16:51:50 -0700 | |
---|---|---|
committer | 2017-09-26 17:04:33 -0700 | |
commit | 0b853efdf0edc7a906a4d08413fa2f1d7f3d9be2 (patch) | |
tree | ee49dea912a83c0e7143ad617cc9e8dd687db394 /tensorflow/compiler/xla/reference_util_test.cc | |
parent | 1ccc394c1010a7d84b71cc193b23578d378c078b (diff) |
[XLA] Split input and output in ConvolutionDimensionNumbers
This allows for additional freedom when reasoning and transforming the input
and output of convolutions.
PiperOrigin-RevId: 170130811
Diffstat (limited to 'tensorflow/compiler/xla/reference_util_test.cc')
-rw-r--r-- | tensorflow/compiler/xla/reference_util_test.cc | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/tensorflow/compiler/xla/reference_util_test.cc b/tensorflow/compiler/xla/reference_util_test.cc index 35b5e8cd52..eb6a71242f 100644 --- a/tensorflow/compiler/xla/reference_util_test.cc +++ b/tensorflow/compiler/xla/reference_util_test.cc @@ -322,8 +322,10 @@ TEST_F(ReferenceUtilTest, ConvGeneralDimensionsWithSamePadding) { // Set the convolution dimension numbers. ConvolutionDimensionNumbers dimension_numbers; - dimension_numbers.set_batch_dimension(2); - dimension_numbers.set_feature_dimension(0); + dimension_numbers.set_input_batch_dimension(2); + dimension_numbers.set_input_feature_dimension(0); + dimension_numbers.set_output_batch_dimension(2); + dimension_numbers.set_output_feature_dimension(0); dimension_numbers.add_spatial_dimensions(1); dimension_numbers.add_spatial_dimensions(3); dimension_numbers.set_kernel_output_feature_dimension(0); @@ -374,8 +376,10 @@ TEST_F(ReferenceUtilTest, ConvGeneralDimensionsWithValidPadding) { // Set the convolution dimension numbers. ConvolutionDimensionNumbers dimension_numbers; - dimension_numbers.set_batch_dimension(2); - dimension_numbers.set_feature_dimension(0); + dimension_numbers.set_input_batch_dimension(2); + dimension_numbers.set_input_feature_dimension(0); + dimension_numbers.set_output_batch_dimension(2); + dimension_numbers.set_output_feature_dimension(0); dimension_numbers.add_spatial_dimensions(1); dimension_numbers.add_spatial_dimensions(3); |