diff options
-rw-r--r-- | packaging/inference-engine-tflite.spec | 4 | ||||
-rw-r--r-- | src/inference_engine_tflite.cpp | 3 |
2 files changed, 4 insertions, 3 deletions
diff --git a/packaging/inference-engine-tflite.spec b/packaging/inference-engine-tflite.spec index 587924a..6bea41e 100644 --- a/packaging/inference-engine-tflite.spec +++ b/packaging/inference-engine-tflite.spec @@ -1,7 +1,7 @@ Name: inference-engine-tflite Summary: Tensorflow-Lite based implementation of inference-engine-interface -Version: 0.0.1 -Release: 14 +Version: 0.0.2 +Release: 0 Group: Multimedia/Libraries License: Apache-2.0 Source0: %{name}-%{version}.tar.gz 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; |