-
Notifications
You must be signed in to change notification settings - Fork 124
/
BUILD
58 lines (52 loc) · 2.1 KB
/
BUILD
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
load("@org_tensorflow//tensorflow/lite/java:aar_with_jni.bzl", "aar_with_jni")
load("@org_tensorflow//tensorflow/lite/core/shims:cc_library_with_tflite.bzl", "android_library_with_tflite")
load("@build_bazel_rules_android//android:rules.bzl", "android_library")
package(
default_visibility = ["//tensorflow_lite_support:users"],
licenses = ["notice"], # Apache 2.0
)
exports_files([
"AndroidManifest.xml",
])
filegroup(
name = "image_segmenter_src",
srcs = glob(["**/*.java"]),
)
# Default target that uses BuiltInOpResolver, registers all built-in OPs.
android_library_with_tflite(
name = "image_segmenter",
tflite_exports = [
"//tensorflow_lite_support/java/src/native/task/vision/segmenter:image_segmenter_native",
],
exports = [
":image_segmenter_java",
],
)
# Java-only target, need to be used together with a native target similar to
# //third_party/tensorflow_lite_support/java/src/native/task/vision/segmenter:image_segmenter_native",
# Use this target when you want to provide a MutableOpResolver with customized
# OPs and/or a subset of BuiltInOps to reduce binary size.
android_library(
name = "image_segmenter_java",
srcs = glob(["*.java"]) + [
"//tensorflow_lite_support/java/src/java/org/tensorflow/lite/task/vision/core:base_vision_api_src",
],
javacopts = ["-source 7 -target 7"],
manifest = "AndroidManifest.xml",
deps = [
"//tensorflow_lite_support/java:tensorflowlite_support_java",
"//tensorflow_lite_support/java/src/java/org/tensorflow/lite/task/core:base_task_api",
"@com_google_auto_value",
"@maven//:androidx_annotation_annotation",
"@maven//:com_google_android_odml_image",
"@org_tensorflow//tensorflow/lite/java:tensorflowlite_java",
],
)
# AAR target for OSS release.
#
# bazel build -c opt --config=monolithic --config=android_arm64 --fat_apk_cpu=x86,x86_64,arm64-v8a,armeabi-v7a \
# tensorflow_lite_support/java/src/java/org/tensorflow/lite/task/vision/segmenter:image-segmenter
aar_with_jni(
name = "image-segmenter",
android_library = ":image_segmenter",
)