diff options
| author |  Jared Duke <jdduke@google.com> | 2018-09-25 17:26:23 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-25 17:33:59 -0700 |
| commit | a7f14807417ea78aee8ea275536902f0aaa94fd4 (patch) | |
| tree | 612e511178404cb3aed788efbfc03931c13a9828 /tensorflow/contrib/lite/java/ovic | |
| parent | f97610daf89572e52912ddc5bf87576cc9e82f66 (diff) | |
Reland "Add Interpreter.Options Java API for interpreter configuration"
The original CL broke the InterpreterTest due to use of a newly
deprecated API. This has been fixed, and deprecated API usage in the
samples has also been updated.
PiperOrigin-RevId: 214532691
Diffstat (limited to 'tensorflow/contrib/lite/java/ovic')
| -rw-r--r-- | tensorflow/contrib/lite/java/ovic/src/main/java/org/tensorflow/ovic/OvicClassifier.java | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/java/ovic/src/main/java/org/tensorflow/ovic/OvicClassifier.java b/tensorflow/contrib/lite/java/ovic/src/main/java/org/tensorflow/ovic/OvicClassifier.java index 4cf51bb0fa..fd610b054f 100644 --- a/tensorflow/contrib/lite/java/ovic/src/main/java/org/tensorflow/ovic/OvicClassifier.java +++ b/tensorflow/contrib/lite/java/ovic/src/main/java/org/tensorflow/ovic/OvicClassifier.java @@ -74,7 +74,7 @@ public class OvicClassifier { } labelList = loadLabelList(labelInputStream); // OVIC uses one thread for CPU inference. - tflite = new Interpreter(model, 1); + tflite = new Interpreter(model, new Interpreter.Options().setNumThreads(1)); inputDims = TestHelper.getInputDims(tflite, 0); if (inputDims.length != 4) { throw new RuntimeException("The model's input dimensions must be 4 (BWHC)."); |