Declare delegate dependencies on Android

2021-09-21 15:06:52 +02:00 · 2021-09-21 15:06:52 +02:00 · 8bd5dac837
parent fd719ac013
commit 8bd5dac837
1 changed files with 7 additions and 1 deletions
--- a/native_client/BUILD
+++ b/native_client/BUILD
@ -136,7 +136,13 @@ cc_library(
        "//tensorflow/lite:model.h",
        "//tensorflow/lite/kernels:register.h",
        "//tensorflow/lite/tools/evaluation:utils.h",
-    ],
+    ] + select({
+        "//tensorflow:android": [
+            "//tensorflow/lite/delegates/gpu:delegate.h",
+            "//tensorflow/lite/delegates/hexagon:hexagon_delegate.h",
+        ],
+        "//conditions:default": [],
+    }),
    srcs = [
        "//tensorflow/lite:libtensorflowlite.so",
    ],