diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/bitcast_op_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/bitcast_op_test.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/tensorflow/python/kernel_tests/bitcast_op_test.py b/tensorflow/python/kernel_tests/bitcast_op_test.py index a535468b05..a2c6b54273 100644 --- a/tensorflow/python/kernel_tests/bitcast_op_test.py +++ b/tensorflow/python/kernel_tests/bitcast_op_test.py @@ -76,12 +76,18 @@ class BitcastTest(test.TestCase): datatype = dtypes.int8 array_ops.bitcast(x, datatype, None) - def testQuantizeType(self): + def testQuantizedType(self): shape = [3, 4] x = np.zeros(shape, np.uint16) datatype = dtypes.quint16 self._testBitcast(x, datatype, shape) + def testUnsignedType(self): + shape = [3, 4] + x = np.zeros(shape, np.int64) + datatype = dtypes.uint64 + self._testBitcast(x, datatype, shape) + if __name__ == "__main__": test.main() |