From 980ce181d9ab929f7efa47cbf7ccd61e5b8683da Mon Sep 17 00:00:00 2001 From: Hana Joo Date: Tue, 24 Feb 2026 08:05:31 -0800 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 874617915 --- .../python/core/clustering/keras/BUILD | 2 ++ .../quantization/keras/collab_opts/cluster_preserve/BUILD | 3 +++ .../core/quantization/keras/collab_opts/prune_preserve/BUILD | 1 + .../python/core/quantization/keras/default_8bit/BUILD | 4 ++++ .../core/quantization/keras/experimental/default_n_bit/BUILD | 3 +++ .../python/core/sparsity/keras/BUILD | 1 + .../python/core/sparsity/keras/tools/BUILD | 2 ++ .../python/examples/clustering/keras/imdb/BUILD | 4 ++++ 8 files changed, 20 insertions(+) diff --git a/tensorflow_model_optimization/python/core/clustering/keras/BUILD b/tensorflow_model_optimization/python/core/clustering/keras/BUILD index 252f047a..e5537a18 100644 --- a/tensorflow_model_optimization/python/core/clustering/keras/BUILD +++ b/tensorflow_model_optimization/python/core/clustering/keras/BUILD @@ -121,6 +121,7 @@ py_test( name = "cluster_test", size = "medium", srcs = ["cluster_test.py"], + strict_deps = False, visibility = ["//visibility:public"], deps = [ ":cluster", @@ -189,6 +190,7 @@ py_test( name = "cluster_integration_test", size = "medium", srcs = ["cluster_integration_test.py"], + strict_deps = False, visibility = ["//visibility:public"], deps = [ ":cluster", 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..8e1f107b 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 @@ -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 = [ @@ -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..8c6fc879 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 @@ -39,6 +39,7 @@ py_test( srcs = [ "prune_preserve_quantize_registry_test.py", ], + strict_deps = False, visibility = ["//visibility:private"], 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..ab421f14 100644 --- a/tensorflow_model_optimization/python/core/quantization/keras/default_8bit/BUILD +++ b/tensorflow_model_optimization/python/core/quantization/keras/default_8bit/BUILD @@ -36,6 +36,7 @@ py_test( srcs = [ "default_8bit_quantizers_test.py", ], + strict_deps = False, deps = [ ":default_8bit_quantizers", # absl/testing:parameterized dep1, @@ -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, @@ -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 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..1d705f9d 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 @@ -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, @@ -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, diff --git a/tensorflow_model_optimization/python/core/sparsity/keras/BUILD b/tensorflow_model_optimization/python/core/sparsity/keras/BUILD index c6d79c4a..e02f73a8 100644 --- a/tensorflow_model_optimization/python/core/sparsity/keras/BUILD +++ b/tensorflow_model_optimization/python/core/sparsity/keras/BUILD @@ -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..f0504eaf 100644 --- a/tensorflow_model_optimization/python/core/sparsity/keras/tools/BUILD +++ b/tensorflow_model_optimization/python/core/sparsity/keras/tools/BUILD @@ -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", @@ -41,6 +42,7 @@ py_test( py_binary( name = "evaluate_pruning", srcs = ["evaluate_pruning.py"], + strict_deps = False, deps = [ ":sparsity_tooling", # google/protobuf:use_fast_cpp_protos dep1, # Automatically added diff --git a/tensorflow_model_optimization/python/examples/clustering/keras/imdb/BUILD b/tensorflow_model_optimization/python/examples/clustering/keras/imdb/BUILD index 6a204573..c041a1e6 100644 --- a/tensorflow_model_optimization/python/examples/clustering/keras/imdb/BUILD +++ b/tensorflow_model_optimization/python/examples/clustering/keras/imdb/BUILD @@ -17,6 +17,7 @@ py_binary( srcs = [ "imdb_lstm.py", ], + strict_deps = False, deps = [ # google/protobuf:use_fast_cpp_protos dep1, # Automatically added # numpy dep1, @@ -31,6 +32,7 @@ py_binary( srcs = [ "imdb_rnn.py", ], + strict_deps = False, deps = [ # google/protobuf:use_fast_cpp_protos dep1, # Automatically added # numpy dep1, @@ -45,6 +47,7 @@ py_binary( srcs = [ "imdb_gru.py", ], + strict_deps = False, deps = [ # google/protobuf:use_fast_cpp_protos dep1, # Automatically added # numpy dep1, @@ -73,6 +76,7 @@ py_binary( srcs = [ "imdb_multiple_cells.py", ], + strict_deps = False, deps = [ ":imdb_utils", # google/protobuf:use_fast_cpp_protos dep1, # Automatically added