Skip to content

Commit 91ce7b4

Browse files
anandoleetensorflower-gardener
anandolee
authored and
tensorflower-gardener
committed
Internal Code Change
PiperOrigin-RevId: 578569429
1 parent 39c8a8c commit 91ce7b4

File tree

11 files changed

+121
-22
lines changed

11 files changed

+121
-22
lines changed

‎tensorflow_privacy/privacy/analysis/BUILD

+20-5
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,10 @@ py_test(
2727
srcs = ["compute_dp_sgd_privacy_test.py"],
2828
python_version = "PY3",
2929
srcs_version = "PY3",
30-
deps = [":compute_dp_sgd_privacy_lib"],
30+
deps = [
31+
":compute_dp_sgd_privacy_lib",
32+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
33+
],
3134
)
3235

3336
py_binary(
@@ -44,7 +47,10 @@ py_library(
4447
py_test(
4548
name = "compute_noise_from_budget_test",
4649
srcs = ["compute_noise_from_budget_test.py"],
47-
deps = [":compute_noise_from_budget_lib"],
50+
deps = [
51+
":compute_noise_from_budget_lib",
52+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
53+
],
4854
)
4955

5056
py_library(
@@ -64,7 +70,10 @@ py_test(
6470
srcs = ["tensor_buffer_eager_test.py"],
6571
python_version = "PY3",
6672
srcs_version = "PY3",
67-
deps = [":tensor_buffer"],
73+
deps = [
74+
":tensor_buffer",
75+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
76+
],
6877
)
6978

7079
py_test(
@@ -73,7 +82,10 @@ py_test(
7382
srcs = ["tensor_buffer_graph_test.py"],
7483
python_version = "PY3",
7584
srcs_version = "PY3",
76-
deps = [":tensor_buffer"],
85+
deps = [
86+
":tensor_buffer",
87+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
88+
],
7789
)
7890

7991
py_library(
@@ -87,5 +99,8 @@ py_test(
8799
srcs = ["tree_aggregation_accountant_test.py"],
88100
python_version = "PY3",
89101
srcs_version = "PY3",
90-
deps = [":tree_aggregation_accountant"],
102+
deps = [
103+
":tree_aggregation_accountant",
104+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
105+
],
91106
)

‎tensorflow_privacy/privacy/dp_query/BUILD

+28-4
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,10 @@ py_test(
1818
srcs = ["dp_query_test.py"],
1919
python_version = "PY3",
2020
srcs_version = "PY3",
21-
deps = [":no_privacy_query"],
21+
deps = [
22+
":no_privacy_query",
23+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
24+
],
2225
)
2326

2427
py_library(
@@ -32,7 +35,10 @@ py_test(
3235
srcs = ["discrete_gaussian_utils_test.py"],
3336
python_version = "PY3",
3437
srcs_version = "PY3",
35-
deps = [":discrete_gaussian_utils"],
38+
deps = [
39+
":discrete_gaussian_utils",
40+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
41+
],
3642
)
3743

3844
py_library(
@@ -54,6 +60,7 @@ py_test(
5460
":discrete_gaussian_query",
5561
":discrete_gaussian_utils",
5662
":test_utils",
63+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
5764
],
5865
)
5966

@@ -76,6 +83,7 @@ py_test(
7683
":discrete_gaussian_utils",
7784
":distributed_discrete_gaussian_query",
7885
":test_utils",
86+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
7987
],
8088
)
8189

@@ -97,6 +105,7 @@ py_test(
97105
deps = [
98106
":distributed_skellam_query",
99107
":test_utils",
108+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
100109
],
101110
)
102111

@@ -116,6 +125,7 @@ py_test(
116125
deps = [
117126
":gaussian_query",
118127
":test_utils",
128+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
119129
],
120130
)
121131

@@ -135,6 +145,7 @@ py_test(
135145
deps = [
136146
":no_privacy_query",
137147
":test_utils",
148+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
138149
],
139150
)
140151

@@ -155,6 +166,7 @@ py_test(
155166
":gaussian_query",
156167
":normalized_query",
157168
":test_utils",
169+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
158170
],
159171
)
160172

@@ -177,6 +189,7 @@ py_test(
177189
":nested_query",
178190
":normalized_query",
179191
":test_utils",
192+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
180193
],
181194
)
182195

@@ -199,6 +212,7 @@ py_test(
199212
deps = [
200213
":quantile_adaptive_clip_sum_query",
201214
":test_utils",
215+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
202216
],
203217
)
204218

@@ -223,6 +237,7 @@ py_test(
223237
deps = [
224238
":quantile_estimator_query",
225239
":test_utils",
240+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
226241
],
227242
)
228243

@@ -244,7 +259,10 @@ py_test(
244259
python_version = "PY3",
245260
shard_count = 10,
246261
srcs_version = "PY3",
247-
deps = [":tree_aggregation"],
262+
deps = [
263+
":tree_aggregation",
264+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
265+
],
248266
)
249267

250268
py_test(
@@ -257,6 +275,7 @@ py_test(
257275
":test_utils",
258276
":tree_aggregation",
259277
":tree_aggregation_query",
278+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
260279
],
261280
)
262281

@@ -286,7 +305,10 @@ py_test(
286305
srcs = ["tree_range_query_test.py"],
287306
python_version = "PY3",
288307
srcs_version = "PY3",
289-
deps = [":tree_range_query"],
308+
deps = [
309+
":tree_range_query",
310+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
311+
],
290312
)
291313

292314
py_test(
@@ -297,6 +319,7 @@ py_test(
297319
deps = [
298320
":restart_query",
299321
":tree_aggregation_query",
322+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
300323
],
301324
)
302325

@@ -328,5 +351,6 @@ py_test(
328351
deps = [
329352
":quantile_adaptive_clip_tree_query",
330353
":test_utils",
354+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
331355
],
332356
)

‎tensorflow_privacy/privacy/estimators/BUILD

+4
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,7 @@ py_test(
7272
":binary_class_head",
7373
":test_utils",
7474
"//tensorflow_privacy/privacy/optimizers:dp_optimizer_keras",
75+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
7576
],
7677
)
7778

@@ -85,6 +86,7 @@ py_test(
8586
":multi_class_head",
8687
":test_utils",
8788
"//tensorflow_privacy/privacy/optimizers:dp_optimizer_keras",
89+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
8890
],
8991
)
9092

@@ -98,6 +100,7 @@ py_test(
98100
":multi_label_head",
99101
":test_utils",
100102
"//tensorflow_privacy/privacy/optimizers:dp_optimizer_keras",
103+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
101104
],
102105
)
103106

@@ -111,5 +114,6 @@ py_test(
111114
":dnn",
112115
":test_utils",
113116
"//tensorflow_privacy/privacy/optimizers:dp_optimizer_keras",
117+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
114118
],
115119
)

