Declare delegate dependencies on Android

This commit is contained in:
Reuben Morais 2021-09-21 15:06:52 +02:00
parent fd719ac013
commit 8bd5dac837
1 changed files with 7 additions and 1 deletions

View File

@ -136,7 +136,13 @@ cc_library(
"//tensorflow/lite:model.h", "//tensorflow/lite:model.h",
"//tensorflow/lite/kernels:register.h", "//tensorflow/lite/kernels:register.h",
"//tensorflow/lite/tools/evaluation:utils.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 = [ srcs = [
"//tensorflow/lite:libtensorflowlite.so", "//tensorflow/lite:libtensorflowlite.so",
], ],