Browse Source

fix strided slice new axis mask exceed limit

pull/15576/head
zhaozhenlong 5 years ago
parent
commit
6be1d58983
1 changed files with 0 additions and 1 deletions
  1. +0
    -1
      mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/strided_slice_infer.c

+ 0
- 1
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/strided_slice_infer.c View File

@@ -177,7 +177,6 @@ void ApplyNewAxisMask(StridedSliceTransferBuffer *transfer_buffer, StridedSliceP
size_t *out_shape_size) {
for (size_t i = 0; i < transfer_buffer->new_axis_mask_size_; i++) {
if (transfer_buffer->new_axis_mask_[i]) {
transfer_buffer->ndim_ += 1;
ShapeInsert(in_shape, out_shape_size, i, 1);
transfer_buffer->begins_[i] = 0;
transfer_buffer->ends_[i] = 1;


Loading…
Cancel
Save