Skip to content

O
opencv
Commit a34c02a0 authored by zuoshaobo's avatar zuoshaobo
Browse files
Options

add support for tf.add_n operation

parent 341e2fa7
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
modules/dnn/src/tensorflow/tf_importer.cpp
View file @ a34c02a0
...@@ -942,7 +942,7 @@ void TFImporter::populateNet(Net dstNet) ...@@ -942,7 +942,7 @@ void TFImporter::populateNet(Net dstNet)
if (getDataLayout(name, data_layouts) == DATA_LAYOUT_UNKNOWN) if (getDataLayout(name, data_layouts) == DATA_LAYOUT_UNKNOWN)
data_layouts[name] = DATA_LAYOUT_NHWC; data_layouts[name] = DATA_LAYOUT_NHWC;
} }
else if (type == "BiasAdd" || type == "Add" || type == "Sub") else if (type == "BiasAdd" || type == "Add" || type == "Sub" || type=="AddN")
{ {
bool haveConst = false; bool haveConst = false;
for(int ii = 0; !haveConst && ii < layer.input_size(); ++ii) for(int ii = 0; !haveConst && ii < layer.input_size(); ++ii)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment