1# TensorFlow Lite Task Library - Text
2
3load(
4    "@org_tensorflow//tensorflow/lite/experimental/ios:ios.bzl",
5    "TFL_MINIMUM_OS_VERSION",
6    "tflite_ios_static_framework",
7)
8load(
9    "//tensorflow_lite_support/ios:ios.bzl",
10    "strip_c_api_include_path_prefix",
11)
12
13package(
14    default_visibility = ["//tensorflow_lite_support:users"],
15    licenses = ["notice"],  # Apache 2.0
16)
17
18strip_c_api_include_path_prefix(
19    name = "strip_c_api_include_path",
20    hdr_labels = [
21        "//tensorflow_lite_support/cc/task/text/nlclassifier:bert_nl_classifier_c_api.h",
22        "//tensorflow_lite_support/cc/task/text/nlclassifier:nl_classifier_c_api.h",
23        "//tensorflow_lite_support/cc/task/text/nlclassifier:nl_classifier_c_api_common.h",
24        "//tensorflow_lite_support/cc/task/text/qa:bert_qa_c_api.h",
25    ],
26)
27
28# This target builds a monolithic static framework for the TFLite Text API,
29# which includes the TFLite runtime in it.
30#
31# bazel build -c opt --config=ios_fat //tensorflow_lite_support/ios:TensorFlowLiteTaskTextC_framework
32tflite_ios_static_framework(
33    name = "TensorFlowLiteTaskTextC_framework",
34    hdrs = [
35        ":bert_nl_classifier_c_api.h",
36        ":bert_qa_c_api.h",
37        ":nl_classifier_c_api.h",
38        ":nl_classifier_c_api_common.h",
39    ],
40    allowlist_symbols_file = ":allowlist_TensorFlowLiteTaskText.txt",
41    bundle_name = "TensorFlowLiteTaskTextC",
42    minimum_os_version = TFL_MINIMUM_OS_VERSION,
43    deps = [
44        "//tensorflow_lite_support/cc/task/text/nlclassifier:bert_nl_classifier_c_api",
45        "//tensorflow_lite_support/cc/task/text/nlclassifier:nl_classifier_c_api",
46        "//tensorflow_lite_support/cc/task/text/qa:bert_qa_c_api",
47    ],
48)
49