diff --git a/tensorflow_model_optimization/python/core/clustering/keras/BUILD b/tensorflow_model_optimization/python/core/clustering/keras/BUILD index 5497fcf3..0fa162bf 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 @@ -140,7 +139,7 @@ py_test( ], ) -py_strict_test( +py_test( name = "clustering_centroids_test", size = "medium", srcs = ["clustering_centroids_test.py"], @@ -156,7 +155,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "cluster_wrapper_test", size = "medium", srcs = ["cluster_wrapper_test.py"], @@ -173,7 +172,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "clustering_registry_test", size = "medium", srcs = ["clustering_registry_test.py"], @@ -207,7 +206,7 @@ py_test( ], ) -py_strict_test( +py_test( name = "cluster_distributed_test", srcs = ["cluster_distributed_test.py"], visibility = ["//visibility:public"], @@ -225,7 +224,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "mnist_clusterable_layer_test", srcs = ["mnist_clusterable_layer_test.py"], tags = ["requires-net:external"], @@ -241,7 +240,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 0bfad37a..71237197 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_applicable_licenses = ["//tensorflow_model_optimization:license"], @@ -35,7 +35,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "core_encoder_test", size = "small", srcs = ["core_encoder_test.py"], @@ -59,7 +59,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "encoding_stage_test", size = "small", srcs = ["encoding_stage_test.py"], @@ -85,7 +85,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "gather_encoder_test", size = "small", srcs = ["gather_encoder_test.py"], @@ -113,7 +113,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 3f6b477c..e0323807 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_applicable_licenses = ["//tensorflow_model_optimization:license"], @@ -33,7 +33,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 5b00df8d..aa914448 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_applicable_licenses = ["//tensorflow_model_optimization:license"], @@ -33,7 +33,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 93fd0900..bfdd4673 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_applicable_licenses = ["//tensorflow_model_optimization:license"], @@ -33,7 +33,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "clipping_test", size = "small", srcs = ["clipping_test.py"], @@ -59,7 +59,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "kashin_test", size = "medium", srcs = ["kashin_test.py"], @@ -84,7 +84,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "quantization_test", size = "small", srcs = ["quantization_test.py"], @@ -108,7 +108,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 76b4c8d0..b1ac72f1 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_applicable_licenses = ["//tensorflow_model_optimization:license"], @@ -34,7 +34,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 0614d337..32637dc5 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_applicable_licenses = ["//tensorflow_model_optimization:license"], @@ -33,7 +33,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "py_utils_test", size = "small", srcs = ["py_utils_test.py"], @@ -56,7 +56,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 e94f46c0..9273eb0e 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_applicable_licenses = ["//tensorflow_model_optimization:license"], @@ -62,7 +62,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 32942e49..6ad39df7 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_applicable_licenses = ["//tensorflow_model_optimization:license"], @@ -39,7 +39,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "quant_ops_test", size = "small", srcs = ["quant_ops_test.py"], @@ -67,7 +67,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "quantizers_test", srcs = [ "quantizers_test.py", @@ -137,7 +137,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "quantize_annotate_test", srcs = [ "quantize_annotate_test.py", @@ -168,7 +168,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "quantize_aware_activation_test", srcs = [ "quantize_aware_activation_test.py", @@ -202,7 +202,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "quantize_layer_test", srcs = [ "quantize_layer_test.py", @@ -236,7 +236,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "quantize_wrapper_test", srcs = [ "quantize_wrapper_test.py", @@ -278,7 +278,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "quantize_test", srcs = ["quantize_test.py"], visibility = ["//visibility:public"], @@ -299,7 +299,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "quantize_integration_test", srcs = ["quantize_integration_test.py"], deps = [ @@ -316,7 +316,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "quantize_models_test", size = "enormous", srcs = ["quantize_models_test.py"], @@ -332,7 +332,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 3285e91b..61c14d19 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 @@ -116,7 +115,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 72dc8eca..f1bcdc50 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 @@ -127,7 +126,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 09a796a3..9be9cca5 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_applicable_licenses = ["//tensorflow_model_optimization:license"], @@ -31,7 +31,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "transforms_test", srcs = [ "transforms_test.py", @@ -58,7 +58,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 0baa55bc..22343370 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 @@ -86,7 +85,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "pruning_schedule_test", size = "medium", srcs = ["pruning_schedule_test.py"], @@ -134,7 +133,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "pruning_callbacks_test", size = "medium", srcs = ["pruning_callbacks_test.py"], @@ -203,7 +202,7 @@ py_library( ], ) -py_strict_test( +py_test( name = "prune_test", size = "medium", srcs = ["prune_test.py"], @@ -222,7 +221,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "prune_integration_test", size = "medium", timeout = "long", @@ -244,7 +243,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "prune_distributed_test", size = "medium", timeout = "long", @@ -264,7 +263,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "prune_registry_test", size = "medium", srcs = ["prune_registry_test.py"], @@ -279,7 +278,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "pruning_wrapper_test", size = "medium", srcs = ["pruning_wrapper_test.py"], @@ -293,7 +292,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "pruning_impl_test", size = "medium", srcs = ["pruning_impl_test.py"], @@ -310,7 +309,7 @@ py_strict_test( ], ) -py_strict_test( +py_test( name = "pruning_utils_test", size = "medium", srcs = ["pruning_utils_test.py"],