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

Fix command-line parsing for --use_nnapi

Browse Source 
PiperOrigin-RevId: 249506404
This commit is contained in:
Jared Duke 2019-05-22 13:18:11 -07:00 committed by TensorFlower Gardener
parent 9067ab24bf
commit 43ae6487dc
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -24,7 +24,7 @@ namespace {
void InitKernelTest(int* argc, char** argv) { void InitKernelTest(int* argc, char** argv) {
bool use_nnapi = false; bool use_nnapi = false;
std::vector<tflite::Flag> flags = { std::vector<tflite::Flag> flags = {
tflite::Flag::CreateFlag("--use_nnapi", &use_nnapi, "Use NNAPI"), tflite::Flag::CreateFlag("use_nnapi", &use_nnapi, "Use NNAPI"),
}; };
tflite::Flags::Parse(argc, const_cast<const char**>(argv), flags); tflite::Flags::Parse(argc, const_cast<const char**>(argv), flags);