Update tensorflow to version 2.16.1.

PiperOrigin-RevId: 657740192

RELEASE.md

@@ -8,6 +8,10 @@ and this project adheres to
 
 ## Unreleased
 
+### Changed
+
+*   `tensorflow` verison increased to 2.16.1.
+
 ## Release 0.84.0
 
 ### Added

WORKSPACE

@@ -18,64 +18,64 @@ load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
 
 http_archive(
     name = "bazel_skylib",
-    url = "https://github.com/bazelbuild/bazel-skylib/archive/refs/tags/1.3.0.tar.gz",
     sha256 = "3b620033ca48fcd6f5ef2ac85e0f6ec5639605fa2f627968490e52fc91a9932f",
     strip_prefix = "bazel-skylib-1.3.0",
+    url = "https://github.com/bazelbuild/bazel-skylib/archive/refs/tags/1.3.0.tar.gz",
 )
 
 http_archive(
     name = "com_github_grpc_grpc",
-    url = "https://github.com/grpc/grpc/archive/refs/tags/v1.50.0.tar.gz",
-    sha256 = "76900ab068da86378395a8e125b5cc43dfae671e09ff6462ddfef18676e2165a",
-    strip_prefix = "grpc-1.50.0",
+    sha256 = "c682fc39baefc6e804d735e6b48141157b7213602cc66dbe0bf375b904d8b5f9",
+    strip_prefix = "grpc-1.64.2",
+    url = "https://github.com/grpc/grpc/archive/refs/tags/v1.64.2.tar.gz",
 )
 
 http_archive(
     name = "com_google_benchmark",
-    url = "https://github.com/google/benchmark/archive/refs/tags/v1.8.3.tar.gz",
     sha256 = "6bc180a57d23d4d9515519f92b0c83d61b05b5bab188961f36ac7b06b0d9e9ce",
     strip_prefix = "benchmark-1.8.3",
+    url = "https://github.com/google/benchmark/archive/refs/tags/v1.8.3.tar.gz",
 )
 
 http_archive(
-  name = "com_google_cc_differential_privacy",
-  url = "https://github.com/google/differential-privacy/archive/refs/tags/v3.0.0.tar.gz",
-  sha256 = "6e6e1cd7a819695caae408f4fa938129ab7a86e83fe2410137c85e50131abbe0",
-  strip_prefix = "differential-privacy-3.0.0/cc",
+    name = "com_google_cc_differential_privacy",
+    sha256 = "6e6e1cd7a819695caae408f4fa938129ab7a86e83fe2410137c85e50131abbe0",
+    strip_prefix = "differential-privacy-3.0.0/cc",
+    url = "https://github.com/google/differential-privacy/archive/refs/tags/v3.0.0.tar.gz",
 )
 
 http_archive(
-  name = "com_google_differential_privacy",
-  url = "https://github.com/google/differential-privacy/archive/refs/tags/v3.0.0.tar.gz",
-  sha256 = "6e6e1cd7a819695caae408f4fa938129ab7a86e83fe2410137c85e50131abbe0",
-  strip_prefix = "differential-privacy-3.0.0",
+    name = "com_google_differential_privacy",
+    sha256 = "6e6e1cd7a819695caae408f4fa938129ab7a86e83fe2410137c85e50131abbe0",
+    strip_prefix = "differential-privacy-3.0.0",
+    url = "https://github.com/google/differential-privacy/archive/refs/tags/v3.0.0.tar.gz",
 )
 
 # This commit is determined by
 # https://github.com/tensorflow/tensorflow/blob/v2.16.1/third_party/absl/workspace.bzl#L10.
 http_archive(
     name = "com_google_absl",
-    url = "https://github.com/abseil/abseil-cpp/archive/fb3621f4f897824c0dbe0615fa94543df6192f30.tar.gz",
     sha256 = "0320586856674d16b0b7a4d4afb22151bdc798490bb7f295eddd8f6a62b46fea",
     strip_prefix = "abseil-cpp-fb3621f4f897824c0dbe0615fa94543df6192f30",
+    url = "https://github.com/abseil/abseil-cpp/archive/fb3621f4f897824c0dbe0615fa94543df6192f30.tar.gz",
 )
 
 http_archive(
     name = "com_google_googletest",
-    url = "https://github.com/google/googletest/archive/refs/tags/release-1.12.1.tar.gz",
     sha256 = "81964fe578e9bd7c94dfdb09c8e4d6e6759e19967e397dbea48d1c10e45d0df2",
     strip_prefix = "googletest-release-1.12.1",
+    url = "https://github.com/google/googletest/archive/refs/tags/release-1.12.1.tar.gz",
 )
 
 http_archive(
     name = "com_google_protobuf",
-    url = "https://github.com/protocolbuffers/protobuf/archive/refs/tags/v3.21.9.tar.gz",
-    sha256 = "1add10f9bd92775b91f326da259f243881e904dd509367d5031d4c782ba82810",
-    strip_prefix = "protobuf-3.21.9",
+    sha256 = "27ff8f361e7908b08ab93cb0c05562f6f92b51bca4f0c60638477264e1597125",
+    strip_prefix = "protobuf-26.1",
+    url = "https://github.com/protocolbuffers/protobuf/archive/refs/tags/v26.1.tar.gz",
 )
 
