Function InitKernelTest

tensorflow/lite/kernels/test_main.cc:24–34

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::Flags::Parse(argc, const_cast<const char**>(argv), flags);

  if (use_nnapi) {
    tflite::SingleOpModel::SetForceUseNnapi(true);
  }
}

}  // namespace