Browse Source

convert Maximum

tags/20170919
nihuini 8 years ago
parent
commit
04cd833cae
1 changed files with 5 additions and 3 deletions
  1. +5
    -3
      tools/tensorflow/tensorflow2ncnn.cpp

+ 5
- 3
tools/tensorflow/tensorflow2ncnn.cpp View File

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


Loading…
Cancel
Save