experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Constructor with Graph for TensorFlowInferenceInterface (#12669)

Browse Source 
Added a new Constructor with `Graph` object only, it will be very useful when user is doing some custom graph loading.

Signature:
`public TensorFlowInferenceInterface(Graph g)`
This commit is contained in:
resec 2017-09-17 15:39:42 -05:00 committed by drpngx
parent 7aa94908e0
commit 74cfc64734
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tensorflow/contrib/android/java/org/tensorflow/contrib/android/TensorFlowInferenceInterface.java
View File

@ -159,6 +159,22 @@ public class TensorFlowInferenceInterface {
throw new RuntimeException("Failed to load model from the input stream", e); throw new RuntimeException("Failed to load model from the input stream", e);
} }
} }
/*
* Construct a TensorFlowInferenceInterface with provided Graph
*
* @param g The Graph to use to construct this interface.
*/
public TensorFlowInferenceInterface(Graph g) {
prepareNativeRuntime();
// modelName is redundant here, here is for
// avoiding error in initialization as modelName is marked final.
this.modelName = "";
this.g = g;
this.sess = new Session(g);
this.runner = sess.runner();
}
/** /**
* Runs inference between the previously registered input nodes (via feed*) and the requested * Runs inference between the previously registered input nodes (via feed*) and the requested