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/core/kernels | |
| parent | 3c6e6885f32e7638ece306dad3a5081b06137bdc (diff) | |
Fixes possible out-of-bounds access by strided slice.
PiperOrigin-RevId: 215269882
Diffstat (limited to 'tensorflow/core/kernels')
| -rw-r--r-- | tensorflow/core/kernels/strided_slice_op.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/strided_slice_op.cc b/tensorflow/core/kernels/strided_slice_op.cc index f0575de4d9..3e8a4c5b72 100644 --- a/tensorflow/core/kernels/strided_slice_op.cc +++ b/tensorflow/core/kernels/strided_slice_op.cc @@ -149,7 +149,7 @@ class StridedSliceOp : public OpKernel { // NDIM and T if (is_simple_slice && std::is_same<Device, CPUDevice>::value && input_dims == 2 && processing_shape.dims() == 2 && - final_shape.dims() == 2) { + final_shape.dims() == 2 && new_axis_mask == 0) { MemCpyFunctor<T> functor; if (functor.Copy(input, begin, end, result)) { return; |