diff options
Diffstat (limited to 'src/inference_engine_tflite.cpp')
| -rw-r--r-- | src/inference_engine_tflite.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/inference_engine_tflite.cpp b/src/inference_engine_tflite.cpp index 33e7c04..2043ff5 100644 --- a/src/inference_engine_tflite.cpp +++ b/src/inference_engine_tflite.cpp @@ -116,7 +116,8 @@ namespace TFLiteImpl LOGI("Inferece targets are: [%d]", mTargetTypes); if (mTargetTypes == INFERENCE_TARGET_GPU) { - TfLiteDelegate *delegate = TfLiteGpuDelegateV2Create(nullptr); + TfLiteGpuDelegateOptionsV2 options = TfLiteGpuDelegateOptionsV2Default(); + TfLiteDelegate *delegate = TfLiteGpuDelegateV2Create(&options); if (!delegate){ LOGE("Failed to GPU delegate"); return INFERENCE_ENGINE_ERROR_INVALID_OPERATION; |