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;
}
/**
* 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) {
this.acceleratorName = name;
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) {
this.cacheDir = cacheDir;
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) {
this.modelToken = modelToken;
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
* 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) {
this.useNnapiCpu = !enable;