diff --git a/tensorflow_model_optimization/python/core/clustering/keras/BUILD b/tensorflow_model_optimization/python/core/clustering/keras/BUILD index e5537a18..6df51344 100644 --- a/tensorflow_model_optimization/python/core/clustering/keras/BUILD +++ b/tensorflow_model_optimization/python/core/clustering/keras/BUILD @@ -1,4 +1,3 @@ -load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test") # Placeholder: load py_library # Placeholder: load py_test @@ -137,7 +136,7 @@ py_test( ], ) -py_strict_test( +py_test( name = "clustering_centroids_test", size = "medium", srcs = ["clustering_centroids_test.py"], @@ -153,7 +152,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "cluster_wrapper_test", size = "medium", srcs = ["cluster_wrapper_test.py"], @@ -170,7 +169,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "clustering_registry_test", size = "medium", srcs = ["clustering_registry_test.py"], @@ -204,7 +203,7 @@ py_test( ], ) -py_strict_test( +py_test( name = "cluster_distributed_test", srcs = ["cluster_distributed_test.py"], visibility = ["//visibility:public"], @@ -222,7 +221,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "mnist_clusterable_layer_test", srcs = ["mnist_clusterable_layer_test.py"], tags = ["requires-net:external"], @@ -238,7 +237,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "mnist_clustering_test", srcs = ["mnist_clustering_test.py"], tags = ["requires-net:external"], 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 36dd5110..8353e0af 100644 --- a/tensorflow_model_optimization/python/core/internal/tensor_encoding/core/BUILD +++ b/tensorflow_model_optimization/python/core/internal/tensor_encoding/core/BUILD @@ -1,5 +1,5 @@ -load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test") # Placeholder: load py_library +# Placeholder: load py_test package(default_visibility = [ "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__", @@ -32,7 +32,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "core_encoder_test", size = "small", srcs = ["core_encoder_test.py"], @@ -56,7 +56,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "encoding_stage_test", size = "small", srcs = ["encoding_stage_test.py"], @@ -82,7 +82,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "gather_encoder_test", size = "small", srcs = ["gather_encoder_test.py"], @@ -110,7 +110,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "simple_encoder_test", size = "small", srcs = ["simple_encoder_test.py"], 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 47cedc23..19f5a24b 100644 --- a/tensorflow_model_optimization/python/core/internal/tensor_encoding/encoders/BUILD +++ b/tensorflow_model_optimization/python/core/internal/tensor_encoding/encoders/BUILD @@ -1,5 +1,5 @@ -load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test") # Placeholder: load py_library +# Placeholder: load py_test package(default_visibility = [ "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__", @@ -30,7 +30,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "common_encoders_test", size = "small", srcs = ["common_encoders_test.py"], 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 954113fc..92c714f3 100644 --- a/tensorflow_model_optimization/python/core/internal/tensor_encoding/stages/BUILD +++ b/tensorflow_model_optimization/python/core/internal/tensor_encoding/stages/BUILD @@ -1,5 +1,5 @@ -load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test") # Placeholder: load py_library +# Placeholder: load py_test package(default_visibility = [ "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__", @@ -30,7 +30,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "stages_impl_test", size = "medium", srcs = ["stages_impl_test.py"], 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 69530287..bbf4ab37 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,5 +1,5 @@ -load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test") # Placeholder: load py_library +# Placeholder: load py_test package(default_visibility = [ "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__", @@ -30,7 +30,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "clipping_test", size = "small", srcs = ["clipping_test.py"], @@ -56,7 +56,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "kashin_test", size = "medium", srcs = ["kashin_test.py"], @@ -81,7 +81,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "quantization_test", size = "small", srcs = ["quantization_test.py"], @@ -105,7 +105,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "misc_test", size = "small", srcs = ["misc_test.py"], 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 fd2c830f..99471ac4 100644 --- a/tensorflow_model_optimization/python/core/internal/tensor_encoding/testing/BUILD +++ b/tensorflow_model_optimization/python/core/internal/tensor_encoding/testing/BUILD @@ -1,5 +1,5 @@ -load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test") # Placeholder: load py_library +# Placeholder: load py_test package(default_visibility = [ "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__", @@ -31,7 +31,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "test_utils_test", size = "medium", srcs = ["test_utils_test.py"], 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 08fbbfd6..11c85dc8 100644 --- a/tensorflow_model_optimization/python/core/internal/tensor_encoding/utils/BUILD +++ b/tensorflow_model_optimization/python/core/internal/tensor_encoding/utils/BUILD @@ -1,5 +1,5 @@ -load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test") # Placeholder: load py_library +# Placeholder: load py_test package(default_visibility = [ "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__", @@ -30,7 +30,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "py_utils_test", size = "small", srcs = ["py_utils_test.py"], @@ -53,7 +53,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "tf_utils_test", size = "medium", srcs = ["tf_utils_test.py"], diff --git a/tensorflow_model_optimization/python/core/keras/BUILD b/tensorflow_model_optimization/python/core/keras/BUILD index faab2735..581d1e90 100644 --- a/tensorflow_model_optimization/python/core/keras/BUILD +++ b/tensorflow_model_optimization/python/core/keras/BUILD @@ -1,5 +1,5 @@ -load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test") # Placeholder: load py_library +# Placeholder: load py_test package(default_visibility = [ "//tensorflow_model_optimization:__subpackages__", @@ -59,7 +59,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "metrics_test", srcs = ["metrics_test.py"], deps = [ diff --git a/tensorflow_model_optimization/python/core/quantization/keras/BUILD b/tensorflow_model_optimization/python/core/quantization/keras/BUILD index c1a5adbc..09e0d21b 100644 --- a/tensorflow_model_optimization/python/core/quantization/keras/BUILD +++ b/tensorflow_model_optimization/python/core/quantization/keras/BUILD @@ -1,5 +1,5 @@ -load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test") # Placeholder: load py_library +# Placeholder: load py_test package(default_visibility = [ "//tensorflow_model_optimization:__subpackages__", @@ -36,7 +36,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "quant_ops_test", size = "small", srcs = ["quant_ops_test.py"], @@ -64,7 +64,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "quantizers_test", srcs = [ "quantizers_test.py", @@ -134,7 +134,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "quantize_annotate_test", srcs = [ "quantize_annotate_test.py", @@ -165,7 +165,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "quantize_aware_activation_test", srcs = [ "quantize_aware_activation_test.py", @@ -199,7 +199,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "quantize_layer_test", srcs = [ "quantize_layer_test.py", @@ -233,7 +233,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "quantize_wrapper_test", srcs = [ "quantize_wrapper_test.py", @@ -275,7 +275,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "quantize_test", srcs = ["quantize_test.py"], visibility = ["//visibility:public"], @@ -296,7 +296,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "quantize_integration_test", srcs = ["quantize_integration_test.py"], deps = [ @@ -313,7 +313,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "quantize_models_test", size = "enormous", srcs = ["quantize_models_test.py"], @@ -329,7 +329,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "quantize_functional_test", size = "large", srcs = ["quantize_functional_test.py"], 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 ab421f14..894d8c68 100644 --- a/tensorflow_model_optimization/python/core/quantization/keras/default_8bit/BUILD +++ b/tensorflow_model_optimization/python/core/quantization/keras/default_8bit/BUILD @@ -1,4 +1,3 @@ -load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test") # Placeholder: load py_library # Placeholder: load py_test @@ -113,7 +112,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "default_8bit_transforms_test", size = "large", srcs = [ 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 1d705f9d..47167654 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 @@ -1,4 +1,3 @@ -load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test") # Placeholder: load py_library # Placeholder: load py_test @@ -124,7 +123,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "default_n_bit_transforms_test", size = "large", srcs = [ 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..e5021fcc 100644 --- a/tensorflow_model_optimization/python/core/quantization/keras/graph_transformations/BUILD +++ b/tensorflow_model_optimization/python/core/quantization/keras/graph_transformations/BUILD @@ -1,5 +1,5 @@ -load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test") # Placeholder: load py_library +# Placeholder: load py_test package(default_visibility = [ "//tensorflow_model_optimization:__subpackages__", @@ -28,7 +28,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "transforms_test", srcs = [ "transforms_test.py", @@ -55,7 +55,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "model_transformer_test", srcs = [ "model_transformer_test.py", diff --git a/tensorflow_model_optimization/python/core/sparsity/keras/BUILD b/tensorflow_model_optimization/python/core/sparsity/keras/BUILD index e02f73a8..1dcd4ecd 100644 --- a/tensorflow_model_optimization/python/core/sparsity/keras/BUILD +++ b/tensorflow_model_optimization/python/core/sparsity/keras/BUILD @@ -1,4 +1,3 @@ -load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test") # Placeholder: load py_library # Placeholder: load py_test @@ -83,7 +82,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "pruning_schedule_test", size = "medium", srcs = ["pruning_schedule_test.py"], @@ -131,7 +130,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "pruning_callbacks_test", size = "medium", srcs = ["pruning_callbacks_test.py"], @@ -200,7 +199,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "prune_test", size = "medium", srcs = ["prune_test.py"], @@ -219,7 +218,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "prune_integration_test", size = "medium", timeout = "long", @@ -241,7 +240,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "prune_distributed_test", size = "medium", timeout = "long", @@ -261,7 +260,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "prune_registry_test", size = "medium", srcs = ["prune_registry_test.py"], @@ -276,7 +275,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "pruning_wrapper_test", size = "medium", srcs = ["pruning_wrapper_test.py"], @@ -290,7 +289,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "pruning_impl_test", size = "medium", srcs = ["pruning_impl_test.py"], @@ -307,7 +306,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "pruning_utils_test", size = "medium", srcs = ["pruning_utils_test.py"],