Commit 4a33014e authored by Alexander Alekhin's avatar Alexander Alekhin

Merge pull request #15391 from dkurt:fix_15141

parents 114674c3 57676cf6
...@@ -787,7 +787,7 @@ void RemoveIdentityOps(tensorflow::GraphDef& net) ...@@ -787,7 +787,7 @@ void RemoveIdentityOps(tensorflow::GraphDef& net)
const tensorflow::NodeDef &layer = net.node(li); const tensorflow::NodeDef &layer = net.node(li);
String type = layer.op(); String type = layer.op();
if (type == "Identity" || type == "Dropout") { if (type == "Identity" || type == "Dropout" || type == "PlaceholderWithDefault") {
identity_ops_idx.push_back(li); identity_ops_idx.push_back(li);
identity_ops[layer.name()] = layer.input(0); identity_ops[layer.name()] = layer.input(0);
} }
...@@ -1031,7 +1031,7 @@ void removePhaseSwitches(tensorflow::GraphDef& net) ...@@ -1031,7 +1031,7 @@ void removePhaseSwitches(tensorflow::GraphDef& net)
} }
} }
nodesToRemove.push_back(i); nodesToRemove.push_back(i);
if (node.op() == "Merge") if (node.op() == "Merge" || node.op() == "Switch")
mergeOpSubgraphNodes.push(i); mergeOpSubgraphNodes.push(i);
} }
} }
......
...@@ -729,6 +729,7 @@ TEST_P(Test_TensorFlow_layers, subpixel) ...@@ -729,6 +729,7 @@ TEST_P(Test_TensorFlow_layers, subpixel)
TEST_P(Test_TensorFlow_layers, keras_mobilenet_head) TEST_P(Test_TensorFlow_layers, keras_mobilenet_head)
{ {
runTensorFlowNet("keras_mobilenet_head"); runTensorFlowNet("keras_mobilenet_head");
runTensorFlowNet("keras_learning_phase");
} }
TEST_P(Test_TensorFlow_layers, resize_bilinear) TEST_P(Test_TensorFlow_layers, resize_bilinear)
......
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