diff options
author | 2018-09-26 14:22:38 -0700 | |
---|---|---|
committer | 2018-09-26 14:27:23 -0700 | |
commit | 9280b3c8a41150022d3ea508f01959ac954c9f73 (patch) | |
tree | f5e6dab79e0fba1705524324d2591caee6d1ea2c /tensorflow/contrib/lite/java/src/test | |
parent | 72b927960625cd2920fea06e242df1ff0d220c77 (diff) |
Add an experimental Java API to allow half precision for FP32 calculation.
PiperOrigin-RevId: 214668283
Diffstat (limited to 'tensorflow/contrib/lite/java/src/test')
-rw-r--r-- | tensorflow/contrib/lite/java/src/test/java/org/tensorflow/lite/InterpreterTest.java | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/java/src/test/java/org/tensorflow/lite/InterpreterTest.java b/tensorflow/contrib/lite/java/src/test/java/org/tensorflow/lite/InterpreterTest.java index dfdd7d22b0..fdd5063156 100644 --- a/tensorflow/contrib/lite/java/src/test/java/org/tensorflow/lite/InterpreterTest.java +++ b/tensorflow/contrib/lite/java/src/test/java/org/tensorflow/lite/InterpreterTest.java @@ -323,7 +323,9 @@ public final class InterpreterTest { MappedByteBuffer mappedByteBuffer = fileChannel.map(FileChannel.MapMode.READ_ONLY, 0, fileChannel.size()); Interpreter interpreter = - new Interpreter(mappedByteBuffer, new Interpreter.Options().setUseNNAPI(true)); + new Interpreter( + mappedByteBuffer, + new Interpreter.Options().setUseNNAPI(true).setAllowFp16PrecisionForFp32(true)); float[] oneD = {1.23f, 6.54f, 7.81f}; float[][] twoD = {oneD, oneD, oneD, oneD, oneD, oneD, oneD, oneD}; float[][][] threeD = {twoD, twoD, twoD, twoD, twoD, twoD, twoD, twoD}; |