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

Add documentation for NnApiDelegate cache configuration API

Browse Source 
PiperOrigin-RevId: 305424094
Change-Id: I69120ba57b901daa8dc04cac455e0701d19976b6
This commit is contained in:
Stefano Galarraga 2020-04-08 00:36:13 -07:00 committed by TensorFlower Gardener
parent bba4004ed2
commit 74c4ae61bb
1 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tensorflow/lite/delegates/nnapi/java/src/main/java/org/tensorflow/lite/nnapi/NnApiDelegate.java
View File

@ -64,16 +64,34 @@ public class NnApiDelegate implements Delegate, AutoCloseable {
return this; return this;
} }
/**
* Specifies the name of the target accelerator to be used by NNAPI. If this parameter is
* specified the {@link #setUseNnapiCpu(boolean)} method won't have any effect.
*
* <p>Only effective on Android 10 (API level 29) and above.
*/
public Options setAcceleratorName(String name) { public Options setAcceleratorName(String name) {
this.acceleratorName = name; this.acceleratorName = name;
return this; return this;
} }
/**
* Configure the location to be used to store model compilation cache entries. If either
* {@code cacheDir} or {@code modelToken} parameters are unset NNAPI caching will be disabled.
*
* <p>Only effective on Android 10 (API level 29) and above.
*/
public Options setCacheDir(String cacheDir) { public Options setCacheDir(String cacheDir) {
this.cacheDir = cacheDir; this.cacheDir = cacheDir;
return this; return this;
} }
/**
* Sets the token to be used to identify this model in the model compilation cache. If either
* {@code cacheDir} or {@code modelToken} parameters are unset NNAPI caching will be disabled.
*
* <p>Only effective on Android 10 (API level 29) and above.
*/
public Options setModelToken(String modelToken) { public Options setModelToken(String modelToken) {
this.modelToken = modelToken; this.modelToken = modelToken;
return this; return this;
@ -93,7 +111,7 @@ public class NnApiDelegate implements Delegate, AutoCloseable {
* Enable or disable the NNAPI CPU Device "nnapi-reference". If unset it will use the NNAPI * Enable or disable the NNAPI CPU Device "nnapi-reference". If unset it will use the NNAPI
* default settings. * default settings.
* *
* <p>Only effective on Android 10 and above. * <p>Only effective on Android 10 (API level 29) and above.
*/ */
public Options setUseNnapiCpu(boolean enable) { public Options setUseNnapiCpu(boolean enable) {
this.useNnapiCpu = !enable; this.useNnapiCpu = !enable;