Add a build target for generating the aar package for AudioTask library.

PiperOrigin-RevId: 370311066

Author: wangtz

tensorflow_lite_support/java/src/java/org/tensorflow/lite/task/audio/BUILD

@@ -1,3 +1,6 @@
+load("@org_tensorflow//tensorflow/lite/java:aar_with_jni.bzl", "aar_with_jni")
+load("@build_bazel_rules_android//android:rules.bzl", "android_library")
+
 package(
     default_visibility = ["//tensorflow_lite_support:users"],
     licenses = ["notice"],  # Apache 2.0
@@ -6,3 +9,30 @@ package(
 exports_files([
     "AndroidManifest.xml",
 ])
+
+android_library(
+    name = "task_library_audio",
+    srcs = [
+        "//tensorflow_lite_support/java/src/java/org/tensorflow/lite/task/audio/classifier:audio_classifier_src",
+    ],
+    # TODO(b/163039980): Use JAVACOPTS in TF. "-Xep:RemoveUnusedImports:ERROR" wierdly break the build.
+    javacopts = ["-source 7 -target 7"],
+    manifest = "AndroidManifest.xml",
+    visibility = ["//visibility:public"],
+    deps = [
+        "//tensorflow_lite_support/java:tensorflowlite_support_java",
+        "//tensorflow_lite_support/java/src/java/org/tensorflow/lite/task/core:base_task_api",
+        "//tensorflow_lite_support/java/src/native/task/audio:task_audio_native",
+        "@com_google_auto_value",
+        "@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/audio:task-library-audio
+aar_with_jni(
+    name = "task-library-audio",
+    android_library = ":task_library_audio",
+)

tensorflow_lite_support/java/src/java/org/tensorflow/lite/task/audio/classifier/BUILD

@@ -1,12 +1,16 @@
 load("@build_bazel_rules_android//android:rules.bzl", "android_library")
 load("@org_tensorflow//tensorflow/java:build_defs.bzl", "JAVACOPTS")
-load("@org_tensorflow//tensorflow/lite/java:aar_with_jni.bzl", "aar_with_jni")
 
 package(
     default_visibility = ["//tensorflow_lite_support:users"],
     licenses = ["notice"],  # Apache 2.0
 )
 
+filegroup(
+    name = "audio_classifier_src",
+    srcs = glob(["**/*.java"]),
+)
+
 # Default target that uses BuiltInOpResolver, registers all built-in OPs.
 android_library(
     name = "audio_classifier",
@@ -22,7 +26,7 @@ android_library(
 # OPs and/or a subset of BuiltInOps to reduce binary size.
 android_library(
     name = "audio_classifier_java",
-    srcs = glob(["*.java"]),
+    srcs = [":audio_classifier_src"],
     javacopts = JAVACOPTS,
     manifest = "//tensorflow_lite_support/java/src/java/org/tensorflow/lite/task/audio:AndroidManifest.xml",
     deps = [
@@ -32,12 +36,3 @@ android_library(
         "@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/audio/classifier:audio-classifier
-aar_with_jni(
-    name = "audio-classifier",
-    android_library = ":audio_classifier",
-)

tensorflow_lite_support/java/src/native/task/audio/BUILD

@@ -0,0 +1,31 @@
+load("@org_tensorflow//tensorflow/lite:build_def.bzl", "tflite_jni_binary")
+
+package(
+    default_visibility = ["//tensorflow_lite_support:users"],
+    licenses = ["notice"],  # Apache 2.0
+)
+
+cc_library(
+    name = "task_audio_native",
+    srcs = [
+        ":libtask_audio_jni.so",
+    ],
+)
+
+tflite_jni_binary(
+    name = "libtask_audio_jni.so",
+    srcs = [
+        "//tensorflow_lite_support/java/src/native/task/audio/classifier:audio_classifier_jni.cc",
+    ],
+    linkscript = "//tensorflow_lite_support/java:default_version_script.lds",
+    deps = [
+        "//tensorflow_lite_support/cc/port:statusor",
+        "//tensorflow_lite_support/cc/task/audio:audio_classifier",
+        "//tensorflow_lite_support/cc/task/audio/core:audio_buffer",
+        "//tensorflow_lite_support/cc/task/audio/proto:audio_classifier_options_cc_proto",
+        "//tensorflow_lite_support/cc/task/audio/proto:class_proto_inc",
+        "//tensorflow_lite_support/cc/task/audio/proto:classifications_proto_inc",
+        "//tensorflow_lite_support/cc/utils:jni_utils",
+        "//tensorflow_lite_support/java/jni",
+    ],
+)

tensorflow_lite_support/java/src/native/task/audio/classifier/BUILD

@@ -5,6 +5,8 @@ package(
     licenses = ["notice"],  # Apache 2.0
 )
 
+exports_files(["audio_classifier_jni.cc"])
+
 cc_library(
     name = "audio_classifier_native",
     srcs = [

tensorflow_lite_support/java/src/native/task/audio/classifier/audio_classifier_jni.cc

@@ -21,7 +21,7 @@ limitations under the License.
 #include "tensorflow_lite_support/cc/port/statusor.h"
 #include "tensorflow_lite_support/cc/task/audio/audio_classifier.h"
 #include "tensorflow_lite_support/cc/task/audio/core/audio_buffer.h"
-#include "tensorflow_lite_support/cc/task/audio/proto/audio_classifier_options.proto.h"
+#include "tensorflow_lite_support/cc/task/audio/proto/audio_classifier_options.pb.h"
 #include "tensorflow_lite_support/cc/task/audio/proto/class_proto_inc.h"
 #include "tensorflow_lite_support/cc/task/audio/proto/classifications_proto_inc.h"
 #include "tensorflow_lite_support/cc/utils/jni_utils.h"

tensorflow_lite_support/tools/ci_build/build_all.sh

@@ -34,7 +34,8 @@ 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/core:base-task-api.aar \
     //tensorflow_lite_support/java/src/java/org/tensorflow/lite/task/text:task-library-text \
-    //tensorflow_lite_support/java/src/java/org/tensorflow/lite/task/vision:task-library-vision
+    //tensorflow_lite_support/java/src/java/org/tensorflow/lite/task/vision:task-library-vision \
+    //tensorflow_lite_support/java/src/java/org/tensorflow/lite/task/audio:task-library-audio
 
 # Build Coral plugin
 bazel build --sandbox_debug --subcommands --define=darwinn_portable=1 \