Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 4 additions & 2 deletions tensorflow_model_optimization/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -5,17 +5,19 @@
# 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", "py_strict_library")
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_strict_library(
py_library(
name = "tensorflow_model_optimization",
srcs = ["__init__.py"],
strict_deps = True,
deps = [
# distutils dep1,
"//tensorflow_model_optimization/python", # buildcleaner: keep
Expand Down
5 changes: 3 additions & 2 deletions tensorflow_model_optimization/python/BUILD
Original file line number Diff line number Diff line change
@@ -1,12 +1,13 @@
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
# Placeholder: load py_library

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

licenses(["notice"])

py_strict_library(
py_library(
name = "python",
srcs = ["__init__.py"],
strict_deps = True,
deps = [
"//tensorflow_model_optimization/python/core", # buildcleaner: keep
],
Expand Down
8 changes: 5 additions & 3 deletions tensorflow_model_optimization/python/core/BUILD
Original file line number Diff line number Diff line change
@@ -1,12 +1,13 @@
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
# Placeholder: load py_library

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

licenses(["notice"])

py_strict_library(
py_library(
name = "core",
srcs = ["__init__.py"],
strict_deps = True,
deps = [
"//tensorflow_model_optimization/python/core/api", # buildcleaner: keep
"//tensorflow_model_optimization/python/core/clustering", # buildcleaner: keep
Expand All @@ -17,7 +18,8 @@ py_strict_library(
],
)

py_strict_library(
py_library(
name = "version",
srcs = ["version.py"],
strict_deps = True,
)
5 changes: 3 additions & 2 deletions tensorflow_model_optimization/python/core/api/BUILD
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
# Placeholder: load py_library

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

licenses(["notice"])

py_strict_library(
py_library(
name = "api",
srcs = [
"__init__.py",
Expand All @@ -27,6 +27,7 @@ py_strict_library(
"sparsity/__init__.py",
"sparsity/keras/__init__.py",
],
strict_deps = True,
deps = [
"//tensorflow_model_optimization/python/core/clustering/keras:cluster",
"//tensorflow_model_optimization/python/core/clustering/keras:cluster_config",
Expand Down
5 changes: 3 additions & 2 deletions tensorflow_model_optimization/python/core/clustering/BUILD
Original file line number Diff line number Diff line change
@@ -1,12 +1,13 @@
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
# Placeholder: load py_library

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

licenses(["notice"])

py_strict_library(
py_library(
name = "clustering",
srcs = ["__init__.py"],
strict_deps = True,
deps = [
"//tensorflow_model_optimization/python/core/clustering/keras", # buildcleaner: keep
],
Expand Down
30 changes: 20 additions & 10 deletions tensorflow_model_optimization/python/core/clustering/keras/BUILD
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library", "py_strict_test")
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
# Placeholder: load py_library
# Placeholder: load py_test

package(default_visibility = [
Expand All @@ -7,21 +8,23 @@ package(default_visibility = [

licenses(["notice"])

py_strict_library(
py_library(
name = "keras",
srcs = [
"__init__.py",
],
strict_deps = True,
deps = [
":cluster", # buildcleaner: keep
":clustering_callbacks", # buildcleaner: keep
"//tensorflow_model_optimization/python/core/clustering/keras/experimental", # buildcleaner: keep
],
)

py_strict_library(
py_library(
name = "cluster",
srcs = ["cluster.py"],
strict_deps = True,
visibility = ["//visibility:public"],
deps = [
":cluster_config",
Expand All @@ -32,15 +35,17 @@ py_strict_library(
],
)

py_strict_library(
py_library(
name = "cluster_config",
srcs = ["cluster_config.py"],
strict_deps = True,
visibility = ["//visibility:public"],
)

py_strict_library(
py_library(
name = "clustering_registry",
srcs = ["clustering_registry.py"],
strict_deps = True,
visibility = ["//visibility:public"],
deps = [
":clusterable_layer",
Expand All @@ -50,18 +55,20 @@ py_strict_library(
],
)

py_strict_library(
py_library(
name = "clusterable_layer",
srcs = ["clusterable_layer.py"],
strict_deps = True,
visibility = ["//visibility:public"],
deps = [
# six dep1,
],
)

py_strict_library(
py_library(
name = "clustering_centroids",
srcs = ["clustering_centroids.py"],
strict_deps = True,
visibility = ["//visibility:public"],
deps = [
":cluster_config",
Expand All @@ -72,9 +79,10 @@ py_strict_library(
],
)

py_strict_library(
py_library(
name = "cluster_wrapper",
srcs = ["cluster_wrapper.py"],
strict_deps = True,
visibility = ["//visibility:public"],
deps = [
":cluster_config",
Expand All @@ -86,9 +94,10 @@ py_strict_library(
],
)

py_strict_library(
py_library(
name = "clustering_algorithm",
srcs = ["clustering_algorithm.py"],
strict_deps = True,
visibility = ["//visibility:public"],
deps = [
# six dep1,
Expand All @@ -97,9 +106,10 @@ py_strict_library(
],
)

py_strict_library(
py_library(
name = "clustering_callbacks",
srcs = ["clustering_callbacks.py"],
strict_deps = True,
visibility = ["//visibility:public"],
deps = [
# tensorflow dep1,
Expand Down
Original file line number Diff line number Diff line change
@@ -1,24 +1,26 @@
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
# Placeholder: load py_library

package(default_visibility = [
"//tensorflow_model_optimization:__subpackages__",
])

licenses(["notice"])

py_strict_library(
py_library(
name = "experimental",
srcs = [
"__init__.py",
],
strict_deps = True,
deps = [
":cluster", # buildcleaner: keep
],
)

py_strict_library(
py_library(
name = "cluster",
srcs = ["cluster.py"],
strict_deps = True,
visibility = ["//visibility:public"],
deps = [
"//tensorflow_model_optimization/python/core/clustering/keras:cluster",
Expand Down
5 changes: 3 additions & 2 deletions tensorflow_model_optimization/python/core/internal/BUILD
Original file line number Diff line number Diff line change
@@ -1,12 +1,13 @@
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
# Placeholder: load py_library

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

licenses(["notice"])

py_strict_library(
py_library(
name = "internal",
srcs = ["__init__.py"],
strict_deps = True,
deps = [
"//tensorflow_model_optimization/python/core/internal/tensor_encoding",
],
Expand Down
Original file line number Diff line number Diff line change
@@ -1,12 +1,13 @@
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
# Placeholder: load py_library

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

licenses(["notice"])

py_strict_library(
py_library(
name = "tensor_encoding",
srcs = ["__init__.py"],
strict_deps = True,
deps = [
"//tensorflow_model_optimization/python/core/internal/tensor_encoding/core",
"//tensorflow_model_optimization/python/core/internal/tensor_encoding/encoders",
Expand Down
Original file line number Diff line number Diff line change
@@ -1,14 +1,16 @@
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library", "py_strict_test")
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
# Placeholder: load py_library

package(default_visibility = [
"//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__",
])

licenses(["notice"])

py_strict_library(
py_library(
name = "core",
srcs = ["__init__.py"],
strict_deps = True,
visibility = ["//visibility:public"],
deps = [
":core_encoder",
Expand All @@ -18,9 +20,10 @@ py_strict_library(
],
)

py_strict_library(
py_library(
name = "core_encoder",
srcs = ["core_encoder.py"],
strict_deps = True,
deps = [
":encoding_stage",
# six dep1,
Expand All @@ -43,9 +46,10 @@ py_strict_test(
],
)

py_strict_library(
py_library(
name = "encoding_stage",
srcs = ["encoding_stage.py"],
strict_deps = True,
deps = [
# six dep1,
# tensorflow dep1,
Expand All @@ -67,9 +71,10 @@ py_strict_test(
],
)

py_strict_library(
py_library(
name = "gather_encoder",
srcs = ["gather_encoder.py"],
strict_deps = True,
deps = [
":core_encoder",
# tensorflow dep1,
Expand All @@ -94,9 +99,10 @@ py_strict_test(
],
)

py_strict_library(
py_library(
name = "simple_encoder",
srcs = ["simple_encoder.py"],
strict_deps = True,
deps = [
":core_encoder",
# tensorflow dep1,
Expand Down
Original file line number Diff line number Diff line change
@@ -1,23 +1,26 @@
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library", "py_strict_test")
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
# Placeholder: load py_library

package(default_visibility = [
"//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__",
])

licenses(["notice"])

py_strict_library(
py_library(
name = "encoders",
srcs = ["__init__.py"],
strict_deps = True,
visibility = ["//visibility:public"],
deps = [
":common_encoders",
],
)

py_strict_library(
py_library(
name = "common_encoders",
srcs = ["common_encoders.py"],
strict_deps = True,
deps = [
# tensorflow dep1,
"//tensorflow_model_optimization/python/core/internal/tensor_encoding/core:core_encoder",
Expand Down
Loading
Loading