Internal change.

PiperOrigin-RevId: 455173202

Author: schien1729

g3doc/BUILD

@@ -27,7 +27,6 @@ py_library(
         "binary_class_head.py",
     ],
     srcs_version = "PY3",
-    deps = ["//third_party/tensorflow_estimator/python/estimator:binary_class_head"],
 )
 
 py_library(
@@ -36,7 +35,6 @@ py_library(
         "multi_class_head.py",
     ],
     srcs_version = "PY3",
-    deps = ["//third_party/tensorflow_estimator/python/estimator:multi_class_head"],
 )
 
 py_library(
@@ -45,7 +43,6 @@ py_library(
         "multi_label_head.py",
     ],
     srcs_version = "PY3",
-    deps = ["//third_party/tensorflow_estimator/python/estimator:multi_label_head"],
 )
 
 py_library(

tensorflow_privacy/privacy/estimators/BUILD

@@ -53,8 +53,6 @@ py_test(
     deps = [
         ":dp_optimizer",
         "//tensorflow_privacy/privacy/dp_query:gaussian_query",
-        "//third_party/py/tensorflow:tensorflow_compat_v1_estimator",
-        "//third_party/py/tensorflow:tensorflow_estimator",
     ],
 )
 
@@ -64,11 +62,7 @@ py_test(
     srcs = ["dp_optimizer_vectorized_test.py"],
     python_version = "PY3",
     srcs_version = "PY3",
-    deps = [
-        ":dp_optimizer_vectorized",
-        "//third_party/py/tensorflow:tensorflow_compat_v1_estimator",
-        "//third_party/py/tensorflow:tensorflow_estimator",
-    ],
+    deps = [":dp_optimizer_vectorized"],
 )
 
 py_test(
@@ -92,6 +86,5 @@ py_test(
     deps = [
         "//tensorflow_privacy/privacy/optimizers:dp_optimizer_keras",
         "//tensorflow_privacy/privacy/optimizers:dp_optimizer_keras_vectorized",
-        "//third_party/py/tensorflow:tensorflow_estimator",
     ],
 )

tensorflow_privacy/privacy/optimizers/BUILD

@@ -113,8 +113,6 @@ py_test(
     deps = [
         ":membership_inference_attack",
         ":tf_estimator_evaluation",
-        "//third_party/py/tensorflow:tensorflow_compat_v1_estimator",
-        "//third_party/py/tensorflow:tensorflow_estimator",
     ],
 )
 
@@ -126,7 +124,6 @@ py_library(
         ":membership_inference_attack",
         ":utils",
         ":utils_tensorboard",
-        "//third_party/py/tensorflow:tensorflow_estimator",
     ],
 )
 
@@ -138,8 +135,6 @@ py_binary(
     deps = [
         ":membership_inference_attack",
         ":tf_estimator_evaluation",
-        "//third_party/py/tensorflow:tensorflow_compat_v1_estimator",
-        "//third_party/py/tensorflow:tensorflow_estimator",
     ],
 )
 

tensorflow_privacy/privacy/privacy_tests/membership_inference_attack/BUILD

@@ -17,7 +17,6 @@ py_binary(
         ":mnist_dpsgd_tutorial_common",
         "//tensorflow_privacy/privacy/analysis:compute_dp_sgd_privacy_lib",
         "//tensorflow_privacy/privacy/optimizers:dp_optimizer",
-        "//third_party/py/tensorflow:tensorflow_estimator",
     ],
 )
 
@@ -64,8 +63,6 @@ py_binary(
     srcs_version = "PY3",
     deps = [
         "//tensorflow_privacy/privacy/optimizers:dp_optimizer_vectorized",
-        "//third_party/py/tensorflow:tensorflow_compat_v1_estimator",
-        "//third_party/py/tensorflow:tensorflow_estimator",
         "@com_google_differential_py//python/dp_accounting:dp_event",
         "@com_google_differential_py//python/dp_accounting/rdp:rdp_privacy_accountant",
     ],
@@ -80,7 +77,6 @@ py_binary(
         ":mnist_dpsgd_tutorial_common",
         "//tensorflow_privacy/privacy/analysis:compute_dp_sgd_privacy_lib",
         "//tensorflow_privacy/privacy/optimizers:dp_optimizer",
-        "//third_party/py/tensorflow:tensorflow_estimator",
     ],
 )
 
@@ -92,8 +88,6 @@ py_binary(
     deps = [
         "//tensorflow_privacy/privacy/analysis:rdp_accountant",
         "//tensorflow_privacy/privacy/optimizers:dp_optimizer",
-        "//third_party/py/tensorflow:tensorflow_compat_v1_estimator",
-        "//third_party/py/tensorflow:tensorflow_estimator",
         "@com_google_differential_py//python/dp_accounting:dp_event",
         "@com_google_differential_py//python/dp_accounting/rdp:rdp_privacy_accountant",
     ],
@@ -106,8 +100,6 @@ py_binary(
     srcs_version = "PY3",
     deps = [
         "//tensorflow_privacy/privacy/optimizers:dp_optimizer",
-        "//third_party/py/tensorflow:tensorflow_compat_v1_estimator",
-        "//third_party/py/tensorflow:tensorflow_estimator",
         "@com_google_differential_py//python/dp_accounting:dp_event",
         "@com_google_differential_py//python/dp_accounting/rdp:rdp_privacy_accountant",
     ],
@@ -119,8 +111,6 @@ py_binary(
     deps = [
         "//tensorflow_privacy/privacy/analysis:gdp_accountant",
         "//tensorflow_privacy/privacy/optimizers:dp_optimizer",
-        "//third_party/py/tensorflow:tensorflow_compat_v1_estimator",
-        "//third_party/py/tensorflow:tensorflow_estimator",
     ],
 )
 

tutorials/BUILD

@@ -7,8 +7,4 @@ py_binary(
     srcs = ["mnist_scratch.py"],
     python_version = "PY3",
     srcs_version = "PY3",
-    deps = [
-        "//third_party/py/tensorflow:tensorflow_compat_v1_estimator",
-        "//third_party/py/tensorflow:tensorflow_estimator",
-    ],
 )