From 38ce6d027b090c247cc2755e0a185b2e030d7e00 Mon Sep 17 00:00:00 2001
From: Yash Katariya <yashkatariya@google.com>
Date: Wed, 4 May 2022 14:39:00 -0700
Subject: [PATCH] Update TF commit for release

PiperOrigin-RevId: 446555288
---
 WORKSPACE | 6 +++---
 setup.py  | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/WORKSPACE b/WORKSPACE
index fd66ebadca84..ccafde06a8b1 100644
--- a/WORKSPACE
+++ b/WORKSPACE
@@ -7,10 +7,10 @@ load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
 #    and update the sha256 with the result.
 http_archive(
     name = "org_tensorflow",
-    sha256 = "bee3000a7892008b32c65a5d6ba5d0a1f490e9e749d6789c5100d47068b1bbd8",
-    strip_prefix = "tensorflow-46cb56d405f0800e1741f68bba098d8eea5d0025",
+    sha256 = "76d7cbb4ce7d6b506e1845bb1bf7fe0ed89888d19069363629992da5059dc700",
+    strip_prefix = "tensorflow-994bca455e4d15b843115d4e43b0989726b7d767",
     urls = [
-        "https://github.com/tensorflow/tensorflow/archive/46cb56d405f0800e1741f68bba098d8eea5d0025.tar.gz",
+        "https://github.com/tensorflow/tensorflow/archive/994bca455e4d15b843115d4e43b0989726b7d767.tar.gz",
     ],
 )
 
diff --git a/setup.py b/setup.py
index 0be99846b3b8..c1eefa989209 100644
--- a/setup.py
+++ b/setup.py
@@ -21,7 +21,7 @@
 _default_cuda_version = '11'
 _available_cudnn_versions = ['82', '805']
 _default_cudnn_version = '82'
-_libtpu_version = '0.1.dev20220503'
+_libtpu_version = '0.1.dev20220504'
 
 _dct = {}
 with open('jax/version.py') as f: