blob: 3ea6bbf9ecf71031a6ac74f80c83843f6c1fe5b7 [file] [log] [blame]
diff --git a/tensorflow/lite/BUILD b/tensorflow/lite/BUILD
index c0b1cd16..2406b723 100644
--- a/tensorflow/lite/BUILD
+++ b/tensorflow/lite/BUILD
@@ -1045,6 +1045,8 @@ tflite_cc_shared_object(
"//conditions:default": [
"-Wl,-z,defs",
"-Wl,--version-script,$(location //tensorflow/lite:tflite_version_script.lds)",
+ "-lEGL",
+ "-lGLESv2",
],
}),
per_os_targets = True,
@@ -1053,6 +1055,7 @@ tflite_cc_shared_object(
":tflite_exported_symbols.lds",
":tflite_version_script.lds",
"//tensorflow/lite/kernels:builtin_ops_all_linked",
+ "//tensorflow/lite/delegates/gpu:delegate",
],
)
diff --git a/tensorflow/lite/examples/label_image/BUILD b/tensorflow/lite/examples/label_image/BUILD
index b6d0f108..b3147244 100644
--- a/tensorflow/lite/examples/label_image/BUILD
+++ b/tensorflow/lite/examples/label_image/BUILD
@@ -25,7 +25,10 @@ cc_binary(
"-lm", # some builtin ops, e.g., tanh, need -lm
"-Wl,-rpath=/data/local/tmp", # for hexagon delegate
],
- "//conditions:default": [],
+ "//conditions:default": [
+ "-lEGL",
+ "-lGLESv2",
+ ],
}),
deps = [
":bitmap_helpers",
@@ -68,7 +71,9 @@ cc_library(
"//tensorflow/lite/delegates/gpu:delegate",
"//tensorflow/lite/delegates/hexagon:hexagon_delegate",
],
- "//conditions:default": [],
+ "//conditions:default": [
+ "//tensorflow/lite/delegates/gpu:delegate",
+ ],
}),
)
diff --git a/tensorflow/lite/tools/benchmark/BUILD b/tensorflow/lite/tools/benchmark/BUILD
index a5813112..7f280d17 100644
--- a/tensorflow/lite/tools/benchmark/BUILD
+++ b/tensorflow/lite/tools/benchmark/BUILD
@@ -36,7 +36,10 @@ cc_binary(
"-lm", # some builtin ops, e.g., tanh, need -lm
"-Wl,--rpath=/data/local/tmp/", # Hexagon delegate libraries should be in /data/local/tmp
],
- "//conditions:default": [],
+ "//conditions:default": [
+ "-lEGL",
+ "-lGLESv2",
+ ],
}),
tags = ["builder_default_android_arm64"],
deps = [
@@ -191,7 +194,9 @@ cc_library(
"//tensorflow:android": [
"//tensorflow/lite/delegates/gpu:delegate",
],
- "//conditions:default": [],
+ "//conditions:default": [
+ "//tensorflow/lite/delegates/gpu:delegate",
+ ],
}),
)
diff --git a/tensorflow/lite/tools/delegates/BUILD b/tensorflow/lite/tools/delegates/BUILD
index b0ddff8c..e1e56f7a 100644
--- a/tensorflow/lite/tools/delegates/BUILD
+++ b/tensorflow/lite/tools/delegates/BUILD
@@ -78,12 +78,8 @@ cc_library(
deps = [
":delegate_provider_hdr",
"//tensorflow/lite/tools/evaluation:utils",
+ "//tensorflow/lite/delegates/gpu:delegate",
] + select({
- "//tensorflow/lite/delegates/gpu:supports_gpu_delegate": [
- "//tensorflow/lite/delegates/gpu:delegate",
- ],
- "//conditions:default": [],
- }) + select({
"//tensorflow:ios": [
"//tensorflow/lite/delegates/gpu:metal_delegate",
],
@@ -117,6 +113,7 @@ cc_library(
deps = [
":delegate_provider_hdr",
"//tensorflow/lite/tools/evaluation:utils",
+ "//tensorflow/lite/delegates/gpu:delegate",
] + select({
"//tensorflow:android_arm": [
"//tensorflow/lite/delegates/hexagon:hexagon_delegate",
@@ -144,6 +141,7 @@ cc_library(
deps = [
":delegate_provider_hdr",
"//tensorflow/lite/tools/evaluation:utils",
+ "//tensorflow/lite/delegates/gpu:delegate",
] + select({
"//tensorflow:ios": [
"//tensorflow/lite/delegates/coreml:coreml_delegate",
@@ -165,6 +163,7 @@ cc_library(
deps = [
":delegate_provider_hdr",
"//tensorflow/lite/tools/evaluation:utils",
+ "//tensorflow/lite/delegates/gpu:delegate",
],
alwayslink = 1,
)
diff --git a/tensorflow/lite/tools/evaluation/BUILD b/tensorflow/lite/tools/evaluation/BUILD
index ba745c6a..a1b7a49e 100644
--- a/tensorflow/lite/tools/evaluation/BUILD
+++ b/tensorflow/lite/tools/evaluation/BUILD
@@ -43,12 +43,8 @@ cc_library(
"//tensorflow/lite/c:common",
"//tensorflow/lite/delegates/nnapi:nnapi_delegate",
"//tensorflow/lite/delegates/xnnpack:xnnpack_delegate",
+ "//tensorflow/lite/delegates/gpu:delegate",
] + select({
- "//tensorflow/lite/delegates/gpu:supports_gpu_delegate": [
- "//tensorflow/lite/delegates/gpu:delegate",
- ],
- "//conditions:default": [],
- }) + select({
"//tensorflow:android_arm": [
"//tensorflow/lite/delegates/hexagon:hexagon_delegate",
],