Skip to content

Commit

Permalink
停用 nnapi 的有关设置
Browse files Browse the repository at this point in the history
  • Loading branch information
Steve-Mr committed Apr 20, 2024
1 parent 7429c0d commit 5db2e73
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 7 deletions.
2 changes: 1 addition & 1 deletion app/build.gradle
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ android {
minSdkVersion 29
targetSdkVersion 34
versionCode 4
versionName "3.0-alpha-240420"
versionName "3.0-alpha-240420-02"
testInstrumentationRunner "androidx.test.runner.AndroidJUnitRunner"
}
buildFeatures {
Expand Down
12 changes: 6 additions & 6 deletions app/src/main/java/com/maary/shareas/WallpaperViewModel.kt
Original file line number Diff line number Diff line change
Expand Up @@ -652,12 +652,12 @@ class WallpaperViewModel : ViewModel() {

val sessionOptions: OrtSession.SessionOptions = OrtSession.SessionOptions()
sessionOptions.registerCustomOpLibrary(OrtxPackage.getLibraryPath())
sessionOptions.addNnapi(
EnumSet.of(
NNAPIFlags.USE_FP16,
NNAPIFlags.CPU_DISABLED
)
)
// sessionOptions.addNnapi(
// EnumSet.of(
// NNAPIFlags.USE_FP16,
// NNAPIFlags.CPU_DISABLED
// )
// )
ortSession = ortEnv.createSession(readModel(context, model), sessionOptions)

// 1. 输入是一个 bimap
Expand Down

0 comments on commit 5db2e73

Please sign in to comment.