AddV2 from TensorFlow

2019-12-30 20:06:58 +03:00
parent b4759d7272
commit 76cfa65d55
2 changed files with 7 additions and 3 deletions
@@ -996,7 +996,7 @@ void TFImporter::populateNet(Net dstNet)
            if (getDataLayout(name, data_layouts) == DATA_LAYOUT_UNKNOWN)
                data_layouts[name] = DATA_LAYOUT_NHWC;
        }
-        else if (type == "BiasAdd" || type == "Add" || type == "Sub" || type=="AddN")
+        else if (type == "BiasAdd" || type == "Add" || type == "AddV2" || type == "Sub" || type=="AddN")
        {
            bool haveConst = false;
            for(int ii = 0; !haveConst && ii < layer.input_size(); ++ii)