chromium/third_party/tflite_support/src/tensorflow_lite_support/python/task/vision/pybinds/BUILD

load("//tensorflow_lite_support/python/task:build_defs.bzl", "pybind_extension_may_pack_coral")

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

pybind_extension_may_pack_coral(
    name = "_pywrap_image_embedder",
    srcs = [
        "_pywrap_image_embedder.cc",
    ],
    module_name = "_pywrap_image_embedder",
    deps = [
        "//tensorflow_lite_support/cc/task/processor/proto:bounding_box_cc_proto",
        "//tensorflow_lite_support/cc/task/processor/proto:embedding_cc_proto",
        "//tensorflow_lite_support/cc/task/processor/proto:embedding_options_cc_proto",
        "//tensorflow_lite_support/cc/task/vision:image_embedder",
        "//tensorflow_lite_support/cc/task/vision/utils:image_utils",
        "//tensorflow_lite_support/python/task/core/pybinds:task_utils",
        "@pybind11",
        "@pybind11_protobuf//pybind11_protobuf:native_proto_caster",
    ],
)

pybind_extension_may_pack_coral(
    name = "_pywrap_image_classifier",
    srcs = [
        "_pywrap_image_classifier.cc",
    ],
    module_name = "_pywrap_image_classifier",
    deps = [
        "//tensorflow_lite_support/cc/task/processor/proto:bounding_box_cc_proto",
        "//tensorflow_lite_support/cc/task/processor/proto:classification_options_cc_proto",
        "//tensorflow_lite_support/cc/task/processor/proto:classifications_cc_proto",
        "//tensorflow_lite_support/cc/task/vision:image_classifier",
        "//tensorflow_lite_support/cc/task/vision/utils:image_utils",
        "//tensorflow_lite_support/python/task/core/pybinds:task_utils",
        "@pybind11",
        "@pybind11_protobuf//pybind11_protobuf:native_proto_caster",
    ],
)

pybind_extension_may_pack_coral(
    name = "_pywrap_image_segmenter",
    srcs = [
        "_pywrap_image_segmenter.cc",
    ],
    module_name = "_pywrap_image_segmenter",
    deps = [
        "//tensorflow_lite_support/cc/task/processor/proto:segmentation_options_cc_proto",
        "//tensorflow_lite_support/cc/task/vision:image_segmenter",
        "//tensorflow_lite_support/cc/task/vision/utils:image_utils",
        "//tensorflow_lite_support/python/task/core/pybinds:task_utils",
        "@pybind11",
        "@pybind11_protobuf//pybind11_protobuf:native_proto_caster",
    ],
)

pybind_extension_may_pack_coral(
    name = "_pywrap_image_searcher",
    srcs = [
        "_pywrap_image_searcher.cc",
    ],
    module_name = "_pywrap_image_searcher",
    deps = [
        "//tensorflow_lite_support/cc/task/processor/proto:bounding_box_cc_proto",
        "//tensorflow_lite_support/cc/task/vision:image_searcher",
        "//tensorflow_lite_support/cc/task/vision/utils:image_utils",
        "//tensorflow_lite_support/python/task/core/pybinds:task_utils",
        "@pybind11",
        "@pybind11_protobuf//pybind11_protobuf:native_proto_caster",
    ],
)

pybind_extension_may_pack_coral(
    name = "_pywrap_object_detector",
    srcs = [
        "_pywrap_object_detector.cc",
    ],
    module_name = "_pywrap_object_detector",
    deps = [
        "//tensorflow_lite_support/cc/task/processor/proto:detection_options_cc_proto",
        "//tensorflow_lite_support/cc/task/processor/proto:detections_cc_proto",
        "//tensorflow_lite_support/cc/task/vision:object_detector",
        "//tensorflow_lite_support/cc/task/vision/utils:image_utils",
        "//tensorflow_lite_support/python/task/core/pybinds:task_utils",
        "@pybind11",
        "@pybind11_protobuf//pybind11_protobuf:native_proto_caster",
    ],
)