Update TF commit for release

PiperOrigin-RevId: 446555288

Author: yashk2810

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",
     ],
 )
 

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: