mirror of
https://github.com/opencv/opencv.git
synced 2026-07-31 08:13:04 +04:00
Merge remote-tracking branch 'upstream/3.4' into merge-3.4
This commit is contained in:
@@ -449,7 +449,7 @@ public:
|
||||
|
||||
InferenceEngine::Builder::Layer l = ieLayer;
|
||||
addConstantData("weights", wrapToInfEngineBlob(blobs[0], {(size_t)blobs[0].size[0], (size_t)blobs[0].size[1], 1, 1}, InferenceEngine::Layout::OIHW), l);
|
||||
if (blobs.size() > 1)
|
||||
if (bias)
|
||||
addConstantData("biases", wrapToInfEngineBlob(blobs[1], {(size_t)outNum}, InferenceEngine::Layout::C), l);
|
||||
|
||||
return Ptr<BackendNode>(new InfEngineBackendNode(l));
|
||||
|
||||
@@ -787,7 +787,7 @@ void RemoveIdentityOps(tensorflow::GraphDef& net)
|
||||
const tensorflow::NodeDef &layer = net.node(li);
|
||||
String type = layer.op();
|
||||
|
||||
if (type == "Identity" || type == "Dropout") {
|
||||
if (type == "Identity" || type == "Dropout" || type == "PlaceholderWithDefault") {
|
||||
identity_ops_idx.push_back(li);
|
||||
identity_ops[layer.name()] = layer.input(0);
|
||||
}
|
||||
@@ -1031,7 +1031,7 @@ void removePhaseSwitches(tensorflow::GraphDef& net)
|
||||
}
|
||||
}
|
||||
nodesToRemove.push_back(i);
|
||||
if (node.op() == "Merge")
|
||||
if (node.op() == "Merge" || node.op() == "Switch")
|
||||
mergeOpSubgraphNodes.push(i);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user