sbjkx
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!6488 add numberSplit default value to fix caffe_slice_parser bug 

Merge pull request !6488 from yeyunpeng2020/master
This commit is contained in:
mindspore-ci-bot 2020-09-18 19:45:19 +08:00 committed by Gitee
parent a845d36c26 c17c549a40
commit f527c9c539
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/lite/java/build_aar.sh
View File

@ -28,7 +28,8 @@ build_mslite_arm64() {
mkdir -p ${BASE_PATH}/java/app/libs/arm64-v8a/
rm -rf ${BASE_PATH}/java/app/libs/arm64-v8a/*
cp ${TOP_PATH}/output/mindspore-lite-${VERSION_STR}-runtime-arm64-cpu/lib/libmindspore-lite.so ${BASE_PATH}/java/app/libs/arm64-v8a/
cp ${TOP_PATH}/output/mindspore-lite-${VERSION_STR}-runtime-arm64-cpu/lib/liboptimize.so ${BASE_PATH}/java/app/libs/arm64-v8a/
cp ${TOP_PATH}/output/mindspore-lite-${VERSION_STR}-runtime-arm64-cpu/lib/libmindspore-lite-fp16.so ${BASE_PATH}/java/app/libs/arm64-v8a/
cp ${TOP_PATH}/output/mindspore-lite-${VERSION_STR}-runtime-arm64-cpu/lib/libmindspore-lite-optimize.so ${BASE_PATH}/java/app/libs/arm64-v8a/
cp ${ANDROID_NDK}/sources/cxx-stl/llvm-libc++/libs/arm64-v8a/libc++_shared.so ${BASE_PATH}/java/app/libs/arm64-v8a/
}

1
mindspore/lite/tools/converter/parser/caffe/caffe_slice_parser.cc
View File

@ -41,6 +41,7 @@ STATUS CaffeSliceParser::Parse(const caffe::LayerParameter &proto, const caffe::
const caffe::SliceParameter &slice_param = proto.slice_param();
attr->numberSplit = 2;
if (!slice_param.slice_point().empty()) {
attr->numberSplit = slice_param.slice_point_size() + 1;
std::vector<int32_t> size_splits;