diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/constant_op_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/constant_op_test.py | 15 |
1 files changed, 8 insertions, 7 deletions
diff --git a/tensorflow/python/kernel_tests/constant_op_test.py b/tensorflow/python/kernel_tests/constant_op_test.py index 6167cb9999..68817cc256 100644 --- a/tensorflow/python/kernel_tests/constant_op_test.py +++ b/tensorflow/python/kernel_tests/constant_op_test.py @@ -439,10 +439,10 @@ class ZerosLikeTest(test.TestCase): def testZerosLikeCPU(self): for dtype in [ - dtypes_lib.float32, dtypes_lib.float64, dtypes_lib.int32, - dtypes_lib.uint8, dtypes_lib.int16, dtypes_lib.int8, - dtypes_lib.complex64, dtypes_lib.complex128, dtypes_lib.int64, - dtypes_lib.string + dtypes_lib.float32, dtypes_lib.float64, dtypes_lib.int8, + dtypes_lib.uint8, dtypes_lib.int16, dtypes_lib.uint16, dtypes_lib.int32, + dtypes_lib.int64, dtypes_lib.bool, dtypes_lib.complex64, + dtypes_lib.complex128, dtypes_lib.string ]: self._compareZeros(dtype, fully_defined_shape=False, use_gpu=False) self._compareZeros(dtype, fully_defined_shape=True, use_gpu=False) @@ -573,9 +573,10 @@ class OnesLikeTest(test.TestCase): def testOnesLike(self): for dtype in [ - dtypes_lib.float32, dtypes_lib.float64, dtypes_lib.int32, - dtypes_lib.uint8, dtypes_lib.int16, dtypes_lib.int8, - dtypes_lib.complex64, dtypes_lib.complex128, dtypes_lib.int64 + dtypes_lib.float32, dtypes_lib.float64, dtypes_lib.int8, + dtypes_lib.uint8, dtypes_lib.int16, dtypes_lib.uint16, dtypes_lib.int32, + dtypes_lib.int64, dtypes_lib.bool, dtypes_lib.complex64, + dtypes_lib.complex128 ]: numpy_dtype = dtype.as_numpy_dtype with self.test_session(): |