diff --git a/src/operators/feed_op.cpp b/src/operators/feed_op.cpp index d03f5c3ecb8afd47e2c3b82fdc8935f27b3a45f1..ffd253073a82d3ff7a0f1e4fca268f8ec24a9145 100644 --- a/src/operators/feed_op.cpp +++ b/src/operators/feed_op.cpp @@ -24,6 +24,11 @@ void FeedOp::InferShape() const { int col = this->param_.Col(); auto input_dims = this->param_.InputX()->at(col).dims(); this->param_.Out()->Resize(input_dims); + if (input_dims.size() == 4 || input_dims.size() == 2) { + this->param_.Out()->Resize(input_dims); + } else { + this->param_.Out()->Resize(out_dims); + } } } // namespace operators diff --git a/tools/build.sh b/tools/build.sh index 232d81fb2fa30add4420296513297ed75e709425..f7ae755a818fd605de70739f088a4b49530e22a8 100755 --- a/tools/build.sh +++ b/tools/build.sh @@ -59,7 +59,7 @@ build_for_android() { MODE="Release" - ANDROID_PLATFORM_VERSION="android-22" + ANDROID_PLATFORM_VERSION="android-19" TOOLCHAIN_FILE="./tools/android-cmake/android.toolchain.cmake" ANDROID_ARM_MODE="arm"