From 16a79522962031b50ccb28954a471c70d47fd5e8 Mon Sep 17 00:00:00 2001 From: Hana Joo Date: Mon, 16 Feb 2026 13:45:21 -0800 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 870988069 --- .../keras/collab_opts/cluster_preserve/BUILD | 11 +++++--- .../keras/collab_opts/prune_preserve/BUILD | 7 +++--- .../quantization/keras/default_8bit/BUILD | 16 +++++++----- .../quantization/keras/experimental/BUILD | 4 +-- .../keras/experimental/default_n_bit/BUILD | 15 ++++++----- .../keras/graph_transformations/BUILD | 6 ++--- .../core/quantization/keras/layers/BUILD | 6 ++--- .../python/core/sparsity/BUILD | 2 +- .../python/core/sparsity/keras/BUILD | 25 ++++++++++--------- .../python/core/sparsity/keras/tools/BUILD | 5 ++-- .../examples/clustering/keras/imdb/BUILD | 2 +- 11 files changed, 56 insertions(+), 43 deletions(-) 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 a30b6405..e8f42e6f 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 @@ -12,7 +12,7 @@ py_library( srcs = [ "__init__.py", ], - strict_deps = True, + strict_deps = False, deps = [ ":default_8bit_cluster_preserve_quantize_scheme", # buildcleaner: keep ], @@ -23,7 +23,7 @@ py_library( srcs = [ "cluster_utils.py", ], - strict_deps = True, + strict_deps = False, deps = [ # tensorflow dep1, "//tensorflow_model_optimization/python/core/keras:compat", @@ -35,7 +35,7 @@ py_library( srcs = [ "cluster_preserve_quantize_registry.py", ], - strict_deps = True, + strict_deps = False, deps = [ ":cluster_utils", # tensorflow dep1, @@ -54,6 +54,7 @@ py_test( srcs = [ "cluster_preserve_quantize_registry_test.py", ], + strict_deps = False, visibility = ["//visibility:private"], deps = [ ":cluster_preserve_quantize_registry", @@ -72,6 +73,7 @@ py_test( srcs = [ "mnist_prune_cluster_preserve_qat_test.py", ], + strict_deps = False, tags = ["requires-net:external"], visibility = ["//visibility:private"], deps = [ @@ -91,7 +93,7 @@ py_library( srcs = [ "default_8bit_cluster_preserve_quantize_scheme.py", ], - strict_deps = True, + strict_deps = False, deps = [ ":cluster_preserve_quantize_registry", "//tensorflow_model_optimization/python/core/quantization/keras/default_8bit:default_8bit_quantize_scheme", @@ -103,6 +105,7 @@ py_test( srcs = [ "cluster_preserve_integration_test.py", ], + strict_deps = False, visibility = ["//visibility:private"], deps = [ ":default_8bit_cluster_preserve_quantize_scheme", diff --git a/tensorflow_model_optimization/python/core/quantization/keras/collab_opts/prune_preserve/BUILD b/tensorflow_model_optimization/python/core/quantization/keras/collab_opts/prune_preserve/BUILD index e27cc345..6ff97f40 100644 --- a/tensorflow_model_optimization/python/core/quantization/keras/collab_opts/prune_preserve/BUILD +++ b/tensorflow_model_optimization/python/core/quantization/keras/collab_opts/prune_preserve/BUILD @@ -12,7 +12,7 @@ py_library( srcs = [ "__init__.py", ], - strict_deps = True, + strict_deps = False, deps = [ ":default_8bit_prune_preserve_quantize_scheme", # buildcleaner: keep ], @@ -23,7 +23,7 @@ py_library( srcs = [ "prune_preserve_quantize_registry.py", ], - strict_deps = True, + strict_deps = False, deps = [ # tensorflow dep1, "//tensorflow_model_optimization/python/core/keras:compat", @@ -39,6 +39,7 @@ py_test( srcs = [ "prune_preserve_quantize_registry_test.py", ], + strict_deps = False, visibility = ["//visibility:private"], deps = [ ":prune_preserve_quantize_registry", @@ -56,7 +57,7 @@ py_library( srcs = [ "default_8bit_prune_preserve_quantize_scheme.py", ], - strict_deps = True, + strict_deps = False, visibility = ["//visibility:public"], deps = [ ":prune_preserve_quantize_registry", diff --git a/tensorflow_model_optimization/python/core/quantization/keras/default_8bit/BUILD b/tensorflow_model_optimization/python/core/quantization/keras/default_8bit/BUILD index 18e45eb6..b0e17757 100644 --- a/tensorflow_model_optimization/python/core/quantization/keras/default_8bit/BUILD +++ b/tensorflow_model_optimization/python/core/quantization/keras/default_8bit/BUILD @@ -13,7 +13,7 @@ py_library( srcs = [ "__init__.py", ], - strict_deps = True, + strict_deps = False, deps = [], ) @@ -22,7 +22,7 @@ py_library( srcs = [ "default_8bit_quantizers.py", ], - strict_deps = True, + strict_deps = False, deps = [ # tensorflow dep1, "//tensorflow_model_optimization/python/core/keras:compat", @@ -36,6 +36,7 @@ py_test( srcs = [ "default_8bit_quantizers_test.py", ], + strict_deps = False, deps = [ ":default_8bit_quantizers", # absl/testing:parameterized dep1, @@ -50,7 +51,7 @@ py_library( srcs = [ "default_8bit_quantize_configs.py", ], - strict_deps = True, + strict_deps = False, deps = [ "//tensorflow_model_optimization/python/core/quantization/keras:quantize_config", "//tensorflow_model_optimization/python/core/quantization/keras:quantizers", @@ -62,7 +63,7 @@ py_library( srcs = [ "default_8bit_quantize_registry.py", ], - strict_deps = True, + strict_deps = False, deps = [ # tensorflow dep1, "//tensorflow_model_optimization/python/core/keras:compat", @@ -80,6 +81,7 @@ py_test( srcs = [ "default_8bit_quantize_registry_test.py", ], + strict_deps = False, deps = [ ":default_8bit_quantize_registry", # absl/testing:parameterized dep1, @@ -97,6 +99,7 @@ py_library( srcs = [ "default_8bit_transforms.py", ], + strict_deps = False, deps = [ # numpy dep1, # tensorflow dep1, @@ -139,7 +142,7 @@ py_library( srcs = [ "default_8bit_quantize_layout_transform.py", ], - strict_deps = True, + strict_deps = False, deps = [ ":default_8bit_transforms", # tensorflow dep1, @@ -154,6 +157,7 @@ py_test( name = "quantize_numerical_test", size = "large", srcs = ["quantize_numerical_test.py"], + strict_deps = False, deps = [ # absl/testing:parameterized dep1, # google/protobuf:use_fast_cpp_protos dep1, # Automatically added @@ -170,7 +174,7 @@ py_library( srcs = [ "default_8bit_quantize_scheme.py", ], - strict_deps = True, + strict_deps = False, deps = [ ":default_8bit_quantize_layout_transform", ":default_8bit_quantize_registry", diff --git a/tensorflow_model_optimization/python/core/quantization/keras/experimental/BUILD b/tensorflow_model_optimization/python/core/quantization/keras/experimental/BUILD index a3dc2ca5..1e22221c 100644 --- a/tensorflow_model_optimization/python/core/quantization/keras/experimental/BUILD +++ b/tensorflow_model_optimization/python/core/quantization/keras/experimental/BUILD @@ -11,7 +11,7 @@ py_library( srcs = [ "__init__.py", ], - strict_deps = True, + strict_deps = False, deps = [ ":quantization", # buildcleaner: keep ], @@ -20,7 +20,7 @@ py_library( py_library( name = "quantization", srcs = ["__init__.py"], - strict_deps = True, + strict_deps = False, deps = [ "//tensorflow_model_optimization/python/core/quantization/keras", # buildcleaner: keep ], diff --git a/tensorflow_model_optimization/python/core/quantization/keras/experimental/default_n_bit/BUILD b/tensorflow_model_optimization/python/core/quantization/keras/experimental/default_n_bit/BUILD index 655d9c87..da82f1dd 100644 --- a/tensorflow_model_optimization/python/core/quantization/keras/experimental/default_n_bit/BUILD +++ b/tensorflow_model_optimization/python/core/quantization/keras/experimental/default_n_bit/BUILD @@ -11,7 +11,7 @@ py_library( srcs = [ "__init__.py", ], - strict_deps = True, + strict_deps = False, deps = [], ) @@ -20,7 +20,7 @@ py_library( srcs = [ "default_n_bit_quantizers.py", ], - strict_deps = True, + strict_deps = False, deps = [ # tensorflow dep1, "//tensorflow_model_optimization/python/core/keras:compat", @@ -33,6 +33,7 @@ py_test( srcs = [ "default_n_bit_quantizers_test.py", ], + strict_deps = False, deps = [ ":default_n_bit_quantizers", # absl/testing:parameterized dep1, @@ -47,7 +48,7 @@ py_library( srcs = [ "default_n_bit_quantize_configs.py", ], - strict_deps = True, + strict_deps = False, deps = [ "//tensorflow_model_optimization/python/core/quantization/keras:quantize_config", "//tensorflow_model_optimization/python/core/quantization/keras:quantizers", @@ -59,7 +60,7 @@ py_library( srcs = [ "default_n_bit_quantize_registry.py", ], - strict_deps = True, + strict_deps = False, deps = [ # tensorflow dep1, "//tensorflow_model_optimization/python/core/keras:compat", @@ -76,6 +77,7 @@ py_test( srcs = [ "default_n_bit_quantize_registry_test.py", ], + strict_deps = False, deps = [ ":default_n_bit_quantize_registry", # absl/testing:parameterized dep1, @@ -92,6 +94,7 @@ py_library( srcs = [ "default_n_bit_transforms.py", ], + strict_deps = False, visibility = ["//visibility:public"], deps = [ # numpy dep1, @@ -111,7 +114,7 @@ py_library( srcs = [ "default_n_bit_quantize_layout_transform.py", ], - strict_deps = True, + strict_deps = False, deps = [ ":default_n_bit_transforms", # tensorflow dep1, @@ -149,7 +152,7 @@ py_library( srcs = [ "default_n_bit_quantize_scheme.py", ], - strict_deps = True, + strict_deps = False, visibility = ["//visibility:public"], deps = [ ":default_n_bit_quantize_layout_transform", diff --git a/tensorflow_model_optimization/python/core/quantization/keras/graph_transformations/BUILD b/tensorflow_model_optimization/python/core/quantization/keras/graph_transformations/BUILD index af02209a..898e61cc 100644 --- a/tensorflow_model_optimization/python/core/quantization/keras/graph_transformations/BUILD +++ b/tensorflow_model_optimization/python/core/quantization/keras/graph_transformations/BUILD @@ -12,7 +12,7 @@ py_library( srcs = [ "__init__.py", ], - strict_deps = True, + strict_deps = False, deps = [], ) @@ -21,7 +21,7 @@ py_library( srcs = [ "transforms.py", ], - strict_deps = True, + strict_deps = False, visibility = ["//visibility:public"], deps = [ # six dep1, @@ -46,7 +46,7 @@ py_library( srcs = [ "model_transformer.py", ], - strict_deps = True, + strict_deps = False, visibility = ["//visibility:public"], deps = [ ":transforms", diff --git a/tensorflow_model_optimization/python/core/quantization/keras/layers/BUILD b/tensorflow_model_optimization/python/core/quantization/keras/layers/BUILD index 31360ab2..b7cdacf8 100644 --- a/tensorflow_model_optimization/python/core/quantization/keras/layers/BUILD +++ b/tensorflow_model_optimization/python/core/quantization/keras/layers/BUILD @@ -11,14 +11,14 @@ py_library( srcs = [ "__init__.py", ], - strict_deps = True, + strict_deps = False, deps = [], ) py_library( name = "conv_batchnorm_test_utils", srcs = ["conv_batchnorm_test_utils.py"], - strict_deps = True, + strict_deps = False, deps = [ # tensorflow dep1, "//tensorflow_model_optimization/python/core/keras:compat", @@ -28,7 +28,7 @@ py_library( py_library( name = "dense_batchnorm_test_utils", srcs = ["dense_batchnorm_test_utils.py"], - strict_deps = True, + strict_deps = False, deps = [ # tensorflow dep1, "//tensorflow_model_optimization/python/core/keras:compat", diff --git a/tensorflow_model_optimization/python/core/sparsity/BUILD b/tensorflow_model_optimization/python/core/sparsity/BUILD index 36f28dcc..5c0195a4 100644 --- a/tensorflow_model_optimization/python/core/sparsity/BUILD +++ b/tensorflow_model_optimization/python/core/sparsity/BUILD @@ -7,7 +7,7 @@ licenses(["notice"]) py_library( name = "sparsity", srcs = ["__init__.py"], - strict_deps = True, + strict_deps = False, deps = [ "//tensorflow_model_optimization/python/core/sparsity/keras", # buildcleaner: keep ], diff --git a/tensorflow_model_optimization/python/core/sparsity/keras/BUILD b/tensorflow_model_optimization/python/core/sparsity/keras/BUILD index c6d79c4a..8959d15a 100644 --- a/tensorflow_model_optimization/python/core/sparsity/keras/BUILD +++ b/tensorflow_model_optimization/python/core/sparsity/keras/BUILD @@ -13,7 +13,7 @@ py_library( srcs = [ "__init__.py", ], - strict_deps = True, + strict_deps = False, deps = [ ":prunable_layer", # buildcleaner: keep ":prune", # buildcleaner: keep @@ -26,7 +26,7 @@ py_library( py_library( name = "prune", srcs = ["prune.py"], - strict_deps = True, + strict_deps = False, visibility = ["//visibility:public"], deps = [ ":pruning_schedule", @@ -40,7 +40,7 @@ py_library( py_library( name = "prunable_layer", srcs = ["prunable_layer.py"], - strict_deps = True, + strict_deps = False, visibility = ["//visibility:public"], deps = [ # six dep1, @@ -50,7 +50,7 @@ py_library( py_library( name = "prune_registry", srcs = ["prune_registry.py"], - strict_deps = True, + strict_deps = False, visibility = ["//visibility:public"], deps = [ ":prunable_layer", @@ -62,7 +62,7 @@ py_library( py_library( name = "pruning_policy", srcs = ["pruning_policy.py"], - strict_deps = True, + strict_deps = False, visibility = ["//visibility:public"], deps = [ ":pruning_wrapper", @@ -75,7 +75,7 @@ py_library( py_library( name = "pruning_schedule", srcs = ["pruning_schedule.py"], - strict_deps = True, + strict_deps = False, visibility = ["//visibility:public"], deps = [ # six dep1, @@ -101,7 +101,7 @@ py_strict_test( py_library( name = "pruning_wrapper", srcs = ["pruning_wrapper.py"], - strict_deps = True, + strict_deps = False, visibility = ["//visibility:public"], deps = [ ":prunable_layer", @@ -120,7 +120,7 @@ py_library( py_library( name = "pruning_callbacks", srcs = ["pruning_callbacks.py"], - strict_deps = True, + strict_deps = False, visibility = ["//visibility:public"], deps = [ ":pruning_wrapper", @@ -151,7 +151,7 @@ py_strict_test( py_library( name = "pruning_impl", srcs = ["pruning_impl.py"], - strict_deps = True, + strict_deps = False, visibility = ["//visibility:public"], deps = [ ":pruning_utils", @@ -163,7 +163,7 @@ py_library( py_library( name = "pruning_utils", srcs = ["pruning_utils.py"], - strict_deps = True, + strict_deps = False, visibility = ["//visibility:public"], deps = [ # numpy dep1, @@ -174,7 +174,7 @@ py_library( py_library( name = "estimator_utils", srcs = ["estimator_utils.py"], - strict_deps = True, + strict_deps = False, visibility = ["//visibility:public"], deps = [ ":pruning_wrapper", @@ -188,7 +188,7 @@ py_library( name = "test_utils", testonly = 1, srcs = ["test_utils.py"], - strict_deps = True, + strict_deps = False, visibility = ["//visibility:public"], deps = [ ":prune", @@ -325,6 +325,7 @@ py_test( name = "pruning_policy_test", size = "medium", srcs = ["pruning_policy_test.py"], + strict_deps = False, visibility = ["//visibility:public"], deps = [ ":prune", diff --git a/tensorflow_model_optimization/python/core/sparsity/keras/tools/BUILD b/tensorflow_model_optimization/python/core/sparsity/keras/tools/BUILD index 0414f47e..63937d70 100644 --- a/tensorflow_model_optimization/python/core/sparsity/keras/tools/BUILD +++ b/tensorflow_model_optimization/python/core/sparsity/keras/tools/BUILD @@ -11,7 +11,7 @@ licenses(["notice"]) py_library( name = "sparsity_tooling", srcs = ["sparsity_tooling.py"], - strict_deps = True, + strict_deps = False, visibility = ["//visibility:public"], deps = [ # tensorflow dep1, @@ -27,6 +27,7 @@ py_test( name = "sparsity_tooling_test", size = "medium", srcs = ["sparsity_tooling_test.py"], + strict_deps = False, visibility = ["//visibility:public"], deps = [ ":sparsity_tooling", @@ -53,7 +54,7 @@ py_binary( py_library( name = "check_sparsity_m_by_n", srcs = ["check_sparsity_m_by_n.py"], - strict_deps = True, + strict_deps = False, deps = [ # absl:app dep1, # absl/flags dep1, diff --git a/tensorflow_model_optimization/python/examples/clustering/keras/imdb/BUILD b/tensorflow_model_optimization/python/examples/clustering/keras/imdb/BUILD index 6a204573..a340c3ce 100644 --- a/tensorflow_model_optimization/python/examples/clustering/keras/imdb/BUILD +++ b/tensorflow_model_optimization/python/examples/clustering/keras/imdb/BUILD @@ -59,7 +59,7 @@ py_library( srcs = [ "imdb_utils.py", ], - strict_deps = True, + strict_deps = False, deps = [ # tensorflow dep1, "//tensorflow_model_optimization/python/core/clustering/keras:cluster",