model-optimization/tensorflow_model_optimization/BUILD at f6858b9c639ae460c9a7600ed04a032b715a4dd8 · tensorflow/model-optimization · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
# Description:
#
#  TensorFlow Optimization is a repository for the training-time
#  portion of the Tensorflow Model Optimization Toolkit,
#  used to optimize machine learning models for deployment and execution.
#
#  https://github.com/tensorflow/model-optimization
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_binary")
# Placeholder: load py_library

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

licenses(["notice"])

exports_files(["LICENSE"])

py_library(
    name = "tensorflow_model_optimization",
    srcs = ["__init__.py"],
    strict_deps = True,
    deps = [
        # distutils dep1,
        "//tensorflow_model_optimization/python",  # buildcleaner: keep
        "//tensorflow_model_optimization/python/core:version",
        "//tensorflow_model_optimization/python/core/api",
    ],
)

py_strict_binary(
    name = "build_docs",
    srcs = ["build_docs.py"],
    deps = [
        # absl:app dep1,
        # absl/flags dep1,
        # google/protobuf:use_fast_cpp_protos dep1,  # Automatically added
        # tensorflow_docs/api_generator:generate_lib dep1,
        "//third_party/tensorflow_model_optimization",
    ],
)