add load TF saved model as TFNet inference model (intel-analytics#1745)

dding3 · Nov 11, 2019 · c2fed7a · c2fed7a
1 parent d9c8bcf
commit c2fed7a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/spark/dl/src/main/java/com/intel/analytics/bigdl/dllib/inference/AbstractInferenceModel.java b/spark/dl/src/main/java/com/intel/analytics/bigdl/dllib/inference/AbstractInferenceModel.java
@@ -59,6 +59,14 @@ public void loadTF(String modelPath, int intraOpParallelismThreads, int interOpP
     doLoadTF(modelPath, intraOpParallelismThreads, interOpParallelismThreads, usePerSessionThreads);
   }
 
+  public void loadTF(String modelPath, String[] inputs, String[] outputs) {
+    doLoadTF(modelPath, inputs, outputs);
+  }
+
+  public void loadTF(String modelPath, String[] inputs, String[] outputs, int intraOpParallelismThreads, int interOpParallelismThreads, boolean usePerSessionThreads) {
+    doLoadTF(modelPath, inputs, outputs, intraOpParallelismThreads, interOpParallelismThreads, usePerSessionThreads);
+  }
+
   public void loadTF(String modelPath, String objectDetectionModelType) {
     doLoadTF(modelPath, objectDetectionModelType);
   }