-# TODO: b/333391041 - Temporarily disable the direct dependency on
-# `eigen`, for now we pick this dependency up from the TensorFlow workspace.
+# TODO: b/333391041 - Temporarily disable the direct dependency on `eigen`, for
+# now we pick this dependency up from the TensorFlow workspace.
 # http_archive(
 #     name = "eigen",
 #     url = "https://gitlab.com/libeigen/eigen/archive/refs/tags/3.4.0.tar.gz",
@@ -91,76 +91,97 @@ http_archive(
 # https://github.com/google-parfait/tensorflow-federated/blob/main/requirements.txt.
 http_archive(
     name = "org_tensorflow",
-    url = "https://github.com/tensorflow/tensorflow/archive/refs/tags/v2.14.0.tar.gz",
     patches = [
         "//third_party/tensorflow:internal_visibility.patch",
         "//third_party/tensorflow:python_toolchain.patch",
         "//third_party/tensorflow:tf2xla_visibility.patch",
     ],
-    sha256 = "ce357fd0728f0d1b0831d1653f475591662ec5bca736a94ff789e6b1944df19f",
-    strip_prefix = "tensorflow-2.14.0",
+    sha256 = "c729e56efc945c6df08efe5c9f5b8b89329c7c91b8f40ad2bb3e13900bd4876d",
+    strip_prefix = "tensorflow-2.16.1",
+    url = "https://github.com/tensorflow/tensorflow/archive/refs/tags/v2.16.1.tar.gz",
 )
 
 # This commit is determined by
 # https://github.com/tensorflow/tensorflow/blob/v2.16.1/third_party/pybind11_abseil/workspace.bzl#L11.
 http_archive(
     name = "pybind11_abseil",
-    url = "https://github.com/pybind/pybind11_abseil/archive/2c4932ed6f6204f1656e245838f4f5eae69d2e29.tar.gz",
     sha256 = "0223b647b8cc817336a51e787980ebc299c8d5e64c069829bf34b69d72337449",
     strip_prefix = "pybind11_abseil-2c4932ed6f6204f1656e245838f4f5eae69d2e29",
+    url = "https://github.com/pybind/pybind11_abseil/archive/2c4932ed6f6204f1656e245838f4f5eae69d2e29.tar.gz",
 )
 
 http_archive(
     name = "pybind11_bazel",
-    url = "https://github.com/pybind/pybind11_bazel/archive/refs/tags/v2.11.1.tar.gz",
     sha256 = "e8355ee56c2ff772334b4bfa22be17c709e5573f6d1d561c7176312156c27bd4",
     strip_prefix = "pybind11_bazel-2.11.1",
+    url = "https://github.com/pybind/pybind11_bazel/archive/refs/tags/v2.11.1.tar.gz",
 )
 
 # This commit is determined by
 # https://github.com/tensorflow/tensorflow/blob/v2.16.1/tensorflow/workspace2.bzl#L788.
 http_archive(
     name = "pybind11_protobuf",
-    url = "https://github.com/pybind/pybind11_protobuf/archive/80f3440cd8fee124e077e2e47a8a17b78b451363.tar.gz",
     sha256 = "ba2c54a8b4d1dd0a68c58159e37b1f863c0d9d1dc815558288195493bcc31682",
     strip_prefix = "pybind11_protobuf-80f3440cd8fee124e077e2e47a8a17b78b451363",
+    url = "https://github.com/pybind/pybind11_protobuf/archive/80f3440cd8fee124e077e2e47a8a17b78b451363.tar.gz",
 )
 
 http_archive(
     name = "rules_license",
-    url = "https://github.com/bazelbuild/rules_license/archive/refs/tags/0.0.8.tar.gz",
     sha256 = "8c1155797cb5f5697ea8c6eac6c154cf51aa020e368813d9d9b949558c84f2da",
     strip_prefix = "rules_license-0.0.8",
+    url = "https://github.com/bazelbuild/rules_license/archive/refs/tags/0.0.8.tar.gz",
 )
 
 http_archive(
     name = "rules_python",
-    url = "https://github.com/bazelbuild/rules_python/archive/refs/tags/0.31.0.tar.gz",
     sha256 = "c68bdc4fbec25de5b5493b8819cfc877c4ea299c0dcb15c244c5a00208cde311",
     strip_prefix = "rules_python-0.31.0",
+    url = "https://github.com/bazelbuild/rules_python/archive/refs/tags/0.31.0.tar.gz",
 )
 
+# TODO: b/347940678 - Temporarily disable the direct dependency on `tsl`, for
+# now we pick this dependency up from the TensorFlow workspace.
+# http_archive(
+#     name = "tsl",
+#     url = "https://github.com/google/tsl/archive/5fdb99d897eadddd136c41156fbda82c6244d60a.tar.gz",
+#     sha256 = "414f9929d6a631aa225093e24e657e1a28c5e84da8e280f1636db65fe24e4b83",
+#     strip_prefix = "tsl-5fdb99d897eadddd136c41156fbda82c6244d60a",
+# )
+
+# TODO: b/347937714 - Temporarily disable the direct dependency on `xla`, for
+# now we pick this dependency up from the TensorFlow workspace.
+# http_archive(
+#     name = "xla",
+#     url = "https://github.com/openxla/xla/archive/940e3a27542b7ce76666173e7b287aa2a9263916.tar.gz",
+#     sha256 = "bcdc778e5a456839869dea796117b723bdea488075bd9555fe118fd8d6fcf25e",
+#     strip_prefix = "xla-940e3a27542b7ce76666173e7b287aa2a9263916",
+# )
+
 #
 # Inlined transitive dependencies, grouped by direct dependency.
 #
 
 # Required by pybind11_abseil and pybind11_protobuf.
 http_archive(
     name = "pybind11",
-    url = "https://github.com/pybind/pybind11/archive/refs/tags/v2.9.2.tar.gz",
+    build_file = "@pybind11_bazel//:pybind11.BUILD",
     sha256 = "6bd528c4dbe2276635dc787b6b1f2e5316cf6b49ee3e150264e455a0d68d19c1",
     strip_prefix = "pybind11-2.9.2",
-    build_file = "@pybind11_bazel//:pybind11.BUILD",
+    url = "https://github.com/pybind/pybind11/archive/refs/tags/v2.9.2.tar.gz",
 )
 
+# TODO: b/347937669 - Temporarily disable the direct dependency on
+# `com_github_grpc_grpc`, for now we pick this dependency up from the
+# TensorFlow workspace.
 # Required by com_github_grpc_grpc. This commit is determined by
 # https://github.com/grpc/grpc/blob/v1.50.0/bazel/grpc_deps.bzl#L344.
-http_archive(
-    name = "upb",
-    url = "https://github.com/protocolbuffers/upb/archive/e4635f223e7d36dfbea3b722a4ca4807a7e882e2.tar.gz",
-    sha256 = "017a7e8e4e842d01dba5dc8aa316323eee080cd1b75986a7d1f94d87220e6502",
-    strip_prefix = "upb-e4635f223e7d36dfbea3b722a4ca4807a7e882e2",
-)
+# http_archive(
+#     name = "upb",
+#     url = "https://github.com/protocolbuffers/upb/archive/e4635f223e7d36dfbea3b722a4ca4807a7e882e2.tar.gz",
+#     sha256 = "017a7e8e4e842d01dba5dc8aa316323eee080cd1b75986a7d1f94d87220e6502",
+#     strip_prefix = "upb-e4635f223e7d36dfbea3b722a4ca4807a7e882e2",
+# )
 
 #
 # Transitive dependencies, grouped by direct dependency.

pyproject.toml

@@ -46,7 +46,7 @@ dependencies = [
   'scipy~=1.9.3',
   'tensorflow-model-optimization==0.7.5',
   'tensorflow-privacy==0.9.0',
-  'tensorflow>=2.14.0,==2.14.*',
+  'tensorflow~=2.16.1',
   'tqdm~=4.64',
   'typing-extensions>=4.5.0,==4.5.*',
   # TODO: b/315515548 - Required because current dependencies are pulling in

requirements.txt

@@ -31,7 +31,7 @@ cachetools~=5.3
 dm-tree==0.1.8
 dp-accounting==0.4.3
 google-vizier==0.1.11
-grpcio~=1.46
+grpcio~=1.64
 jaxlib==0.4.14
 jax==0.4.14
 numpy~=1.25
@@ -44,7 +44,7 @@ typing-extensions>=4.5.0,==4.5.*
 
 # The version of this dependency should match the version in
 # https://github.com/google-parfait/tensorflow-federated/blob/main/WORKSPACE.
-tensorflow>=2.14.0,==2.14.*
+tensorflow~=2.16.1
 
 # TODO: b/315515548 - Required because current dependencies are pulling in later
 # versions which are causing opaque failures.

tensorflow_federated/cc/core/impl/aggregation/tensorflow/BUILD

@@ -29,10 +29,10 @@ cc_library(
         "@com_google_absl//absl/status",
         "@com_google_absl//absl/status:statusor",
         "@com_google_absl//absl/strings",
+        "@local_tsl//tsl/platform:refcount",
         "@org_tensorflow//tensorflow/core:framework",
         "@org_tensorflow//tensorflow/core:protos_all_cc",
         "@org_tensorflow//tensorflow/core/platform:tstring",
-        "@org_tensorflow//tensorflow/tsl/platform:refcount",
     ],
 )
 
@@ -130,8 +130,8 @@ cc_library(
         "@com_google_absl//absl/status:statusor",
         "@com_google_absl//absl/strings",
         "@com_google_absl//absl/strings:cord",
+        "@local_tsl//tsl/platform:env",
         "@org_tensorflow//tensorflow/core/platform:env",
-        "@org_tensorflow//tensorflow/tsl/platform:env",
     ],
 )
 
