Browse Source

!15576 fix newAxisMask making ndim incorrect

From: @zhaozhenlong
Reviewed-by: @zhanghaibo5,@zhang_xue_tong
Signed-off-by: @zhang_xue_tong
pull/15576/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
e26d3f5a21
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) { size_t *out_shape_size) {
for (size_t i = 0; i < transfer_buffer->new_axis_mask_size_; i++) { for (size_t i = 0; i < transfer_buffer->new_axis_mask_size_; i++) {
if (transfer_buffer->new_axis_mask_[i]) { if (transfer_buffer->new_axis_mask_[i]) {
transfer_buffer->ndim_ += 1;
ShapeInsert(in_shape, out_shape_size, i, 1); ShapeInsert(in_shape, out_shape_size, i, 1);
transfer_buffer->begins_[i] = 0; transfer_buffer->begins_[i] = 0;
transfer_buffer->ends_[i] = 1; transfer_buffer->ends_[i] = 1;


Loading…
Cancel
Save