Add DP-enabled binary-class head and multi-class heads for Estimator.

PiperOrigin-RevId: 325921076 Change-Id: I958d492afb9b0d53300559d1880372b27400154e
2020-08-10 17:19:05 -07:00 · 2020-08-10 17:19:05 -07:00 · f621eebcff
commit f621eebcff
parent 956c8578c3
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/python/BUILD
+++ b/tensorflow/python/BUILD
@ -102,6 +102,7 @@ py_library(
        "//tensorflow/tools/api/tests:__pkg__",
        "//tensorflow/tools/compatibility/update:__pkg__",
        "//tensorflow_estimator:__subpackages__",
+        "//third_party/py/tensorflow_privacy:__subpackages__",  # TODO(b/163395075): remove when fixed
    ],
    deps = [
        ":layers",