Output the intermediate graph

1 files changed, 28 insertions, 0 deletions

diff --git a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/tensorflow/Issue9662TestCase.java b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/tensorflow/Issue9662TestCase.java
new file mode 100644
index 00000000000..be0ab4b894a
--- /dev/null
+++ b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/tensorflow/Issue9662TestCase.java
@@ -0,0 +1,28 @@
+// Copyright 2019 Oath Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package ai.vespa.rankingexpression.importer.tensorflow;
+
+import ai.vespa.rankingexpression.importer.ImportedModel;
+import ai.vespa.rankingexpression.importer.configmodelview.ImportedMlFunction;
+import org.junit.Assert;
+import org.junit.Test;
+
+import static org.junit.Assert.assertNotNull;
+
+/**
+ * @author bratseth
+ */
+public class Issue9662TestCase {
+
+    @Test
+    public void testImporting() {
+        TestableTensorFlowModel model = new TestableTensorFlowModel("test", "src/test/models/tensorflow/9662");
+        ImportedModel.Signature signature = model.get().signature("serving_default");
+        Assert.assertEquals("Should have no skipped outputs",
+                            0, model.get().signature("serving_default").skippedOutputs().size());
+
+        ImportedMlFunction output = signature.outputFunction("y", "y");
+        assertNotNull(output);
+        model.assertEqualResultSum("input", "dnn/outputs/add", 0.00001);
+    }
+
+}