diff options
author | Alexandre Passos <apassos@google.com> | 2018-10-01 13:28:17 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-01 13:32:57 -0700 |
commit | 1630584951975479dee852cf6f7603fe6819fde1 (patch) | |
tree | da3595845c3eda62c72a919b613b17f556707dc5 /tensorflow/python/kernel_tests | |
parent | 3c6e6885f32e7638ece306dad3a5081b06137bdc (diff) |
Fixes possible out-of-bounds access by strided slice.
PiperOrigin-RevId: 215269882
Diffstat (limited to 'tensorflow/python/kernel_tests')
-rw-r--r-- | tensorflow/python/kernel_tests/array_ops_test.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/python/kernel_tests/array_ops_test.py b/tensorflow/python/kernel_tests/array_ops_test.py index c5547b19be..dcc594789e 100644 --- a/tensorflow/python/kernel_tests/array_ops_test.py +++ b/tensorflow/python/kernel_tests/array_ops_test.py @@ -615,6 +615,14 @@ class StridedSliceTest(test_util.TensorFlowTestCase): _ = checker[:, 0] _ = checker[:, :, 0] + def testBothNewAxisAndShrink(self): + with self.test_session(use_gpu=True): + ones = array_ops.placeholder(shape=[2, 2], dtype=dtypes.int16) + self.assertAllEqual( + ones[array_ops.newaxis, :, 0].eval( + feed_dict={ones: [[1, 1], [1, 1]]}), + [[1, 1]]) + def testTensorIndexing(self): with self.test_session(use_gpu=True): raw = [[[[[1, 2, 4, 5], [5, 6, 7, 8], [9, 10, 11, 12]]], |