From 63247a32a7aa1c1942d54d346b611ef163eca1b1 Mon Sep 17 00:00:00 2001 From: nihuini Date: Mon, 31 Aug 2020 16:43:49 +0800 Subject: [PATCH] mlir fix keras conv2d dense bias, fix pooling type --- tools/mlir/mlir2ncnn.cpp | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tools/mlir/mlir2ncnn.cpp b/tools/mlir/mlir2ncnn.cpp index 71ebd79d8..dd0cfaf72 100644 --- a/tools/mlir/mlir2ncnn.cpp +++ b/tools/mlir/mlir2ncnn.cpp @@ -697,6 +697,8 @@ int main(int argc, char** argv) fprintf(pp, " 4=%d", -233); } + fprintf(pp, " 5=1"); // bias_term + std::vector v = get_attr_af(W); std::vector bv = get_attr_af(B); @@ -741,6 +743,7 @@ int main(int argc, char** argv) int weight_data_size = shape[0] * shape[1]; fprintf(pp, " 0=%d", num_output); + fprintf(pp, " 1=1"); // bias_term fprintf(pp, " 2=%d", weight_data_size); std::vector v = get_attr_af(W); @@ -809,6 +812,8 @@ int main(int argc, char** argv) std::vector strides = get_operation_attr_ai(operation, "strides"); std::string padding = get_operation_attr_s(operation, "padding"); + fprintf(pp, " 0=1"); // avg pool + if (ksize.size() == 4) { fprintf(pp, " 1=%d", ksize[2]); @@ -1254,6 +1259,8 @@ int main(int argc, char** argv) std::vector strides = get_operation_attr_ai(operation, "strides"); std::string padding = get_operation_attr_s(operation, "padding"); + fprintf(pp, " 0=0"); // max pool + if (ksize.size() == 4) { fprintf(pp, " 1=%d", ksize[2]);