Fix command-line parsing for --use_nnapi

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

View File

@ -24,7 +24,7 @@ namespace {
void InitKernelTest(int* argc, char** argv) {
bool use_nnapi = false;
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);