From c4c36a0addd38f523482e6997fa52d9c781d9239 Mon Sep 17 00:00:00 2001 From: Hana Joo Date: Sun, 8 Feb 2026 03:34:29 -0800 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 867158637 --- .../python/core/clustering/keras/BUILD | 2 ++ .../core/quantization/keras/collab_opts/cluster_preserve/BUILD | 3 +++ .../core/quantization/keras/collab_opts/prune_preserve/BUILD | 1 + .../python/core/quantization/keras/default_8bit/BUILD | 3 +++ .../core/quantization/keras/experimental/default_n_bit/BUILD | 2 ++ tensorflow_model_optimization/python/core/sparsity/keras/BUILD | 1 + .../python/core/sparsity/keras/tools/BUILD | 1 + 7 files changed, 13 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..830270d5 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, @@ -154,6 +156,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..3727cb74 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, 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..a6cc8e83 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",