Skip to content

Commit 6a014e2

Browse files
h-jootensorflower-gardener
authored andcommitted
Automated Code Change
PiperOrigin-RevId: 855977983
1 parent b0b0ad5 commit 6a014e2

File tree

10 files changed

+90
-49
lines changed

10 files changed

+90
-49
lines changed

tensorflow_model_optimization/python/core/quantization/keras/collab_opts/prune_preserve/BUILD

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

44
package(default_visibility = [
@@ -7,21 +7,23 @@ package(default_visibility = [
77

88
licenses(["notice"])
99

10-
py_strict_library(
10+
py_library(
1111
name = "prune_preserve",
1212
srcs = [
1313
"__init__.py",
1414
],
15+
strict_deps = True,
1516
deps = [
1617
":default_8bit_prune_preserve_quantize_scheme", # buildcleaner: keep
1718
],
1819
)
1920

20-
py_strict_library(
21+
py_library(
2122
name = "prune_preserve_quantize_registry",
2223
srcs = [
2324
"prune_preserve_quantize_registry.py",
2425
],
26+
strict_deps = True,
2527
deps = [
2628
# tensorflow dep1,
2729
"//tensorflow_model_optimization/python/core/keras:compat",
@@ -49,11 +51,12 @@ py_test(
4951
],
5052
)
5153

52-
py_strict_library(
54+
py_library(
5355
name = "default_8bit_prune_preserve_quantize_scheme",
5456
srcs = [
5557
"default_8bit_prune_preserve_quantize_scheme.py",
5658
],
59+
strict_deps = True,
5760
visibility = ["//visibility:public"],
5861
deps = [
5962
":prune_preserve_quantize_registry",

tensorflow_model_optimization/python/core/quantization/keras/default_8bit/BUILD

Lines changed: 13 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
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")
22
# Placeholder: load py_library
33
# Placeholder: load py_test
44

@@ -8,19 +8,21 @@ package(default_visibility = [
88

99
licenses(["notice"])
1010

11-
py_strict_library(
11+
py_library(
1212
name = "default_8bit",
1313
srcs = [
1414
"__init__.py",
1515
],
16+
strict_deps = True,
1617
deps = [],
1718
)
1819

19-
py_strict_library(
20+
py_library(
2021
name = "default_8bit_quantizers",
2122
srcs = [
2223
"default_8bit_quantizers.py",
2324
],
25+
strict_deps = True,
2426
deps = [
2527
# tensorflow dep1,
2628
"//tensorflow_model_optimization/python/core/keras:compat",
@@ -43,22 +45,24 @@ py_test(
4345
],
4446
)
4547

46-
py_strict_library(
48+
py_library(
4749
name = "default_8bit_quantize_configs",
4850
srcs = [
4951
"default_8bit_quantize_configs.py",
5052
],
53+
strict_deps = True,
5154
deps = [
5255
"//tensorflow_model_optimization/python/core/quantization/keras:quantize_config",
5356
"//tensorflow_model_optimization/python/core/quantization/keras:quantizers",
5457
],
5558
)
5659

57-
py_strict_library(
60+
py_library(
5861
name = "default_8bit_quantize_registry",
5962
srcs = [
6063
"default_8bit_quantize_registry.py",
6164
],
65+
strict_deps = True,
6266
deps = [
6367
# tensorflow dep1,
6468
"//tensorflow_model_optimization/python/core/keras:compat",
@@ -130,11 +134,12 @@ py_strict_test(
130134
],
131135
)
132136

133-
py_strict_library(
137+
py_library(
134138
name = "default_8bit_quantize_layout_transform",
135139
srcs = [
136140
"default_8bit_quantize_layout_transform.py",
137141
],
142+
strict_deps = True,
138143
deps = [
139144
":default_8bit_transforms",
140145
# tensorflow dep1,
@@ -160,11 +165,12 @@ py_test(
160165
],
161166
)
162167

163-
py_strict_library(
168+
py_library(
164169
name = "default_8bit_quantize_scheme",
165170
srcs = [
166171
"default_8bit_quantize_scheme.py",
167172
],
173+
strict_deps = True,
168174
deps = [
169175
":default_8bit_quantize_layout_transform",
170176
":default_8bit_quantize_registry",

tensorflow_model_optimization/python/core/quantization/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
":quantization", # buildcleaner: keep
1617
],
1718
)
1819

19-
py_strict_library(
20+
py_library(
2021
name = "quantization",
2122
srcs = ["__init__.py"],
23+
strict_deps = True,
2224
deps = [
2325
"//tensorflow_model_optimization/python/core/quantization/keras", # buildcleaner: keep
2426
],

tensorflow_model_optimization/python/core/quantization/keras/experimental/default_n_bit/BUILD

Lines changed: 13 additions & 7 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", "py_strict_test")
1+
load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
22
# Placeholder: load py_library
33
# Placeholder: load py_test
44

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

77
licenses(["notice"])
88

9-
py_strict_library(
9+
py_library(
1010
name = "default_n_bit",
1111
srcs = [
1212
"__init__.py",
1313
],
14+
strict_deps = True,
1415
deps = [],
1516
)
1617

17-
py_strict_library(
18+
py_library(
1819
name = "default_n_bit_quantizers",
1920
srcs = [
2021
"default_n_bit_quantizers.py",
2122
],
23+
strict_deps = True,
2224
deps = [
2325
# tensorflow dep1,
2426
"//tensorflow_model_optimization/python/core/keras:compat",
@@ -40,22 +42,24 @@ py_test(
4042
],
4143
)
4244

43-
py_strict_library(
45+
py_library(
4446
name = "default_n_bit_quantize_configs",
4547
srcs = [
4648
"default_n_bit_quantize_configs.py",
4749
],
50+
strict_deps = True,
4851
deps = [
4952
"//tensorflow_model_optimization/python/core/quantization/keras:quantize_config",
5053
"//tensorflow_model_optimization/python/core/quantization/keras:quantizers",
5154
],
5255
)
5356

54-
py_strict_library(
57+
py_library(
5558
name = "default_n_bit_quantize_registry",
5659
srcs = [
5760
"default_n_bit_quantize_registry.py",
5861
],
62+
strict_deps = True,
5963
deps = [
6064
# tensorflow dep1,
6165
"//tensorflow_model_optimization/python/core/keras:compat",
@@ -102,11 +106,12 @@ py_library(
102106
],
103107
)
104108

105-
py_strict_library(
109+
py_library(
106110
name = "default_n_bit_quantize_layout_transform",
107111
srcs = [
108112
"default_n_bit_quantize_layout_transform.py",
109113
],
114+
strict_deps = True,
110115
deps = [
111116
":default_n_bit_transforms",
112117
# tensorflow dep1,
@@ -139,11 +144,12 @@ py_strict_test(
139144
],
140145
)
141146

142-
py_strict_library(
147+
py_library(
143148
name = "default_n_bit_quantize_scheme",
144149
srcs = [
145150
"default_n_bit_quantize_scheme.py",
146151
],
152+
strict_deps = True,
147153
visibility = ["//visibility:public"],
148154
deps = [
149155
":default_n_bit_quantize_layout_transform",

tensorflow_model_optimization/python/core/quantization/keras/graph_transformations/BUILD

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,24 +1,27 @@
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:__subpackages__",
56
])
67

78
licenses(["notice"])
89

9-
py_strict_library(
10+
py_library(
1011
name = "graph_transformations",
1112
srcs = [
1213
"__init__.py",
1314
],
15+
strict_deps = True,
1416
deps = [],
1517
)
1618

17-
py_strict_library(
19+
py_library(
1820
name = "transforms",
1921
srcs = [
2022
"transforms.py",
2123
],
24+
strict_deps = True,
2225
visibility = ["//visibility:public"],
2326
deps = [
2427
# six dep1,
@@ -38,11 +41,12 @@ py_strict_test(
3841
],
3942
)
4043

41-
py_strict_library(
44+
py_library(
4245
name = "model_transformer",
4346
srcs = [
4447
"model_transformer.py",
4548
],
49+
strict_deps = True,
4650
visibility = ["//visibility:public"],
4751
deps = [
4852
":transforms",

tensorflow_model_optimization/python/core/quantization/keras/layers/BUILD

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,31 +1,34 @@
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 = "layers",
1111
srcs = [
1212
"__init__.py",
1313
],
14+
strict_deps = True,
1415
deps = [],
1516
)
1617

17-
py_strict_library(
18+
py_library(
1819
name = "conv_batchnorm_test_utils",
1920
srcs = ["conv_batchnorm_test_utils.py"],
21+
strict_deps = True,
2022
deps = [
2123
# tensorflow dep1,
2224
"//tensorflow_model_optimization/python/core/keras:compat",
2325
],
2426
)
2527

26-
py_strict_library(
28+
py_library(
2729
name = "dense_batchnorm_test_utils",
2830
srcs = ["dense_batchnorm_test_utils.py"],
31+
strict_deps = True,
2932
deps = [
3033
# tensorflow dep1,
3134
"//tensorflow_model_optimization/python/core/keras:compat",

tensorflow_model_optimization/python/core/sparsity/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 = "sparsity",
99
srcs = ["__init__.py"],
10+
strict_deps = True,
1011
deps = [
1112
"//tensorflow_model_optimization/python/core/sparsity/keras", # buildcleaner: keep
1213
],

0 commit comments

Comments
 (0)