blob: 7b340e205621e1a56c102d0c4d12f9896be2e937 [file] [log] [blame]
# Description:
# Java Native Interface (JNI) library intended for implementing the
# TensorFlow Lite GPU delegate Java API using the TensorFlow Lite CC library.
load("//tensorflow/lite:build_def.bzl", "tflite_copts")
package(
default_visibility = ["//visibility:public"],
licenses = ["notice"], # Apache 2.0
)
cc_library(
name = "native",
srcs = ["gpu_delegate_jni.cc"],
copts = tflite_copts(),
linkopts = select({
"//tensorflow:android": [
"-lGLESv3",
"-lEGL",
],
"//conditions:default": [],
}),
tags = [
"manual",
"notap",
],
deps = [
"//tensorflow/lite/delegates/gpu:delegate",
"//tensorflow/lite/delegates/gpu/common:gpu_info",
"//tensorflow/lite/delegates/gpu/gl:egl_environment",
"//tensorflow/lite/delegates/gpu/gl:request_gpu_info",
"//tensorflow/lite/experimental/acceleration/compatibility:android_info",
"//tensorflow/lite/experimental/acceleration/compatibility:gpu_compatibility_recommender",
"//tensorflow/lite/java/jni",
"@com_google_absl//absl/status",
],
alwayslink = 1,
)