Skip to content

Commit f19b4e6

Browse files
Internal Change
PiperOrigin-RevId: 413924111
1 parent 91d8b27 commit f19b4e6

File tree

2 files changed

+7
-0
lines changed
  • tensorflow_model_optimization/python/core
    • clustering/keras
    • common/keras/compression/algorithms

2 files changed

+7
-0
lines changed

tensorflow_model_optimization/python/core/clustering/keras/BUILD

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -211,6 +211,7 @@ py_strict_test(
211211
name = "mnist_clusterable_layer_test",
212212
srcs = ["mnist_clusterable_layer_test.py"],
213213
python_version = "PY3",
214+
tags = ["requires-net:external"],
214215
visibility = ["//visibility:public"],
215216
deps = [
216217
":cluster",
@@ -225,6 +226,7 @@ py_strict_test(
225226
name = "mnist_clustering_test",
226227
srcs = ["mnist_clustering_test.py"],
227228
python_version = "PY3",
229+
tags = ["requires-net:external"],
228230
visibility = ["//visibility:public"],
229231
deps = [
230232
":cluster",

tensorflow_model_optimization/python/core/common/keras/compression/algorithms/BUILD

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ py_strict_test(
2020
timeout = "long",
2121
srcs = ["same_training_and_inference_test.py"],
2222
python_version = "PY3",
23+
tags = ["requires-net:external"],
2324
deps = [
2425
":same_training_and_inference",
2526
# tensorflow dep1,
@@ -42,6 +43,7 @@ py_strict_test(
4243
timeout = "long",
4344
srcs = ["different_training_and_inference_test.py"],
4445
python_version = "PY3",
46+
tags = ["requires-net:external"],
4547
deps = [
4648
":different_training_and_inference",
4749
# tensorflow dep1,
@@ -64,6 +66,7 @@ py_strict_test(
6466
timeout = "long",
6567
srcs = ["bias_only_test.py"],
6668
python_version = "PY3",
69+
tags = ["requires-net:external"],
6770
deps = [
6871
":bias_only",
6972
# tensorflow dep1,
@@ -88,6 +91,7 @@ py_strict_test(
8891
name = "weight_clustering_test",
8992
srcs = ["weight_clustering_test.py"],
9093
python_version = "PY3",
94+
tags = ["requires-net:external"],
9195
deps = [
9296
":weight_clustering",
9397
# tensorflow dep1,
@@ -110,6 +114,7 @@ py_strict_test(
110114
timeout = "long",
111115
srcs = ["periodical_update_and_scheduling_test.py"],
112116
python_version = "PY3",
117+
tags = ["requires-net:external"],
113118
deps = [
114119
":periodical_update_and_scheduling",
115120
# tensorflow dep1,

0 commit comments

Comments
 (0)