summaryrefslogtreecommitdiffstats
path: root/searchlib
diff options
context:
space:
mode:
authorJon Bratseth <bratseth@oath.com>2018-08-13 09:52:08 +0200
committerGitHub <noreply@github.com>2018-08-13 09:52:08 +0200
commit8420c115a34990d592a287132d5753d8956c4770 (patch)
treed6df8b74bba2fcde95bd08db1b6982d694f3e6a4 /searchlib
parent5deabf6d73002692aaaf9e4b4221235f413dc831 (diff)
parent7ccae75e7f026df580debe58156aa0ee4585714e (diff)
Merge pull request #6550 from vespa-engine/bratseth/generate-rank-profiles-for-all-models
Bratseth/generate rank profiles for all models
Diffstat (limited to 'searchlib')
-rw-r--r--searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/TensorFlowImporter.java1
1 files changed, 1 insertions, 0 deletions
diff --git a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/TensorFlowImporter.java b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/TensorFlowImporter.java
index ff584559a83..303ba228fa6 100644
--- a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/TensorFlowImporter.java
+++ b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/TensorFlowImporter.java
@@ -23,6 +23,7 @@ public class TensorFlowImporter extends ModelImporter {
* @param modelName the name of the model to import, consisting of characters in [A-Za-z0-9_]
* @param modelDir the directory containing the TensorFlow model files to import
*/
+ @Override
public ImportedModel importModel(String modelName, String modelDir) {
try (SavedModelBundle model = SavedModelBundle.load(modelDir, "serve")) {
return importModel(modelName, model);