From 4d72fd299c689ae01233a95a31ef899f43881efe Mon Sep 17 00:00:00 2001 From: Hana Joo Date: Wed, 14 Jan 2026 00:38:21 -0800 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 856082507 --- tensorflow_model_optimization/BUILD | 6 ++- tensorflow_model_optimization/python/BUILD | 5 ++- .../python/core/BUILD | 8 ++-- .../python/core/api/BUILD | 5 ++- .../python/core/clustering/BUILD | 5 ++- .../python/core/clustering/keras/BUILD | 30 ++++++++----- .../core/clustering/keras/experimental/BUILD | 8 ++-- .../python/core/internal/BUILD | 5 ++- .../core/internal/tensor_encoding/BUILD | 5 ++- .../core/internal/tensor_encoding/core/BUILD | 18 +++++--- .../internal/tensor_encoding/encoders/BUILD | 9 ++-- .../internal/tensor_encoding/stages/BUILD | 9 ++-- .../tensor_encoding/stages/research/BUILD | 18 +++++--- .../internal/tensor_encoding/testing/BUILD | 9 ++-- .../core/internal/tensor_encoding/utils/BUILD | 12 ++++-- .../python/core/keras/BUILD | 18 +++++--- .../python/core/keras/testing/BUILD | 5 ++- .../python/core/quantization/BUILD | 5 ++- .../python/core/quantization/keras/BUILD | 42 ++++++++++++------- .../core/quantization/keras/collab_opts/BUILD | 5 ++- .../keras/collab_opts/cluster_preserve/BUILD | 14 ++++--- 21 files changed, 157 insertions(+), 84 deletions(-) diff --git a/tensorflow_model_optimization/BUILD b/tensorflow_model_optimization/BUILD index d9443828..20b13381 100644 --- a/tensorflow_model_optimization/BUILD +++ b/tensorflow_model_optimization/BUILD @@ -5,7 +5,8 @@ # 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"]) @@ -13,9 +14,10 @@ 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 diff --git a/tensorflow_model_optimization/python/BUILD b/tensorflow_model_optimization/python/BUILD index 501ec572..1aca10c9 100644 --- a/tensorflow_model_optimization/python/BUILD +++ b/tensorflow_model_optimization/python/BUILD @@ -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 ], diff --git a/tensorflow_model_optimization/python/core/BUILD b/tensorflow_model_optimization/python/core/BUILD index d714a599..3d15a666 100644 --- a/tensorflow_model_optimization/python/core/BUILD +++ b/tensorflow_model_optimization/python/core/BUILD @@ -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 @@ -17,7 +18,8 @@ py_strict_library( ], ) -py_strict_library( +py_library( name = "version", srcs = ["version.py"], + strict_deps = True, ) diff --git a/tensorflow_model_optimization/python/core/api/BUILD b/tensorflow_model_optimization/python/core/api/BUILD index 118425b1..d1505853 100644 --- a/tensorflow_model_optimization/python/core/api/BUILD +++ b/tensorflow_model_optimization/python/core/api/BUILD @@ -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", @@ -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", diff --git a/tensorflow_model_optimization/python/core/clustering/BUILD b/tensorflow_model_optimization/python/core/clustering/BUILD index 8126b974..894c0606 100644 --- a/tensorflow_model_optimization/python/core/clustering/BUILD +++ b/tensorflow_model_optimization/python/core/clustering/BUILD @@ -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 ], diff --git a/tensorflow_model_optimization/python/core/clustering/keras/BUILD b/tensorflow_model_optimization/python/core/clustering/keras/BUILD index db9bc474..252f047a 100644 --- a/tensorflow_model_optimization/python/core/clustering/keras/BUILD +++ b/tensorflow_model_optimization/python/core/clustering/keras/BUILD @@ -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 = [ @@ -7,11 +8,12 @@ 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 @@ -19,9 +21,10 @@ py_strict_library( ], ) -py_strict_library( +py_library( name = "cluster", srcs = ["cluster.py"], + strict_deps = True, visibility = ["//visibility:public"], deps = [ ":cluster_config", @@ -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", @@ -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", @@ -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", @@ -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, @@ -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, diff --git a/tensorflow_model_optimization/python/core/clustering/keras/experimental/BUILD b/tensorflow_model_optimization/python/core/clustering/keras/experimental/BUILD index 843ddb32..78798b08 100644 --- a/tensorflow_model_optimization/python/core/clustering/keras/experimental/BUILD +++ b/tensorflow_model_optimization/python/core/clustering/keras/experimental/BUILD @@ -1,4 +1,4 @@ -load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library") +# Placeholder: load py_library package(default_visibility = [ "//tensorflow_model_optimization:__subpackages__", @@ -6,19 +6,21 @@ package(default_visibility = [ 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", diff --git a/tensorflow_model_optimization/python/core/internal/BUILD b/tensorflow_model_optimization/python/core/internal/BUILD index 5fcc3418..caaa8ccc 100644 --- a/tensorflow_model_optimization/python/core/internal/BUILD +++ b/tensorflow_model_optimization/python/core/internal/BUILD @@ -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", ], diff --git a/tensorflow_model_optimization/python/core/internal/tensor_encoding/BUILD b/tensorflow_model_optimization/python/core/internal/tensor_encoding/BUILD index d6cdac46..4f420a23 100644 --- a/tensorflow_model_optimization/python/core/internal/tensor_encoding/BUILD +++ b/tensorflow_model_optimization/python/core/internal/tensor_encoding/BUILD @@ -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", diff --git a/tensorflow_model_optimization/python/core/internal/tensor_encoding/core/BUILD b/tensorflow_model_optimization/python/core/internal/tensor_encoding/core/BUILD index 183a8572..36dd5110 100644 --- a/tensorflow_model_optimization/python/core/internal/tensor_encoding/core/BUILD +++ b/tensorflow_model_optimization/python/core/internal/tensor_encoding/core/BUILD @@ -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 package(default_visibility = [ "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__", @@ -6,9 +7,10 @@ package(default_visibility = [ licenses(["notice"]) -py_strict_library( +py_library( name = "core", srcs = ["__init__.py"], + strict_deps = True, visibility = ["//visibility:public"], deps = [ ":core_encoder", @@ -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, @@ -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, @@ -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, @@ -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, diff --git a/tensorflow_model_optimization/python/core/internal/tensor_encoding/encoders/BUILD b/tensorflow_model_optimization/python/core/internal/tensor_encoding/encoders/BUILD index d198eae0..47cedc23 100644 --- a/tensorflow_model_optimization/python/core/internal/tensor_encoding/encoders/BUILD +++ b/tensorflow_model_optimization/python/core/internal/tensor_encoding/encoders/BUILD @@ -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 package(default_visibility = [ "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__", @@ -6,18 +7,20 @@ package(default_visibility = [ 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", diff --git a/tensorflow_model_optimization/python/core/internal/tensor_encoding/stages/BUILD b/tensorflow_model_optimization/python/core/internal/tensor_encoding/stages/BUILD index e79076a1..954113fc 100644 --- a/tensorflow_model_optimization/python/core/internal/tensor_encoding/stages/BUILD +++ b/tensorflow_model_optimization/python/core/internal/tensor_encoding/stages/BUILD @@ -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 package(default_visibility = [ "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__", @@ -6,9 +7,10 @@ package(default_visibility = [ licenses(["notice"]) -py_strict_library( +py_library( name = "stages", srcs = ["__init__.py"], + strict_deps = True, visibility = ["//visibility:public"], deps = [ ":stages_impl", @@ -16,9 +18,10 @@ py_strict_library( ], ) -py_strict_library( +py_library( name = "stages_impl", srcs = ["stages_impl.py"], + strict_deps = True, deps = [ # numpy dep1, # tensorflow dep1, diff --git a/tensorflow_model_optimization/python/core/internal/tensor_encoding/stages/research/BUILD b/tensorflow_model_optimization/python/core/internal/tensor_encoding/stages/research/BUILD index 82c58480..69530287 100644 --- a/tensorflow_model_optimization/python/core/internal/tensor_encoding/stages/research/BUILD +++ b/tensorflow_model_optimization/python/core/internal/tensor_encoding/stages/research/BUILD @@ -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 package(default_visibility = [ "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__", @@ -6,9 +7,10 @@ package(default_visibility = [ licenses(["notice"]) -py_strict_library( +py_library( name = "research", srcs = ["__init__.py"], + strict_deps = True, visibility = ["//visibility:public"], deps = [ ":clipping", @@ -18,9 +20,10 @@ py_strict_library( ], ) -py_strict_library( +py_library( name = "clipping", srcs = ["clipping.py"], + strict_deps = True, deps = [ # tensorflow dep1, "//tensorflow_model_optimization/python/core/internal/tensor_encoding/core:encoding_stage", @@ -41,9 +44,10 @@ py_strict_test( ], ) -py_strict_library( +py_library( name = "kashin", srcs = ["kashin.py"], + strict_deps = True, deps = [ # numpy dep1, # tensorflow dep1, @@ -66,9 +70,10 @@ py_strict_test( ], ) -py_strict_library( +py_library( name = "quantization", srcs = ["quantization.py"], + strict_deps = True, deps = [ # tensorflow dep1, "//tensorflow_model_optimization/python/core/internal/tensor_encoding/core:encoding_stage", @@ -90,9 +95,10 @@ py_strict_test( ], ) -py_strict_library( +py_library( name = "misc", srcs = ["misc.py"], + strict_deps = True, deps = [ # tensorflow dep1, "//tensorflow_model_optimization/python/core/internal/tensor_encoding/core:encoding_stage", diff --git a/tensorflow_model_optimization/python/core/internal/tensor_encoding/testing/BUILD b/tensorflow_model_optimization/python/core/internal/tensor_encoding/testing/BUILD index 1f611447..fd2c830f 100644 --- a/tensorflow_model_optimization/python/core/internal/tensor_encoding/testing/BUILD +++ b/tensorflow_model_optimization/python/core/internal/tensor_encoding/testing/BUILD @@ -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 package(default_visibility = [ "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__", @@ -6,18 +7,20 @@ package(default_visibility = [ licenses(["notice"]) -py_strict_library( +py_library( name = "testing", srcs = ["__init__.py"], + strict_deps = True, visibility = ["//visibility:public"], deps = [ ":test_utils", ], ) -py_strict_library( +py_library( name = "test_utils", srcs = ["test_utils.py"], + strict_deps = True, deps = [ # absl/testing:parameterized dep1, # numpy dep1, diff --git a/tensorflow_model_optimization/python/core/internal/tensor_encoding/utils/BUILD b/tensorflow_model_optimization/python/core/internal/tensor_encoding/utils/BUILD index 174e9fa4..08fbbfd6 100644 --- a/tensorflow_model_optimization/python/core/internal/tensor_encoding/utils/BUILD +++ b/tensorflow_model_optimization/python/core/internal/tensor_encoding/utils/BUILD @@ -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 package(default_visibility = [ "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__", @@ -6,9 +7,10 @@ package(default_visibility = [ licenses(["notice"]) -py_strict_library( +py_library( name = "utils", srcs = ["__init__.py"], + strict_deps = True, visibility = ["//visibility:public"], deps = [ ":py_utils", @@ -16,9 +18,10 @@ py_strict_library( ], ) -py_strict_library( +py_library( name = "py_utils", srcs = ["py_utils.py"], + strict_deps = True, deps = [ # numpy dep1, # six dep1, @@ -40,9 +43,10 @@ py_strict_test( ], ) -py_strict_library( +py_library( name = "tf_utils", srcs = ["tf_utils.py"], + strict_deps = True, deps = [ # numpy dep1, # tensorflow dep1, diff --git a/tensorflow_model_optimization/python/core/keras/BUILD b/tensorflow_model_optimization/python/core/keras/BUILD index 032633f0..faab2735 100644 --- a/tensorflow_model_optimization/python/core/keras/BUILD +++ b/tensorflow_model_optimization/python/core/keras/BUILD @@ -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 package(default_visibility = [ "//tensorflow_model_optimization:__subpackages__", @@ -6,9 +7,10 @@ package(default_visibility = [ licenses(["notice"]) -py_strict_library( +py_library( name = "keras", srcs = ["__init__.py"], + strict_deps = True, deps = [ ":compat", ":utils", @@ -16,9 +18,10 @@ py_strict_library( ) # TODO(alanchiao): split into test_utils and utils. -py_strict_library( +py_library( name = "test_utils", srcs = ["test_utils.py"], + strict_deps = True, deps = [ ":compat", # numpy dep1, @@ -26,17 +29,19 @@ py_strict_library( ], ) -py_strict_library( +py_library( name = "compat", srcs = ["compat.py"], + strict_deps = True, deps = [ # tensorflow dep1, ], ) -py_strict_library( +py_library( name = "utils", srcs = ["utils.py"], + strict_deps = True, visibility = ["//visibility:public"], deps = [ # tensorflow dep1, @@ -45,9 +50,10 @@ py_strict_library( ], ) -py_strict_library( +py_library( name = "metrics", srcs = ["metrics.py"], + strict_deps = True, deps = [ # python/eager:monitoring tensorflow dep2, ], diff --git a/tensorflow_model_optimization/python/core/keras/testing/BUILD b/tensorflow_model_optimization/python/core/keras/testing/BUILD index 5512d16c..212c0e8c 100644 --- a/tensorflow_model_optimization/python/core/keras/testing/BUILD +++ b/tensorflow_model_optimization/python/core/keras/testing/BUILD @@ -1,4 +1,4 @@ -load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library") +# Placeholder: load py_library package(default_visibility = [ "//tensorflow_model_optimization:__subpackages__", @@ -6,10 +6,11 @@ package(default_visibility = [ licenses(["notice"]) -py_strict_library( +py_library( name = "test_utils_mnist", testonly = 1, srcs = ["test_utils_mnist.py"], + strict_deps = True, visibility = ["//visibility:public"], deps = [ # numpy dep1, diff --git a/tensorflow_model_optimization/python/core/quantization/BUILD b/tensorflow_model_optimization/python/core/quantization/BUILD index 8cf88190..a649c0c9 100644 --- a/tensorflow_model_optimization/python/core/quantization/BUILD +++ b/tensorflow_model_optimization/python/core/quantization/BUILD @@ -1,12 +1,13 @@ -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 = "quantization", srcs = ["__init__.py"], + strict_deps = True, deps = [ "//tensorflow_model_optimization/python/core/quantization/keras", # buildcleaner: keep ], diff --git a/tensorflow_model_optimization/python/core/quantization/keras/BUILD b/tensorflow_model_optimization/python/core/quantization/keras/BUILD index 7fb0dbd9..c1a5adbc 100644 --- a/tensorflow_model_optimization/python/core/quantization/keras/BUILD +++ b/tensorflow_model_optimization/python/core/quantization/keras/BUILD @@ -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 package(default_visibility = [ "//tensorflow_model_optimization:__subpackages__", @@ -7,11 +8,12 @@ package(default_visibility = [ licenses(["notice"]) -py_strict_library( +py_library( name = "keras", srcs = [ "__init__.py", ], + strict_deps = True, deps = [ ":quantize", # build-cleaner: keep # APIs are not exposed, but still needed for internal imports. @@ -23,9 +25,10 @@ py_strict_library( ], ) -py_strict_library( +py_library( name = "quant_ops", srcs = ["quant_ops.py"], + strict_deps = True, deps = [ # absl/logging dep1, # tensorflow dep1, @@ -46,11 +49,12 @@ py_strict_test( ], ) -py_strict_library( +py_library( name = "quantizers", srcs = [ "quantizers.py", ], + strict_deps = True, visibility = ["//visibility:public"], deps = [ ":quant_ops", @@ -77,11 +81,12 @@ py_strict_test( ], ) -py_strict_library( +py_library( name = "quantize_config", srcs = [ "quantize_config.py", ], + strict_deps = True, visibility = ["//visibility:public"], deps = [ ":quantizers", @@ -91,33 +96,36 @@ py_strict_library( ], ) -py_strict_library( +py_library( name = "quantize_registry", srcs = [ "quantize_registry.py", ], + strict_deps = True, visibility = ["//visibility:public"], deps = [ # six dep1, ], ) -py_strict_library( +py_library( name = "quantize_layout_transform", srcs = [ "quantize_layout_transform.py", ], + strict_deps = True, visibility = ["//visibility:public"], deps = [ # six dep1, ], ) -py_strict_library( +py_library( name = "quantize_annotate", srcs = [ "quantize_annotate.py", ], + strict_deps = True, visibility = ["//visibility:public"], deps = [ ":utils", @@ -142,11 +150,12 @@ py_strict_test( ], ) -py_strict_library( +py_library( name = "quantize_aware_activation", srcs = [ "quantize_aware_activation.py", ], + strict_deps = True, visibility = ["//visibility:public"], deps = [ ":utils", @@ -174,11 +183,12 @@ py_strict_test( ], ) -py_strict_library( +py_library( name = "quantize_layer", srcs = [ "quantize_layer.py", ], + strict_deps = True, visibility = ["//visibility:public"], deps = [ ":quantizers", @@ -205,11 +215,12 @@ py_strict_test( ], ) -py_strict_library( +py_library( name = "quantize_wrapper", srcs = [ "quantize_wrapper.py", ], + strict_deps = True, visibility = ["//visibility:public"], deps = [ ":quantize_aware_activation", @@ -240,11 +251,12 @@ py_strict_test( ], ) -py_strict_library( +py_library( name = "quantize", srcs = [ "quantize.py", ], + strict_deps = True, visibility = ["//visibility:public"], deps = [ ":quantize_annotate", @@ -333,11 +345,12 @@ py_strict_test( ], ) -py_strict_library( +py_library( name = "utils", srcs = [ "utils.py", ], + strict_deps = True, visibility = ["//visibility:public"], deps = [ # tensorflow dep1, @@ -345,11 +358,12 @@ py_strict_library( ], ) -py_strict_library( +py_library( name = "quantize_scheme", srcs = [ "quantize_scheme.py", ], + strict_deps = True, visibility = ["//visibility:public"], deps = [ # six dep1, diff --git a/tensorflow_model_optimization/python/core/quantization/keras/collab_opts/BUILD b/tensorflow_model_optimization/python/core/quantization/keras/collab_opts/BUILD index 990c3912..389110ad 100644 --- a/tensorflow_model_optimization/python/core/quantization/keras/collab_opts/BUILD +++ b/tensorflow_model_optimization/python/core/quantization/keras/collab_opts/BUILD @@ -1,12 +1,13 @@ -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 = "collab_opts", srcs = ["__init__.py"], + strict_deps = True, deps = [ "//tensorflow_model_optimization/python/core/quantization/keras/collab_opts/cluster_preserve", # buildcleaner: keep "//tensorflow_model_optimization/python/core/quantization/keras/collab_opts/prune_preserve", # buildcleaner: keep diff --git a/tensorflow_model_optimization/python/core/quantization/keras/collab_opts/cluster_preserve/BUILD b/tensorflow_model_optimization/python/core/quantization/keras/collab_opts/cluster_preserve/BUILD index 201a37b1..a30b6405 100644 --- a/tensorflow_model_optimization/python/core/quantization/keras/collab_opts/cluster_preserve/BUILD +++ b/tensorflow_model_optimization/python/core/quantization/keras/collab_opts/cluster_preserve/BUILD @@ -1,4 +1,4 @@ -load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library") +# Placeholder: load py_library # Placeholder: load py_test package(default_visibility = [ @@ -7,32 +7,35 @@ package(default_visibility = [ licenses(["notice"]) -py_strict_library( +py_library( name = "cluster_preserve", srcs = [ "__init__.py", ], + strict_deps = True, deps = [ ":default_8bit_cluster_preserve_quantize_scheme", # buildcleaner: keep ], ) -py_strict_library( +py_library( name = "cluster_utils", srcs = [ "cluster_utils.py", ], + strict_deps = True, deps = [ # tensorflow dep1, "//tensorflow_model_optimization/python/core/keras:compat", ], ) -py_strict_library( +py_library( name = "cluster_preserve_quantize_registry", srcs = [ "cluster_preserve_quantize_registry.py", ], + strict_deps = True, deps = [ ":cluster_utils", # tensorflow dep1, @@ -83,11 +86,12 @@ py_test( ], ) -py_strict_library( +py_library( name = "default_8bit_cluster_preserve_quantize_scheme", srcs = [ "default_8bit_cluster_preserve_quantize_scheme.py", ], + strict_deps = True, deps = [ ":cluster_preserve_quantize_registry", "//tensorflow_model_optimization/python/core/quantization/keras/default_8bit:default_8bit_quantize_scheme",