Skip to content

Commit a6a51ae

Browse files
Automated Code Change
PiperOrigin-RevId: 858775602
1 parent b0b0ad5 commit a6a51ae

File tree

21 files changed

+157
-84
lines changed

21 files changed

+157
-84
lines changed

tensorflow_model_optimization/BUILD

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,17 +5,19 @@
55
# used to optimize machine learning models for deployment and execution.
66
#
77
# https://github.com/tensorflow/model-optimization
8-
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_binary", "py_strict_library")
8+
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_binary")
9+
# Placeholder: load py_library
910

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

1213
licenses(["notice"])
1314

1415
exports_files(["LICENSE"])
1516

16-
py_strict_library(
17+
py_library(
1718
name = "tensorflow_model_optimization",
1819
srcs = ["__init__.py"],
20+
strict_deps = True,
1921
deps = [
2022
# distutils dep1,
2123
"//tensorflow_model_optimization/python", # buildcleaner: keep

tensorflow_model_optimization/python/BUILD

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,13 @@
1-
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
1+
# Placeholder: load py_library
22

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

55
licenses(["notice"])
66

7-
py_strict_library(
7+
py_library(
88
name = "python",
99
srcs = ["__init__.py"],
10+
strict_deps = True,
1011
deps = [
1112
"//tensorflow_model_optimization/python/core", # buildcleaner: keep
1213
],
Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,13 @@
1-
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
1+
# Placeholder: load py_library
22

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

55
licenses(["notice"])
66

7-
py_strict_library(
7+
py_library(
88
name = "core",
99
srcs = ["__init__.py"],
10+
strict_deps = True,
1011
deps = [
1112
"//tensorflow_model_optimization/python/core/api", # buildcleaner: keep
1213
"//tensorflow_model_optimization/python/core/clustering", # buildcleaner: keep
@@ -17,7 +18,8 @@ py_strict_library(
1718
],
1819
)
1920

20-
py_strict_library(
21+
py_library(
2122
name = "version",
2223
srcs = ["version.py"],
24+
strict_deps = True,
2325
)

tensorflow_model_optimization/python/core/api/BUILD

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
1-
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
1+
# Placeholder: load py_library
22

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

55
licenses(["notice"])
66

7-
py_strict_library(
7+
py_library(
88
name = "api",
99
srcs = [
1010
"__init__.py",
@@ -27,6 +27,7 @@ py_strict_library(
2727
"sparsity/__init__.py",
2828
"sparsity/keras/__init__.py",
2929
],
30+
strict_deps = True,
3031
deps = [
3132
"//tensorflow_model_optimization/python/core/clustering/keras:cluster",
3233
"//tensorflow_model_optimization/python/core/clustering/keras:cluster_config",

tensorflow_model_optimization/python/core/clustering/BUILD

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,13 @@
1-
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
1+
# Placeholder: load py_library
22

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

55
licenses(["notice"])
66

7-
py_strict_library(
7+
py_library(
88
name = "clustering",
99
srcs = ["__init__.py"],
10+
strict_deps = True,
1011
deps = [
1112
"//tensorflow_model_optimization/python/core/clustering/keras", # buildcleaner: keep
1213
],

tensorflow_model_optimization/python/core/clustering/keras/BUILD

Lines changed: 20 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
1-
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library", "py_strict_test")
1+
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
2+
# Placeholder: load py_library
23
# Placeholder: load py_test
34

45
package(default_visibility = [
@@ -7,21 +8,23 @@ package(default_visibility = [
78

89
licenses(["notice"])
910

10-
py_strict_library(
11+
py_library(
1112
name = "keras",
1213
srcs = [
1314
"__init__.py",
1415
],
16+
strict_deps = True,
1517
deps = [
1618
":cluster", # buildcleaner: keep
1719
":clustering_callbacks", # buildcleaner: keep
1820
"//tensorflow_model_optimization/python/core/clustering/keras/experimental", # buildcleaner: keep
1921
],
2022
)
2123

22-
py_strict_library(
24+
py_library(
2325
name = "cluster",
2426
srcs = ["cluster.py"],
27+
strict_deps = True,
2528
visibility = ["//visibility:public"],
2629
deps = [
2730
":cluster_config",
@@ -32,15 +35,17 @@ py_strict_library(
3235
],
3336
)
3437

35-
py_strict_library(
38+
py_library(
3639
name = "cluster_config",
3740
srcs = ["cluster_config.py"],
41+
strict_deps = True,
3842
visibility = ["//visibility:public"],
3943
)
4044

41-
py_strict_library(
45+
py_library(
4246
name = "clustering_registry",
4347
srcs = ["clustering_registry.py"],
48+
strict_deps = True,
4449
visibility = ["//visibility:public"],
4550
deps = [
4651
":clusterable_layer",
@@ -50,18 +55,20 @@ py_strict_library(
5055
],
5156
)
5257

53-
py_strict_library(
58+
py_library(
5459
name = "clusterable_layer",
5560
srcs = ["clusterable_layer.py"],
61+
strict_deps = True,
5662
visibility = ["//visibility:public"],
5763
deps = [
5864
# six dep1,
5965
],
6066
)
6167

62-
py_strict_library(
68+
py_library(
6369
name = "clustering_centroids",
6470
srcs = ["clustering_centroids.py"],
71+
strict_deps = True,
6572
visibility = ["//visibility:public"],
6673
deps = [
6774
":cluster_config",
@@ -72,9 +79,10 @@ py_strict_library(
7279
],
7380
)
7481

75-
py_strict_library(
82+
py_library(
7683
name = "cluster_wrapper",
7784
srcs = ["cluster_wrapper.py"],
85+
strict_deps = True,
7886
visibility = ["//visibility:public"],
7987
deps = [
8088
":cluster_config",
@@ -86,9 +94,10 @@ py_strict_library(
8694
],
8795
)
8896

89-
py_strict_library(
97+
py_library(
9098
name = "clustering_algorithm",
9199
srcs = ["clustering_algorithm.py"],
100+
strict_deps = True,
92101
visibility = ["//visibility:public"],
93102
deps = [
94103
# six dep1,
@@ -97,9 +106,10 @@ py_strict_library(
97106
],
98107
)
99108

100-
py_strict_library(
109+
py_library(
101110
name = "clustering_callbacks",
102111
srcs = ["clustering_callbacks.py"],
112+
strict_deps = True,
103113
visibility = ["//visibility:public"],
104114
deps = [
105115
# tensorflow dep1,

tensorflow_model_optimization/python/core/clustering/keras/experimental/BUILD

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,24 +1,26 @@
1-
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
1+
# Placeholder: load py_library
22

33
package(default_visibility = [
44
"//tensorflow_model_optimization:__subpackages__",
55
])
66

77
licenses(["notice"])
88

9-
py_strict_library(
9+
py_library(
1010
name = "experimental",
1111
srcs = [
1212
"__init__.py",
1313
],
14+
strict_deps = True,
1415
deps = [
1516
":cluster", # buildcleaner: keep
1617
],
1718
)
1819

19-
py_strict_library(
20+
py_library(
2021
name = "cluster",
2122
srcs = ["cluster.py"],
23+
strict_deps = True,
2224
visibility = ["//visibility:public"],
2325
deps = [
2426
"//tensorflow_model_optimization/python/core/clustering/keras:cluster",

tensorflow_model_optimization/python/core/internal/BUILD

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,13 @@
1-
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
1+
# Placeholder: load py_library
22

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

55
licenses(["notice"])
66

7-
py_strict_library(
7+
py_library(
88
name = "internal",
99
srcs = ["__init__.py"],
10+
strict_deps = True,
1011
deps = [
1112
"//tensorflow_model_optimization/python/core/internal/tensor_encoding",
1213
],

tensorflow_model_optimization/python/core/internal/tensor_encoding/BUILD

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,13 @@
1-
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
1+
# Placeholder: load py_library
22

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

55
licenses(["notice"])
66

7-
py_strict_library(
7+
py_library(
88
name = "tensor_encoding",
99
srcs = ["__init__.py"],
10+
strict_deps = True,
1011
deps = [
1112
"//tensorflow_model_optimization/python/core/internal/tensor_encoding/core",
1213
"//tensorflow_model_optimization/python/core/internal/tensor_encoding/encoders",

tensorflow_model_optimization/python/core/internal/tensor_encoding/core/BUILD

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,16 @@
1-
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library", "py_strict_test")
1+
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
2+
# Placeholder: load py_library
23

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

78
licenses(["notice"])
89

9-
py_strict_library(
10+
py_library(
1011
name = "core",
1112
srcs = ["__init__.py"],
13+
strict_deps = True,
1214
visibility = ["//visibility:public"],
1315
deps = [
1416
":core_encoder",
@@ -18,9 +20,10 @@ py_strict_library(
1820
],
1921
)
2022

21-
py_strict_library(
23+
py_library(
2224
name = "core_encoder",
2325
srcs = ["core_encoder.py"],
26+
strict_deps = True,
2427
deps = [
2528
":encoding_stage",
2629
# six dep1,
@@ -43,9 +46,10 @@ py_strict_test(
4346
],
4447
)
4548

46-
py_strict_library(
49+
py_library(
4750
name = "encoding_stage",
4851
srcs = ["encoding_stage.py"],
52+
strict_deps = True,
4953
deps = [
5054
# six dep1,
5155
# tensorflow dep1,
@@ -67,9 +71,10 @@ py_strict_test(
6771
],
6872
)
6973

70-
py_strict_library(
74+
py_library(
7175
name = "gather_encoder",
7276
srcs = ["gather_encoder.py"],
77+
strict_deps = True,
7378
deps = [
7479
":core_encoder",
7580
# tensorflow dep1,
@@ -94,9 +99,10 @@ py_strict_test(
9499
],
95100
)
96101

97-
py_strict_library(
102+
py_library(
98103
name = "simple_encoder",
99104
srcs = ["simple_encoder.py"],
105+
strict_deps = True,
100106
deps = [
101107
":core_encoder",
102108
# tensorflow dep1,

0 commit comments

Comments
 (0)