From 6be1d589833628f0ddd43a9a7e452b598ddc5e9b Mon Sep 17 00:00:00 2001 From: zhaozhenlong Date: Fri, 23 Apr 2021 14:38:38 +0800 Subject: [PATCH] fix strided slice new axis mask exceed limit --- .../kernel_compiler/cpu/nnacl/infer/strided_slice_infer.c | 1 - 1 file changed, 1 deletion(-) diff --git a/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/strided_slice_infer.c b/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/strided_slice_infer.c index 83ba8d3f5b..a118b3abc5 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/strided_slice_infer.c +++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/strided_slice_infer.c @@ -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;