From 5ea6f5d8fed993f91d118aef577710a7343379c7 Mon Sep 17 00:00:00 2001
From: Yong Tang <yong.tang.github@outlook.com>
Date: Wed, 10 Apr 2024 01:48:33 +0000
Subject: [PATCH] Change build to tf 2.16.1

---
 .bazelversion                           |  2 +-
 WORKSPACE                               | 11 +++++------
 tensorflow_io/python/ops/version_ops.py |  4 ++--
 tools/docker/build.Dockerfile           |  2 +-
 4 files changed, 9 insertions(+), 10 deletions(-)

diff --git a/.bazelversion b/.bazelversion
index dfda3e0b4..f22d756da 100644
--- a/.bazelversion
+++ b/.bazelversion
@@ -1 +1 @@
-6.1.0
+6.5.0
diff --git a/WORKSPACE b/WORKSPACE
index e29c56d4a..41e624f60 100644
--- a/WORKSPACE
+++ b/WORKSPACE
@@ -93,10 +93,10 @@ switched_rules_by_language(
 
 http_archive(
     name = "org_tensorflow",
-    sha256 = "9cec5acb0ecf2d47b16891f8bc5bc6fbfdffe1700bdadc0d9ebe27ea34f0c220",
-    strip_prefix = "tensorflow-2.15.0",
+    sha256 = "c729e56efc945c6df08efe5c9f5b8b89329c7c91b8f40ad2bb3e13900bd4876d",
+    strip_prefix = "tensorflow-2.16.1",
     urls = [
-        "https://github.com/tensorflow/tensorflow/archive/refs/tags/v2.15.0.tar.gz",
+        "https://github.com/tensorflow/tensorflow/archive/refs/tags/v2.16.1.tar.gz",
     ],
 )
 
@@ -880,10 +880,9 @@ http_archive(
 # Needed for llvm_toolchain and Golang
 http_archive(
     name = "com_grail_bazel_toolchain",
-    sha256 = "9e6065ded4b7453143e1586d6819729a63cd233114b72bf85ff3435367b02c90",
-    strip_prefix = "bazel-toolchain-edd07e96a2ecaa131af9234d6582875d980c0ac7",
+    sha256 = "4b7999c1fa2c3117bb21651e3c155b152e44ae67b2c311214883d4707dbe183f",
+    strip_prefix = "toolchains_llvm-edd07e96a2ecaa131af9234d6582875d980c0ac7",
     urls = [
-        "https://storage.googleapis.com/mirror.tensorflow.org/github.com/grailbio/bazel-toolchain/archive/edd07e96a2ecaa131af9234d6582875d980c0ac7.tar.gz",
         "https://github.com/grailbio/bazel-toolchain/archive/edd07e96a2ecaa131af9234d6582875d980c0ac7.tar.gz",
     ],
 )
diff --git a/tensorflow_io/python/ops/version_ops.py b/tensorflow_io/python/ops/version_ops.py
index 3daebf24d..aa7b04f03 100644
--- a/tensorflow_io/python/ops/version_ops.py
+++ b/tensorflow_io/python/ops/version_ops.py
@@ -14,5 +14,5 @@
 # ==============================================================================
 """version_ops"""
 
-version = "0.36.0"
-require = "tensorflow>=2.15.0,<2.16.0"
+version = "0.37.0"
+require = "tensorflow>=2.16.0,<2.17.0"
diff --git a/tools/docker/build.Dockerfile b/tools/docker/build.Dockerfile
index cbce6b9fe..189a8f2a6 100644
--- a/tools/docker/build.Dockerfile
+++ b/tools/docker/build.Dockerfile
@@ -9,7 +9,7 @@ ARG BAZEL_OPTIMIZATION
 ADD . /opt/io
 WORKDIR /opt/io
 
-RUN python${PYTHON_VERSION} -m pip install tensorflow==${TENSORFLOW_VERSION}
+RUN python${PYTHON_VERSION} -m pip install tensorflow>=${TENSORFLOW_VERSION}
 
 RUN python$PYTHON_VERSION -m pip uninstall -y tensorflow-io-gcs-filesystem