From 8bd5dac837584bbd6cffadbe8ac573801219559f Mon Sep 17 00:00:00 2001 From: Reuben Morais Date: Tue, 21 Sep 2021 15:06:52 +0200 Subject: [PATCH] Declare delegate dependencies on Android --- native_client/BUILD | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/native_client/BUILD b/native_client/BUILD index d0a5f5d9..03ae756e 100644 --- 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", ],