diff options
| author | Inki Dae <inki.dae@samsung.com> | 2020-11-10 17:10:54 +0900 |
|---|---|---|
| committer | Inki Dae <inki.dae@samsung.com> | 2021-01-05 17:47:56 +0900 |
| commit | 8eb3c1851efb723cc4f638e929bd2681c002e5ed (patch) | |
| tree | 5f26795945663448ca007acf88d57a2f9810e108 | |
| parent | 3b98079db6287cf7320e2ff5bbb8e99499111a81 (diff) | |
| download | inference-engine-tflite-8eb3c1851efb723cc4f638e929bd2681c002e5ed.tar.gz inference-engine-tflite-8eb3c1851efb723cc4f638e929bd2681c002e5ed.tar.bz2 inference-engine-tflite-8eb3c1851efb723cc4f638e929bd2681c002e5ed.zip | |
Add GPU device capabality
Change-Id: I1c7ab126e2ef61c4c2b85e1c125f071629ec21e7
Signed-off-by: Inki Dae <inki.dae@samsung.com>
| -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 4c265fd..13b78a7 100644 --- a/src/inference_engine_tflite.cpp +++ b/src/inference_engine_tflite.cpp @@ -397,7 +397,8 @@ namespace TFLiteImpl return INFERENCE_ENGINE_ERROR_INVALID_PARAMETER; } - capacity->supported_accel_devices = INFERENCE_TARGET_CPU; + capacity->supported_accel_devices = INFERENCE_TARGET_CPU | + INFERENCE_TARGET_GPU; LOGI("LEAVE"); |