chromium/third_party/tflite_support/src/tensorflow_lite_support/custom_ops/kernel/ragged/py/BUILD

# Python wrapper used for test.

load("@org_tensorflow//tensorflow:tensorflow.bzl", "pybind_extension")

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

pybind_extension(
    name = "pywrap_tflite_registerer",
    srcs = [
        "pywrap_tflite_registerer.cc",
    ],
    additional_exported_symbols = ["TFLite_RaggedTensorToTensorRegisterer"],
    enable_stub_generation = True,
    module_name = "pywrap_tflite_registerer",
    pytype_srcs = [
        "pywrap_tflite_registerer.pyi",
    ],
    srcs_version = "PY3ONLY",
    deps = [
        "//tensorflow_lite_support/custom_ops/kernel/ragged:py_tflite_registerer",
        "@local_config_python//:python_headers",
        "@org_tensorflow//tensorflow/lite:framework",
        "@org_tensorflow//tensorflow/lite/kernels:builtin_ops",
        "@pybind11",
    ],
)