Browse Source

convert add_n and ElementWiseSum, fix #1008

tags/20190611
nihui 7 years ago
parent
commit
af950819cd
1 changed files with 9 additions and 0 deletions
  1. +9
    -0
      tools/mxnet/mxnet2ncnn.cpp

+ 9
- 0
tools/mxnet/mxnet2ncnn.cpp View File

@@ -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;


Loading…
Cancel
Save