root
/
MNN
mirror of https://github.com/alibaba/MNN.git
1
0
Fork 0
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix bug in shaping slice

This commit is contained in:
Turing311 2020-04-27 09:46:13 +08:00 committed by GitHub
parent ab7677771c
commit b5aed1d64a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
source/shape/ShapeSlice.cpp
View File

@ -40,6 +40,8 @@ class SliceComputer : public SizeComputer {
// Compute Last
auto& output = outputs[outputs.size() - 1]->buffer();
output.dimensions = input.dimensions;
output.type = input.type;
::memcpy(output.dim, input.dim, input.dimensions * sizeof(halide_dimension_t));
output.dim[axis].extent = input.dim[axis].extent - previous;