diff options
author | 2017-03-21 14:26:26 -0800 | |
---|---|---|
committer | 2017-03-21 15:48:06 -0700 | |
commit | 61f30222eba5e3f1f51dedb3c5493f5f8eb331c8 (patch) | |
tree | 622b69c4e91a1be79bb33681d54842c08b17ee4e /tensorflow/python/layers/utils_test.py | |
parent | a6116e04dbcf397337de8a4c37b531c4f373da04 (diff) |
Add support for the NCHW data_format for 3d operations (convolution, pooling).
This brings NCHW support for 3d in sync with the corresponding 2d ops.
Change: 150811076
Diffstat (limited to 'tensorflow/python/layers/utils_test.py')
-rw-r--r-- | tensorflow/python/layers/utils_test.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/layers/utils_test.py b/tensorflow/python/layers/utils_test.py index d95ea04d42..ace8046a0b 100644 --- a/tensorflow/python/layers/utils_test.py +++ b/tensorflow/python/layers/utils_test.py @@ -25,6 +25,7 @@ from tensorflow.python.platform import test class ConvUtilsTest(test.TestCase): def testConvertDataFormat(self): + self.assertEqual(utils.convert_data_format('channels_first', 5), 'NCDHW') self.assertEqual(utils.convert_data_format('channels_first', 4), 'NCHW') self.assertEqual(utils.convert_data_format('channels_first', 3), 'NCW') self.assertEqual(utils.convert_data_format('channels_last', 4), 'NHWC') |