diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/array_ops_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/array_ops_test.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tensorflow/python/kernel_tests/array_ops_test.py b/tensorflow/python/kernel_tests/array_ops_test.py index 2cfbb63de8..42e63d8b81 100644 --- a/tensorflow/python/kernel_tests/array_ops_test.py +++ b/tensorflow/python/kernel_tests/array_ops_test.py @@ -285,7 +285,8 @@ class ReverseV2Test(test_util.TensorFlowTestCase): def testReverse1DimAuto(self): for dtype in [ - np.uint8, np.int8, np.int32, np.int64, np.bool, np.float16, np.float32, + np.uint8, np.int8, np.uint16, np.int16, np.int32, np.int64, + np.bool, np.float16, np.float32, np.float64, np.complex64, np.complex128, np.array(b"").dtype.type ]: @@ -293,7 +294,8 @@ class ReverseV2Test(test_util.TensorFlowTestCase): def testReverse2DimAuto(self): for dtype in [ - np.uint8, np.int8, np.int32, np.int64, np.bool, np.float16, np.float32, + np.uint8, np.int8, np.uint16, np.int16, np.int32, np.int64, + np.bool, np.float16, np.float32, np.float64, np.complex64, np.complex128, np.array(b"").dtype.type ]: |