diff --git a/src/layer/arm/pooling_arm.cpp b/src/layer/arm/pooling_arm.cpp index 888531dbe..ea72e6b59 100644 --- a/src/layer/arm/pooling_arm.cpp +++ b/src/layer/arm/pooling_arm.cpp @@ -42,7 +42,7 @@ Pooling_arm::Pooling_arm() support_bf16_storage = true; } -int Pooling_arm::create_pipeline(const Option& _opt) +int Pooling_arm::create_pipeline(const Option& /*opt*/) { if (adaptive_pooling) { diff --git a/tools/onnx/onnx2ncnn.cpp b/tools/onnx/onnx2ncnn.cpp index 08941ab9b..3b8edc6f3 100644 --- a/tools/onnx/onnx2ncnn.cpp +++ b/tools/onnx/onnx2ncnn.cpp @@ -2270,6 +2270,13 @@ int main(int argc, char** argv) node_reference[node.input(1)] -= 1; } } + else if (op == "adaptive_avg_pool2d" || op == "adaptive_max_pool2d") + { + if (node.input_size() >= 2) + { + node_reference[node.input(1)] -= 1; + } + } } // for (auto a: node_reference)