@@ -165,8 +165,8 @@ cc_library(
         "@com_google_absl//absl/status:statusor",
         "@com_google_absl//absl/strings",
         "@com_google_absl//absl/strings:cord",
+        "@local_tsl//tsl/platform:env",
         "@org_tensorflow//tensorflow/core/platform:env",
-        "@org_tensorflow//tensorflow/tsl/platform:env",
     ],
 )
 

tensorflow_federated/cc/core/impl/aggregation/tensorflow/converters.cc

@@ -29,7 +29,7 @@
 #include "tensorflow/core/framework/tensor_shape.h"
 #include "tensorflow/core/framework/tensor_shape.pb.h"
 #include "tensorflow/core/platform/tstring.h"
-#include "tensorflow/tsl/platform/refcount.h"
+#include "tsl/platform/refcount.h"
 #include "tensorflow_federated/cc/core/impl/aggregation/base/monitoring.h"
 #include "tensorflow_federated/cc/core/impl/aggregation/core/datatype.h"
 #include "tensorflow_federated/cc/core/impl/aggregation/core/tensor.h"

tensorflow_federated/cc/core/impl/aggregation/tensorflow/tensorflow_checkpoint_builder_factory.cc

@@ -29,7 +29,7 @@
 #include "absl/strings/str_cat.h"
 #include "absl/strings/string_view.h"
 #include "tensorflow/core/platform/file_system.h"
-#include "tensorflow/tsl/platform/env.h"
+#include "tsl/platform/env.h"
 #include "tensorflow_federated/cc/core/impl/aggregation/base/monitoring.h"
 #include "tensorflow_federated/cc/core/impl/aggregation/core/tensor.h"
 #include "tensorflow_federated/cc/core/impl/aggregation/protocol/checkpoint_builder.h"

tensorflow_federated/cc/core/impl/aggregation/tensorflow/tensorflow_checkpoint_parser_factory.cc

@@ -29,7 +29,7 @@
 #include "absl/strings/str_cat.h"
 #include "absl/strings/string_view.h"
 #include "tensorflow/core/platform/file_system.h"
-#include "tensorflow/tsl/platform/env.h"
+#include "tsl/platform/env.h"
 #include "tensorflow_federated/cc/core/impl/aggregation/base/monitoring.h"
 #include "tensorflow_federated/cc/core/impl/aggregation/core/tensor.h"
 #include "tensorflow_federated/cc/core/impl/aggregation/protocol/checkpoint_parser.h"