‎tensorflow_privacy/privacy/estimators/v1/BUILD

+3
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ py_test(
4343
":head",
4444
"//tensorflow_privacy/privacy/estimators:test_utils",
4545
"//tensorflow_privacy/privacy/optimizers:dp_optimizer",
46+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
4647
],
4748
)
4849

@@ -56,6 +57,7 @@ py_test(
5657
":dnn",
5758
"//tensorflow_privacy/privacy/estimators:test_utils",
5859
"//tensorflow_privacy/privacy/optimizers:dp_optimizer",
60+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
5961
],
6062
)
6163

@@ -69,5 +71,6 @@ py_test(
6971
":linear",
7072
"//tensorflow_privacy/privacy/estimators:test_utils",
7173
"//tensorflow_privacy/privacy/optimizers:dp_optimizer",
74+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
7275
],
7376
)

‎tensorflow_privacy/privacy/fast_gradient_clipping/BUILD

+2
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@ py_test(
4242
deps = [
4343
":gradient_clipping_utils",
4444
":type_aliases",
45+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
4546
],
4647
)
4748

@@ -78,5 +79,6 @@ py_test(
7879
":common_test_utils",
7980
":layer_registry",
8081
":type_aliases",
82+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
8183
],
8284
)

‎tensorflow_privacy/privacy/keras_models/BUILD

