Browse Source

!6488 add numberSplit default value to fix caffe_slice_parser bug

Merge pull request !6488 from yeyunpeng2020/master
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
f527c9c539
2 changed files with 3 additions and 1 deletions
  1. +2
    -1
      mindspore/lite/java/build_aar.sh
  2. +1
    -0
      mindspore/lite/tools/converter/parser/caffe/caffe_slice_parser.cc

+ 2
- 1
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
- 0
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;


Loading…
Cancel
Save