experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add test case for uint32/uint64 support of tf.tile

Browse Source 
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
This commit is contained in:
Yong Tang 2020-05-11 16:48:53 +00:00
parent 764e3a790e
commit c65b6f9356
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/kernel_tests/shape_ops_test.py
View File

@ -500,6 +500,8 @@ class TileTest(test.TestCase, parameterized.TestCase):
"int16": (dtypes.int16, int), "int16": (dtypes.int16, int),
"int32": (dtypes.int32, int), "int32": (dtypes.int32, int),
"int64": (dtypes.int64, int), "int64": (dtypes.int64, int),
"uint32": (dtypes.uint32, int),
"uint64": (dtypes.uint64, int),
bytes: (dtypes.string, bytes) bytes: (dtypes.string, bytes)
} }
for dtype_np, (dtype_tf, cast) in types_to_test.items(): for dtype_np, (dtype_tf, cast) in types_to_test.items():