Commit bdd8cc69 authored by Dmitry Kurtaev's avatar Dmitry Kurtaev

Import wrapped Dropout subgraphs from TensorFlow

parent 6a5298a5
...@@ -363,7 +363,7 @@ void RemoveIdentityOps(tensorflow::GraphDef& net) { ...@@ -363,7 +363,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") { if (type == "Identity" || type == "Dropout") {
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);
} }
......
...@@ -138,6 +138,11 @@ TEST(Test_TensorFlow, matmul) ...@@ -138,6 +138,11 @@ TEST(Test_TensorFlow, matmul)
runTensorFlowNet("matmul"); runTensorFlowNet("matmul");
} }
TEST(Test_TensorFlow, defun)
{
runTensorFlowNet("defun_dropout");
}
TEST(Test_TensorFlow, fp16) TEST(Test_TensorFlow, fp16)
{ {
const float l1 = 1e-3; const float l1 = 1e-3;
......
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