TensorFlow-Models/inception/slim/BUILD

# Description:
#   Contains the operations and nets for building TensorFlow-Slim models.

package(default_visibility = ["//inception:internal"])

licenses(["notice"])  # Apache 2.0

exports_files(["LICENSE"])

py_library(
    name = "scopes",
    srcs = ["scopes.py"],
    deps = [
        "@tf//tensorflow:tensorflow_py",
    ],
)

py_test(
    name = "scopes_test",
    size = "small",
    srcs = ["scopes_test.py"],
    deps = [
        ":scopes",
    ],
)

py_library(
    name = "variables",
    srcs = ["variables.py"],
    deps = [
        "@tf//tensorflow:tensorflow_py",
        ":scopes",
    ],
)

py_test(
    name = "variables_test",
    size = "small",
    srcs = ["variables_test.py"],
    deps = [
        ":variables",
    ],
)

py_library(
    name = "losses",
    srcs = ["losses.py"],
    deps = [
        "@tf//tensorflow:tensorflow_py",
    ],
)

py_test(
    name = "losses_test",
    size = "small",
    srcs = ["losses_test.py"],
    deps = [
        ":losses",
    ],
)

py_library(
    name = "ops",
    srcs = ["ops.py"],
    deps = [
        "@tf//tensorflow:tensorflow_py",
        ":losses",
        ":scopes",
        ":variables",
    ],
)

py_test(
    name = "ops_test",
    size = "small",
    srcs = ["ops_test.py"],
    deps = [
        ":ops",
        ":variables",
    ],
)

py_library(
    name = "inception",
    srcs = ["inception_model.py"],
    deps = [
        "@tf//tensorflow:tensorflow_py",
        ":ops",
        ":scopes",
    ],
)

py_test(
    name = "inception_test",
    size = "medium",
    srcs = ["inception_test.py"],
    deps = [
        ":inception",
    ],
)

py_library(
    name = "slim",
    srcs = ["slim.py"],
    deps = [
        ":inception",
        ":losses",
        ":ops",
        ":scopes",
        ":variables",
    ],
)