convert Maximum

chainfeng · Sep 4, 2017 · 04cd833 · 04cd833
1 parent 92eef2b
commit 04cd833
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tools/tensorflow/tensorflow2ncnn.cpp b/tools/tensorflow/tensorflow2ncnn.cpp
@@ -169,7 +169,9 @@ int main(int argc, char** argv)
             weights[output_name] = tensorflow::TensorProto();
             continue;
         }
-        else if (node.op() == "Add" || node.op() == "BiasAdd" || node.op() == "Max" || node.op() == "Mul" || node.op() == "RealDiv" || node.op() == "Sub")
+        else if (node.op() == "Add" || node.op() == "BiasAdd"
+            || node.op() == "Max" || node.op() == "Maximum" || node.op() == "Mul"
+            || node.op() == "RealDiv" || node.op() == "Sub")
         {
             // check weights
             for (int j=0; j<node.input_size(); j++)
@@ -305,7 +307,7 @@ int main(int argc, char** argv)
         {
             fprintf(pp, "%-16s", "InnerProduct");
         }
-        else if (node.op() == "Max")
+        else if (node.op() == "Max" || node.op() == "Maximum")
         {
             // check weights
             tensorflow::TensorProto tensor;
@@ -754,7 +756,7 @@ int main(int argc, char** argv)
 
             fprintf(pp, " %d %d %d", num_output, bias_term, weight_data_size);
         }
-        else if (node.op() == "Max")
+        else if (node.op() == "Max" || node.op() == "Maximum")
         {
             // check weights
             tensorflow::TensorProto tensor;