Merge pull request #9673 from dkurt:feature_dnn_tensorflow_defun

This commit is contained in:
Vadim Pisarevsky 2017-09-21 10:55:25 +00:00
commit 2ad27f5757
2 changed files with 6 additions and 1 deletions

View File

@ -363,7 +363,7 @@ void RemoveIdentityOps(tensorflow::GraphDef& net) {
const tensorflow::NodeDef &layer = net.node(li);
String type = layer.op();
if (type == "Identity") {
if (type == "Identity" || type == "Dropout") {
identity_ops_idx.push_back(li);
identity_ops[layer.name()] = layer.input(0);
}

View File

@ -138,6 +138,11 @@ TEST(Test_TensorFlow, matmul)
runTensorFlowNet("matmul");
}
TEST(Test_TensorFlow, defun)
{
runTensorFlowNet("defun_dropout");
}
TEST(Test_TensorFlow, fp16)
{
const float l1 = 1e-3;