nihuini
8 years ago
1 changed files with 5 additions and 3 deletions
Unified View
Diff Options
+ 5
- 3
tools/tensorflow/tensorflow2ncnn.cpp
View File
| @@ -169,7 +169,9 @@ int main(int argc, char** argv) | |||||
| weights[output_name] = tensorflow::TensorProto(); | weights[output_name] = tensorflow::TensorProto(); | ||||
| continue; | 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 | // check weights | ||||
| for (int j=0; j<node.input_size(); j++) | for (int j=0; j<node.input_size(); j++) | ||||
| @@ -305,7 +307,7 @@ int main(int argc, char** argv) | |||||
| { | { | ||||
| fprintf(pp, "%-16s", "InnerProduct"); | fprintf(pp, "%-16s", "InnerProduct"); | ||||
| } | } | ||||
| else if (node.op() == "Max") | |||||
| else if (node.op() == "Max" || node.op() == "Maximum") | |||||
| { | { | ||||
| // check weights | // check weights | ||||
| tensorflow::TensorProto tensor; | 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); | 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 | // check weights | ||||
| tensorflow::TensorProto tensor; | tensorflow::TensorProto tensor; | ||||