Searched refs:graphDef (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/tools/android/inference_interface/java/org/tensorflow/contrib/android/ |
| D | TensorFlowInferenceInterface.java | 91 byte[] graphDef = new byte[is.available()]; in TensorFlowInferenceInterface()
92 final int numBytesRead = is.read(graphDef); in TensorFlowInferenceInterface()
93 if (numBytesRead != graphDef.length) { in TensorFlowInferenceInterface()
98 + graphDef.length); in TensorFlowInferenceInterface()
105 loadGraph(graphDef, g); in TensorFlowInferenceInterface()
147 byte[] graphDef = baos.toByteArray(); in TensorFlowInferenceInterface()
153 loadGraph(graphDef, g); in TensorFlowInferenceInterface()
552 private void loadGraph(byte[] graphDef, Graph g) throws IOException { in loadGraph() argument
560 g.importGraphDef(graphDef); in loadGraph()
|
| /external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/ |
| D | Graph.java | 114 public void importGraphDef(byte[] graphDef) throws IllegalArgumentException { in importGraphDef() argument
115 importGraphDef(graphDef, ""); in importGraphDef()
126 public void importGraphDef(byte[] graphDef, String prefix) throws IllegalArgumentException { in importGraphDef() argument
127 if (graphDef == null || prefix == null) { in importGraphDef()
131 importGraphDef(nativeHandle, graphDef, prefix); in importGraphDef()
455 private static native void importGraphDef(long handle, byte[] graphDef, String prefix) in importGraphDef() argument
|
| /external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/examples/ |
| D | LabelImage.java | 60 byte[] graphDef = readAllBytesOrExit(Paths.get(modelDir, "tensorflow_inception_graph.pb")); in main()
66 float[] labelProbabilities = executeInceptionGraph(graphDef, image); in main()
110 private static float[] executeInceptionGraph(byte[] graphDef, Tensor<Float> image) { in executeInceptionGraph() argument
112 g.importGraphDef(graphDef); in executeInceptionGraph()
|
| /external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/ |
| D | GraphTest.java | 35 byte[] graphDef; in graphDefRoundTrip()
39 graphDef = g.toGraphDef(); in graphDefRoundTrip()
43 g.importGraphDef(graphDef); in graphDefRoundTrip()
47 g.importGraphDef(graphDef, "BugsBunny"); in graphDefRoundTrip()
|