BUILD

package(default_visibility = ["//visibility:public"])


py_library(
    name = "neuron_py",
    srcs = [
        "__init__.py",
        "api/__init__.py",
    ],
    deps = [
        "//tensorflow/neuron/python:version_py",
        "//tensorflow/neuron/python:saved_model_py",
        "//tensorflow/neuron/python:saved_model_v2_py",
        "//tensorflow/neuron/python:graph_util_py",
        "//tensorflow/neuron/python:predictor_py",
        "//tensorflow/neuron/python:neuron_op_py",
        "//tensorflow/neuron/python:fuse_py",
        "//tensorflow/neuron/python:performance_py",
        "//tensorflow/neuron/python:unittest_py",
        "//tensorflow/neuron/python:auto_multicore_save_model_py",
        "//tensorflow/neuron/python:multicore_py",
    ],
)

filegroup(
    name = "license",
    data = [
        "LICENSE",
        "THIRD-PARTY-LICENSES.txt",
    ],
)

sh_binary(
    name = "build_pip_package",
    srcs = ["build_pip_package.sh"],
    data = [
        ":license",
        ":neuron_py",
        "//tensorflow/neuron/tf2hlo:aws_neuron_tf2hlo",
        "//tensorflow/neuron/runtime:libnrt_shared_library",
    ],
)