Include necessary .h files in the aar lib.

PiperOrigin-RevId: 640318226
tensorflow · Jun 4, 2024 · 864b359 · 864b359
1 parent 5a23cdf
commit 864b359
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tensorflow/lite/java/BUILD b/tensorflow/lite/java/BUILD
@@ -139,11 +139,13 @@ TFLITE_HEADERS = [
     "//tensorflow/lite/c:c_api_opaque.h",
     "//tensorflow/lite/c:c_api_types.h",
     "//tensorflow/lite/c:common.h",
+    "//tensorflow/lite/core/async/c:types.h",
     "//tensorflow/lite/core/c:c_api.h",
     "//tensorflow/lite/core/c:c_api_opaque.h",
     "//tensorflow/lite/core/c:c_api_types.h",
     "//tensorflow/lite/core/c:c_api_experimental.h",
     "//tensorflow/lite/core/c:common.h",
+    "//tensorflow/lite/core/c:operator.h",
     "//tensorflow/lite/delegates/nnapi:nnapi_delegate_c_api.h",
 ]
 # LINT.ThenChange(../c/BUILD:exported_headers)
@@ -199,8 +201,10 @@ aar_with_jni(
         "//tensorflow/lite:builtin_ops.h",
         "//tensorflow/lite/c:c_api.h",
         "//tensorflow/lite/c:c_api_types.h",
+        "//tensorflow/lite/core/async/c:types.h",
         "//tensorflow/lite/core/c:c_api.h",
         "//tensorflow/lite/core/c:c_api_types.h",
+        "//tensorflow/lite/core/c:operator.h",
         # TODO(b/175298345): move the stable parts of common.h into
         # a separate header file that contains no experimental APIs
         # (perhaps also replacing concrete struct types with opaque types