chromium/third_party/tflite_support/src/tensorflow_lite_support/metadata/python/tests/testdata/bert_nl_classifier/BUILD

load("//tensorflow_lite_support/tools/build_rules:http_files.bzl", "tflite_file", "tflite_model")

package(
    default_visibility = ["//tensorflow_lite_support:users"],
    licenses = ["notice"],  # Apache 2.0
)

filegroup(
    name = "test_files",
    srcs = glob([
        "*.json",
        "*.tflite",
        "*.txt",
        "*.model",
    ]) + [
        "mobilebert_vocab",
        ":30k-clean",
        ":bert_nl_classifier_no_metadata",
    ],
)

tflite_model(name = "bert_nl_classifier_no_metadata")

tflite_file(
    name = "mobilebert_vocab",
    extension = "txt",
)

tflite_file(
    name = "30k-clean",
    extension = "model",
)