Internal Code Change

PiperOrigin-RevId: 578569429
tensorflow · Nov 1, 2023 · 91ce7b4 · 91ce7b4
1 parent 39c8a8c
commit 91ce7b4
Show file tree

Hide file tree

Showing 11 changed files with 121 additions and 22 deletions.
diff --git a/tensorflow_privacy/privacy/analysis/BUILD b/tensorflow_privacy/privacy/analysis/BUILD
@@ -27,7 +27,10 @@ py_test(
     srcs = ["compute_dp_sgd_privacy_test.py"],
     python_version = "PY3",
     srcs_version = "PY3",
-    deps = [":compute_dp_sgd_privacy_lib"],
+    deps = [
+        ":compute_dp_sgd_privacy_lib",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
+    ],
 )
 
 py_binary(
@@ -44,7 +47,10 @@ py_library(
 py_test(
     name = "compute_noise_from_budget_test",
     srcs = ["compute_noise_from_budget_test.py"],
-    deps = [":compute_noise_from_budget_lib"],
+    deps = [
+        ":compute_noise_from_budget_lib",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
+    ],
 )
 
 py_library(
@@ -64,7 +70,10 @@ py_test(
     srcs = ["tensor_buffer_eager_test.py"],
     python_version = "PY3",
     srcs_version = "PY3",
-    deps = [":tensor_buffer"],
+    deps = [
+        ":tensor_buffer",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
+    ],
 )
 
 py_test(
@@ -73,7 +82,10 @@ py_test(
     srcs = ["tensor_buffer_graph_test.py"],
     python_version = "PY3",
     srcs_version = "PY3",
-    deps = [":tensor_buffer"],
+    deps = [
+        ":tensor_buffer",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
+    ],
 )
 
 py_library(
@@ -87,5 +99,8 @@ py_test(
     srcs = ["tree_aggregation_accountant_test.py"],
     python_version = "PY3",
     srcs_version = "PY3",
-    deps = [":tree_aggregation_accountant"],
+    deps = [
+        ":tree_aggregation_accountant",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
+    ],
 )
diff --git a/tensorflow_privacy/privacy/dp_query/BUILD b/tensorflow_privacy/privacy/dp_query/BUILD
@@ -18,7 +18,10 @@ py_test(
     srcs = ["dp_query_test.py"],
     python_version = "PY3",
     srcs_version = "PY3",
-    deps = [":no_privacy_query"],
+    deps = [
+        ":no_privacy_query",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
+    ],
 )
 
 py_library(
@@ -32,7 +35,10 @@ py_test(
     srcs = ["discrete_gaussian_utils_test.py"],
     python_version = "PY3",
     srcs_version = "PY3",
-    deps = [":discrete_gaussian_utils"],
+    deps = [
+        ":discrete_gaussian_utils",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
+    ],
 )
 
 py_library(
@@ -54,6 +60,7 @@ py_test(
         ":discrete_gaussian_query",
         ":discrete_gaussian_utils",
         ":test_utils",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -76,6 +83,7 @@ py_test(
         ":discrete_gaussian_utils",
         ":distributed_discrete_gaussian_query",
         ":test_utils",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -97,6 +105,7 @@ py_test(
     deps = [
         ":distributed_skellam_query",
         ":test_utils",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -116,6 +125,7 @@ py_test(
     deps = [
         ":gaussian_query",
         ":test_utils",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -135,6 +145,7 @@ py_test(
     deps = [
         ":no_privacy_query",
         ":test_utils",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -155,6 +166,7 @@ py_test(
         ":gaussian_query",
         ":normalized_query",
         ":test_utils",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -177,6 +189,7 @@ py_test(
         ":nested_query",
         ":normalized_query",
         ":test_utils",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -199,6 +212,7 @@ py_test(
     deps = [
         ":quantile_adaptive_clip_sum_query",
         ":test_utils",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -223,6 +237,7 @@ py_test(
     deps = [
         ":quantile_estimator_query",
         ":test_utils",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -244,7 +259,10 @@ py_test(
     python_version = "PY3",
     shard_count = 10,
     srcs_version = "PY3",
-    deps = [":tree_aggregation"],
+    deps = [
+        ":tree_aggregation",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
+    ],
 )
 
 py_test(
@@ -257,6 +275,7 @@ py_test(
         ":test_utils",
         ":tree_aggregation",
         ":tree_aggregation_query",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -286,7 +305,10 @@ py_test(
     srcs = ["tree_range_query_test.py"],
     python_version = "PY3",
     srcs_version = "PY3",
-    deps = [":tree_range_query"],
+    deps = [
+        ":tree_range_query",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
+    ],
 )
 
 py_test(
@@ -297,6 +319,7 @@ py_test(
     deps = [
         ":restart_query",
         ":tree_aggregation_query",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -328,5 +351,6 @@ py_test(
     deps = [
         ":quantile_adaptive_clip_tree_query",
         ":test_utils",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
diff --git a/tensorflow_privacy/privacy/estimators/BUILD b/tensorflow_privacy/privacy/estimators/BUILD
@@ -72,6 +72,7 @@ py_test(
         ":binary_class_head",
         ":test_utils",
         "//tensorflow_privacy/privacy/optimizers:dp_optimizer_keras",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -85,6 +86,7 @@ py_test(
         ":multi_class_head",
         ":test_utils",
         "//tensorflow_privacy/privacy/optimizers:dp_optimizer_keras",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -98,6 +100,7 @@ py_test(
         ":multi_label_head",
         ":test_utils",
         "//tensorflow_privacy/privacy/optimizers:dp_optimizer_keras",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -111,5 +114,6 @@ py_test(
         ":dnn",
         ":test_utils",
         "//tensorflow_privacy/privacy/optimizers:dp_optimizer_keras",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
diff --git a/tensorflow_privacy/privacy/estimators/v1/BUILD b/tensorflow_privacy/privacy/estimators/v1/BUILD
@@ -43,6 +43,7 @@ py_test(
         ":head",
         "//tensorflow_privacy/privacy/estimators:test_utils",
         "//tensorflow_privacy/privacy/optimizers:dp_optimizer",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -56,6 +57,7 @@ py_test(
         ":dnn",
         "//tensorflow_privacy/privacy/estimators:test_utils",
         "//tensorflow_privacy/privacy/optimizers:dp_optimizer",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -69,5 +71,6 @@ py_test(
         ":linear",
         "//tensorflow_privacy/privacy/estimators:test_utils",
         "//tensorflow_privacy/privacy/optimizers:dp_optimizer",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
diff --git a/tensorflow_privacy/privacy/fast_gradient_clipping/BUILD b/tensorflow_privacy/privacy/fast_gradient_clipping/BUILD
@@ -42,6 +42,7 @@ py_test(
     deps = [
         ":gradient_clipping_utils",
         ":type_aliases",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -78,5 +79,6 @@ py_test(
         ":common_test_utils",
         ":layer_registry",
         ":type_aliases",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
diff --git a/tensorflow_privacy/privacy/keras_models/BUILD b/tensorflow_privacy/privacy/keras_models/BUILD
@@ -29,6 +29,7 @@ py_test(
     deps = [
         "//tensorflow_privacy/privacy/fast_gradient_clipping:layer_registry",
         "//tensorflow_privacy/privacy/keras_models:dp_keras_model",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -38,5 +39,6 @@ py_test(
     deps = [
         ":dp_keras_model",
         "//tensorflow_privacy/privacy/fast_gradient_clipping:layer_registry",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
diff --git a/tensorflow_privacy/privacy/logistic_regression/BUILD b/tensorflow_privacy/privacy/logistic_regression/BUILD
@@ -29,6 +29,7 @@ py_test(
         ":datasets",
         ":multinomial_logistic",
         "//tensorflow_privacy/privacy/analysis:compute_dp_sgd_privacy_lib",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -45,7 +46,10 @@ py_test(
     python_version = "PY3",
     srcs_version = "PY3",
     tags = ["requires-net:external"],
-    deps = [":datasets"],
+    deps = [
+        ":datasets",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
+    ],
 )
 
 py_library(
@@ -64,5 +68,6 @@ py_test(
     deps = [
         ":datasets",
         ":single_layer_softmax",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
diff --git a/tensorflow_privacy/privacy/optimizers/BUILD b/tensorflow_privacy/privacy/optimizers/BUILD
@@ -83,7 +83,10 @@ py_test(
     srcs = ["clip_and_aggregate_gradients_test.py"],
     python_version = "PY3",
     srcs_version = "PY3",
-    deps = [":clip_and_aggregate_gradients"],
+    deps = [
+        ":clip_and_aggregate_gradients",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
+    ],
 )
 
 py_test(
@@ -95,6 +98,7 @@ py_test(
     deps = [
         ":dp_optimizer",
         "//tensorflow_privacy/privacy/dp_query:gaussian_query",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -104,7 +108,10 @@ py_test(
     srcs = ["dp_optimizer_keras_sparse_test.py"],
     python_version = "PY3",
     srcs_version = "PY3",
-    deps = [":dp_optimizer_keras_sparse"],
+    deps = [
+        ":dp_optimizer_keras_sparse",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
+    ],
 )
 
 py_test(
@@ -113,7 +120,10 @@ py_test(
     srcs = ["dp_optimizer_keras_sparse_distributed_test.py"],
     python_version = "PY3",
     srcs_version = "PY3",
-    deps = [":dp_optimizer_keras_sparse"],
+    deps = [
+        ":dp_optimizer_keras_sparse",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
+    ],
 )
 
 py_test(
@@ -122,7 +132,10 @@ py_test(
     srcs = ["dp_optimizer_vectorized_test.py"],
     python_version = "PY3",
     srcs_version = "PY3",
-    deps = [":dp_optimizer_vectorized"],
+    deps = [
+        ":dp_optimizer_vectorized",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
+    ],
 )
 
 py_test(
@@ -134,6 +147,7 @@ py_test(
     deps = [
         ":dp_optimizer",
         "//tensorflow_privacy/privacy/dp_query:gaussian_query",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )
 
@@ -146,5 +160,6 @@ py_test(
     deps = [
         ":dp_optimizer_keras",
         ":dp_optimizer_keras_vectorized",
+        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
     ],
 )