convert _minus_scalar and _mul_scalar

dacson · May 9, 2018 · d6e1b82 · d6e1b82
1 parent a341e74
commit d6e1b82
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/tools/mxnet/mxnet2ncnn.cpp b/tools/mxnet/mxnet2ncnn.cpp
@@ -745,6 +745,14 @@ int main(int argc, char** argv)
 
             fprintf(pp, "%-16s", "Input");
         }
+        else if (n.op == "_minus_scalar")
+        {
+            fprintf(pp, "%-16s", "BinaryOp");
+        }
+        else if (n.op == "_mul_scalar")
+        {
+            fprintf(pp, "%-16s", "BinaryOp");
+        }
         else if (n.op == "abs")
         {
             fprintf(pp, "%-16s", "UnaryOp");
@@ -1062,6 +1070,24 @@ int main(int argc, char** argv)
             // dummy input shape
 //             fprintf(pp, " 0 0 0");
         }
+        else if (n.op == "_minus_scalar")
+        {
+            int op_type = 1;
+            int with_scalar = 1;
+            float scalar = n.attr("scalar");
+            fprintf(pp, " 0=%d", op_type);
+            fprintf(pp, " 1=%d", with_scalar);
+            fprintf(pp, " 2=%f", scalar);
+        }
+        else if (n.op == "_mul_scalar")
+        {
+            int op_type = 2;
+            int with_scalar = 1;
+            float scalar = n.attr("scalar");
+            fprintf(pp, " 0=%d", op_type);
+            fprintf(pp, " 1=%d", with_scalar);
+            fprintf(pp, " 2=%f", scalar);
+        }
         else if (n.op == "abs")
         {
             int op_type = 0;