diff options
Diffstat (limited to 'tensorflow/compiler/tests/nary_ops_test.py')
| -rw-r--r-- | tensorflow/compiler/tests/nary_ops_test.py | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/tensorflow/compiler/tests/nary_ops_test.py b/tensorflow/compiler/tests/nary_ops_test.py index a1f1e67a9f..2660e1d572 100644 --- a/tensorflow/compiler/tests/nary_ops_test.py +++ b/tensorflow/compiler/tests/nary_ops_test.py @@ -116,12 +116,14 @@ class NAryOpsTest(XLATestCase): np.array([1, 1], dtype=np.int32)], expected=np.array([[], []], dtype=np.float32)) - self._testNAry(lambda x: array_ops.strided_slice(*x), - [np.array([[], [], []], dtype=np.float32), - np.array([1, 0], dtype=np.int64), - np.array([3, 0], dtype=np.int64), - np.array([1, 1], dtype=np.int64)], - expected=np.array([[], []], dtype=np.float32)) + if np.int64 in self.int_types: + self._testNAry( + lambda x: array_ops.strided_slice(*x), [ + np.array([[], [], []], dtype=np.float32), np.array( + [1, 0], dtype=np.int64), np.array([3, 0], dtype=np.int64), + np.array([1, 1], dtype=np.int64) + ], + expected=np.array([[], []], dtype=np.float32)) self._testNAry(lambda x: array_ops.strided_slice(*x), [np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]], |