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

Add test case for GitHub issue 45392.

Browse Source 
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
This commit is contained in:
Yong Tang 2020-12-14 23:12:56 +00:00
parent 0666d8bb71
commit f137ea2a23
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tensorflow/python/kernel_tests/sparse_reorder_op_test.py
View File

@ -21,6 +21,7 @@ from __future__ import print_function
import numpy as np
from tensorflow.python.framework import dtypes
from tensorflow.python.framework import errors_impl
from tensorflow.python.framework import sparse_tensor
from tensorflow.python.framework import test_util
from tensorflow.python.ops import array_ops
@ -124,6 +125,17 @@ class SparseReorderTest(test.TestCase):
x_init_value=input_val.values)
self.assertLess(err, 1e-11)
def testShapeOverflow(self):
# Test case for GitHub issue 45392
sp_input = sparse_tensor.SparseTensor(
indices=[[0, 0, 0, 0, 0, 0]], values=[0.0],
dense_shape=[4096, 4096, 4096, 4096, 4096, 4096])
self.assertAllEqual(
(4096, 4096, 4096, 4096, 4096, 4096), sp_input.get_shape())
with self.assertRaisesRegex(errors_impl.InvalidArgumentError,
"Shape would have more than"):
sp_output = sparse_ops.sparse_reorder(sp_input)
if __name__ == "__main__":
test.main()