From af950819cd08a5a5e081f0a4e1afa5aba6cbe9bd Mon Sep 17 00:00:00 2001 From: nihui Date: Fri, 7 Jun 2019 15:30:46 +0800 Subject: [PATCH] convert add_n and ElementWiseSum, fix #1008 --- tools/mxnet/mxnet2ncnn.cpp | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tools/mxnet/mxnet2ncnn.cpp b/tools/mxnet/mxnet2ncnn.cpp index ce0e9ff0e..25391aefe 100644 --- a/tools/mxnet/mxnet2ncnn.cpp +++ b/tools/mxnet/mxnet2ncnn.cpp @@ -965,6 +965,10 @@ int main(int argc, char** argv) fprintf(pp, "%-16s", "TanH"); } } + else if (n.op == "add_n" || n.op == "ElementWiseSum") + { + fprintf(pp, "%-16s", "Eltwise"); + } else if (n.op == "arccos") { fprintf(pp, "%-16s", "UnaryOp"); @@ -1475,6 +1479,11 @@ int main(int argc, char** argv) // fprintf(pp, " 0=%f", 0.f); } } + else if (n.op == "add_n" || n.op == "ElementWiseSum") + { + int op_type = 1; + fprintf(pp, " 0=%d", op_type); + } else if (n.op == "arccos") { int op_type = 13;