From 808fbb19d9dc71abe7a19f09cd5bcb90af90c29a Mon Sep 17 00:00:00 2001
From: Junrou Nishida <homuler@users.noreply.github.com>
Date: Mon, 13 Jan 2025 14:40:00 +0900
Subject: [PATCH] feat: support EDGETPU_NNAPI (#1309)

---
 .../Runtime/Scripts/Tasks/Core/BaseOptions.cs | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/Packages/com.github.homuler.mediapipe/Runtime/Scripts/Tasks/Core/BaseOptions.cs b/Packages/com.github.homuler.mediapipe/Runtime/Scripts/Tasks/Core/BaseOptions.cs
index 341b3b742..6f33530cc 100644
--- a/Packages/com.github.homuler.mediapipe/Runtime/Scripts/Tasks/Core/BaseOptions.cs
+++ b/Packages/com.github.homuler.mediapipe/Runtime/Scripts/Tasks/Core/BaseOptions.cs
@@ -10,8 +10,10 @@ public sealed class BaseOptions
   {
     public enum Delegate
     {
-      CPU,
-      GPU,
+      CPU = 0,
+      GPU = 1,
+      // Edge TPU acceleration using NNAPI delegate.
+      EDGETPU_NNAPI = 2,
     }
 
     public Delegate delegateCase { get; } = Delegate.CPU;
@@ -39,7 +41,18 @@ private Proto.Acceleration acceleration
           case Delegate.GPU:
             return new Proto.Acceleration
             {
-              Gpu = new InferenceCalculatorOptions.Types.Delegate.Types.Gpu { },
+              Gpu = new InferenceCalculatorOptions.Types.Delegate.Types.Gpu
+              {
+                UseAdvancedGpuApi = true,
+              },
+            };
+          case Delegate.EDGETPU_NNAPI:
+            return new Proto.Acceleration
+            {
+              Nnapi = new InferenceCalculatorOptions.Types.Delegate.Types.Nnapi
+              {
+                AcceleratorName = "google-edgetpu",
+              },
             };
           default:
             return null;