Browse Source

modified: ge/host_kernels/strided_slice_kernel.cc

pull/383/head
zhaoxinxin 4 years ago
parent
commit
eb0f6723e0
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/host_kernels/strided_slice_kernel.cc

+ 1
- 1
ge/host_kernels/strided_slice_kernel.cc View File

@@ -309,7 +309,7 @@ void StridedSliceKernel::ExpandStrideWithEllipsisMask(const size_t x_dims_num,
for (size_t j = 1; j < (x_dims_num - orig_begin_vec.size() + 1); ++j) { for (size_t j = 1; j < (x_dims_num - orig_begin_vec.size() + 1); ++j) {
orig_begin_vec.insert((orig_begin_vec.begin() + ellipsis_dim + j), 0); orig_begin_vec.insert((orig_begin_vec.begin() + ellipsis_dim + j), 0);
orig_end_vec.insert((orig_end_vec.begin() + ellipsis_dim + j), x_dims.at(ellipsis_dim +j)); orig_end_vec.insert((orig_end_vec.begin() + ellipsis_dim + j), x_dims.at(ellipsis_dim +j));
orig_stride_vec.insert((orig_begin_vec.begin() + ellipsis_dim + j), 1);
orig_stride_vec.insert((orig_stride_vec.begin() + ellipsis_dim + j), 1);
} }
} }
} }


Loading…
Cancel
Save