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

Merge pull request #42400 from freedomtan:make_nnapi_accelerator_name_work

Browse Source 
PiperOrigin-RevId: 327127418
Change-Id: I8e9810c3c2b0d390a1ad520ca1e3574961394e26
This commit is contained in:
TensorFlower Gardener 2020-08-17 16:56:04 -07:00
commit 33e968d542
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/lite/kernels/test_main.cc
View File

@ -31,8 +31,10 @@ void InitKernelTest(int* argc, char** argv) {
// In Android Q, the NNAPI delegate avoids delegation if the only device
// is the reference CPU. However, for testing purposes, we still want
// delegation coverage, so force use of this reference path.
delegate_providers->MutableParams()->Set<std::string>(
"nnapi_accelerator_name", "nnapi-reference");
auto* params = delegate_providers->MutableParams();
if (!params->HasValueSet<std::string>("nnapi_accelerator_name")) {
params->Set<std::string>("nnapi_accelerator_name", "nnapi-reference");
}
}
}