+2
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ py_test(
2929
deps = [
3030
"//tensorflow_privacy/privacy/fast_gradient_clipping:layer_registry",
3131
"//tensorflow_privacy/privacy/keras_models:dp_keras_model",
32+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
3233
],
3334
)
3435

@@ -38,5 +39,6 @@ py_test(
3839
deps = [
3940
":dp_keras_model",
4041
"//tensorflow_privacy/privacy/fast_gradient_clipping:layer_registry",
42+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
4143
],
4244
)

‎tensorflow_privacy/privacy/logistic_regression/BUILD

+6-1
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ py_test(
2929
":datasets",
3030
":multinomial_logistic",
3131
"//tensorflow_privacy/privacy/analysis:compute_dp_sgd_privacy_lib",
32+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
3233
],
3334
)
3435

@@ -45,7 +46,10 @@ py_test(
4546
python_version = "PY3",
4647
srcs_version = "PY3",
4748
tags = ["requires-net:external"],
48-
deps = [":datasets"],
49+
deps = [
50+
":datasets",
51+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
52+
],
4953
)
5054

5155
py_library(
@@ -64,5 +68,6 @@ py_test(
6468
deps = [
6569
":datasets",
6670
":single_layer_softmax",
71+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
6772
],
6873
)

‎tensorflow_privacy/privacy/optimizers/BUILD

+19-4
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,10 @@ py_test(
8383
srcs = ["clip_and_aggregate_gradients_test.py"],
8484
python_version = "PY3",
8585
srcs_version = "PY3",
86-
deps = [":clip_and_aggregate_gradients"],
86+
deps = [
87+
":clip_and_aggregate_gradients",
88+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
89+
],
8790
)
8891

8992
py_test(
@@ -95,6 +98,7 @@ py_test(
9598
deps = [
9699
":dp_optimizer",
97100
"//tensorflow_privacy/privacy/dp_query:gaussian_query",
101+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
98102
],
99103
)
100104

@@ -104,7 +108,10 @@ py_test(
104108
srcs = ["dp_optimizer_keras_sparse_test.py"],
105109
python_version = "PY3",
106110
srcs_version = "PY3",
107-
deps = [":dp_optimizer_keras_sparse"],
111+
deps = [
112+
":dp_optimizer_keras_sparse",
113+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
114+
],
108115
)
109116

110117
py_test(
@@ -113,7 +120,10 @@ py_test(
113120
srcs = ["dp_optimizer_keras_sparse_distributed_test.py"],
114121
python_version = "PY3",
115122
srcs_version = "PY3",
116-
deps = [":dp_optimizer_keras_sparse"],
123+
deps = [
124+
":dp_optimizer_keras_sparse",
125+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
126+
],
117127
)
118128

119129
py_test(
@@ -122,7 +132,10 @@ py_test(
122132
srcs = ["dp_optimizer_vectorized_test.py"],
123133
python_version = "PY3",
124134
srcs_version = "PY3",
125-
deps = [":dp_optimizer_vectorized"],
135+
deps = [
136+
":dp_optimizer_vectorized",
137+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
138+
],
126139
)
127140

128141
py_test(
@@ -134,6 +147,7 @@ py_test(
134147
deps = [
135148
":dp_optimizer",
136149
"//tensorflow_privacy/privacy/dp_query:gaussian_query",
150+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
137151
],
138152
)
139153

@@ -146,5 +160,6 @@ py_test(
146160
deps = [
147161
":dp_optimizer_keras",
148162
":dp_optimizer_keras_vectorized",
163+
"//third_party/py/google/protobuf:use_fast_cpp_protos", # Automatically added go/proto_python_upb_flip
149164
],
150165
)

0 commit comments

Comments
 (0)