diff --git a/.github/ISSUE_TEMPLATE/bug_report.yml b/.github/ISSUE_TEMPLATE/bug_report.yml
index 1d22372..7968b99 100644
--- a/.github/ISSUE_TEMPLATE/bug_report.yml
+++ b/.github/ISSUE_TEMPLATE/bug_report.yml
@@ -7,7 +7,7 @@
       value: |
         :star2: Thanks for taking the time to report this issue! :star2:
 
-        Please search through [other recent issues](https://github.com/google/iree/issues) to see if your report overlaps with an existing issue.
+        Please search through [other recent issues](https://github.com/iree-org/iree/issues) to see if your report overlaps with an existing issue.
   - type: textarea
     id: what-happened
     attributes:
diff --git a/.github/ISSUE_TEMPLATE/config.yml b/.github/ISSUE_TEMPLATE/config.yml
index c6f5b21..7f14799 100644
--- a/.github/ISSUE_TEMPLATE/config.yml
+++ b/.github/ISSUE_TEMPLATE/config.yml
@@ -4,7 +4,7 @@
     url: https://groups.google.com/forum/#!forum/iree-discuss
     about: For announcements and asynchronous discussion
   - name: 🗣 Start a discussion on GitHub
-    url: https://github.com/google/iree/discussions
+    url: https://github.com/iree-org/iree/discussions
     about: An alternative platform for asynchronous discussion
   - name: 💬 Join us on Discord
     url: https://discord.gg/26P4xW4
diff --git a/.github/workflows/publish_website.yml b/.github/workflows/publish_website.yml
index 74011c3..6dfad87 100644
--- a/.github/workflows/publish_website.yml
+++ b/.github/workflows/publish_website.yml
@@ -20,7 +20,7 @@
 jobs:
   publish_website:
     # Don't run this in everyone's forks.
-    if: github.repository == 'google/iree'
+    if: github.repository == 'iree-org/iree'
     runs-on: ubuntu-18.04
     steps:
       - name: Checkout out repository
diff --git a/.gitmodules b/.gitmodules
index c4feb0b..7a20f93 100644
--- a/.gitmodules
+++ b/.gitmodules
@@ -3,7 +3,7 @@
 	url = https://github.com/google/googletest.git
 [submodule "third_party/llvm-project"]
 	path = third_party/llvm-project
-	url = https://github.com/google/iree-llvm-fork.git
+	url = https://github.com/iree-org/iree-llvm-fork.git
 [submodule "third_party/vulkan_headers"]
 	path = third_party/vulkan_headers
 	url = https://github.com/KhronosGroup/Vulkan-Headers.git
@@ -37,7 +37,7 @@
 	url = https://github.com/axboe/liburing.git
 [submodule "third_party/mlir-hlo"]
 	path = third_party/mlir-hlo
-	url = https://github.com/google/iree-mhlo-fork.git
+	url = https://github.com/iree-org/iree-mhlo-fork.git
 [submodule "third_party/libyaml"]
 	path = third_party/libyaml
 	url = https://github.com/yaml/libyaml.git
diff --git a/CITATION.cff b/CITATION.cff
index d826e6f..49f4a66 100644
--- a/CITATION.cff
+++ b/CITATION.cff
@@ -16,8 +16,8 @@
     email: laurenzo@google.com
     affiliation: Google
 license: "Apache-2.0 WITH LLVM-exception"
-url: "https://google.github.io/iree/"
-repository-code: "https://github.com/google/iree"
+url: "https://iree-org.github.io/iree/"
+repository-code: "https://github.com/iree-org/iree"
 keywords:
   - compiler
   - "machine learning"
diff --git a/CMakeLists.txt b/CMakeLists.txt
index 4bb00fb..f311bb1 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -226,7 +226,7 @@
 
 # STREQUAL feels wrong here - we don't care about the exact true-value used,
 # ON or TRUE or something else. But we haven't been able to think of a less bad
-# alternative. https://github.com/google/iree/pull/8474#discussion_r840790062
+# alternative. https://github.com/iree-org/iree/pull/8474#discussion_r840790062
 if(NOT IREE_ENABLE_TSAN STREQUAL IREE_BYTECODE_MODULE_ENABLE_TSAN)
   message(SEND_ERROR
       "IREE_ENABLE_TSAN and IREE_BYTECODE_MODULE_ENABLE_TSAN must be "
diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md
index d0b73b1..a62bddf 100644
--- a/CONTRIBUTING.md
+++ b/CONTRIBUTING.md
@@ -66,7 +66,7 @@
 ## Peculiarities
 
 Our documentation on
-[repository management](https://github.com/google/iree/blob/main/docs/developers/developing_iree/repository_management.md)
+[repository management](https://github.com/iree-org/iree/blob/main/docs/developers/developing_iree/repository_management.md)
 has more information on some of the oddities in our repository setup and
 workflows. For the most part, these should be transparent to normal developer
 workflows.
diff --git a/README.md b/README.md
index f2408f0..029959a 100644
--- a/README.md
+++ b/README.md
@@ -6,7 +6,7 @@
 that scales up to meet the needs of the datacenter and down to satisfy the
 constraints and special considerations of mobile and edge deployments.
 
-See [our website](https://google.github.io/iree/) for project details, user
+See [our website](https://iree-org.github.io/iree/) for project details, user
 guides, and instructions on building from source.
 
 #### Project Status
@@ -19,7 +19,7 @@
 
 ## Communication Channels
 
-*   [GitHub issues](https://github.com/google/iree/issues): Feature requests,
+*   [GitHub issues](https://github.com/iree-org/iree/issues): Feature requests,
     bugs, and other work tracking
 *   [IREE Discord server](https://discord.gg/26P4xW4): Daily development
     discussions with the core team and collaborators
@@ -57,7 +57,7 @@
 ![IREE Architecture](docs/website/docs/assets/images/iree_architecture_dark.svg#gh-dark-mode-only)
 ![IREE Architecture](docs/website/docs/assets/images/iree_architecture.svg#gh-light-mode-only)
 
-See [our website](https://google.github.io/iree/) for more information.
+See [our website](https://iree-org.github.io/iree/) for more information.
 
 ## Presentations and Talks
 
diff --git a/benchmarks/README.md b/benchmarks/README.md
index db73d0a..cc08c07 100644
--- a/benchmarks/README.md
+++ b/benchmarks/README.md
@@ -63,7 +63,7 @@
 
 1. Install `iree-import-tflite`.
    ```
-   $ python -m pip install iree-tools-tflite -f https://github.com/google/iree/releases
+   $ python -m pip install iree-tools-tflite -f https://github.com/iree-org/iree/releases
    ```
 
 2. Expose and confirm the binary `iree-import-tflite` is in your path by running
@@ -85,7 +85,7 @@
 
 ### Running benchmark suites locally
 
-First you need to have [`iree-import-tflite`](https://google.github.io/iree/getting-started/tflite/)
+First you need to have [`iree-import-tflite`](https://iree-org.github.io/iree/getting-started/tflite/)
 and `requests` in your python environment. Then you can build the target
 `iree-benchmark-suites` to generate the required files:
 
diff --git a/benchmarks/dashboard.md b/benchmarks/dashboard.md
index 8689eee..fca5ca6 100644
--- a/benchmarks/dashboard.md
+++ b/benchmarks/dashboard.md
@@ -81,15 +81,15 @@
 
 This field specifies the IREE HAL driver:
 
-* [`local-task`](https://google.github.io/iree/deployment-configurations/cpu/):
+* [`local-task`](https://iree-org.github.io/iree/deployment-configurations/cpu/):
   For CPU via the local task system. Kernels contain CPU native instructions AOT
   compiled using LLVM. This driver issues workloads to the CPU asynchronously
   and supports multithreading.
-* [`local-sync`](https://google.github.io/iree/deployment-configurations/cpu/):
+* [`local-sync`](https://iree-org.github.io/iree/deployment-configurations/cpu/):
   For CPU via the local 'sync' device. Kernels contain contain CPU native
   instructions AOT compiled using LLVM. This driver issues workloads to the CPU
   synchronously.
-* [`Vulkan`](https://google.github.io/iree/deployment-configurations/gpu-vulkan/):
+* [`Vulkan`](https://iree-org.github.io/iree/deployment-configurations/gpu-vulkan/):
   For GPU via Vulkan. Kernels contain SPIR-V. This driver issues workload to
   the GPU via the Vulkan API.
 
diff --git a/build_tools/benchmarks/comparisons/README.md b/build_tools/benchmarks/comparisons/README.md
index 2d80e20..0124b54 100644
--- a/build_tools/benchmarks/comparisons/README.md
+++ b/build_tools/benchmarks/comparisons/README.md
@@ -30,7 +30,7 @@
 ### Install Android NDK and ADB
 
 Detailed steps
-[here](https://google.github.io/iree/building-from-source/android/#install-android-ndk-and-adb).
+[here](https://iree-org.github.io/iree/building-from-source/android/#install-android-ndk-and-adb).
 
 ### Install the Termux App and the Python Interpreter
 
diff --git a/build_tools/benchmarks/comparisons/setup_desktop.sh b/build_tools/benchmarks/comparisons/setup_desktop.sh
index 92b7b18..1ad2d8f 100644
--- a/build_tools/benchmarks/comparisons/setup_desktop.sh
+++ b/build_tools/benchmarks/comparisons/setup_desktop.sh
@@ -31,7 +31,7 @@
 mkdir "${SOURCE_DIR}"
 cd "${SOURCE_DIR}"
 
-git clone https://github.com/google/iree.git
+git clone https://github.com/iree-org/iree.git
 
 cd iree
 git submodule update --init
diff --git a/build_tools/benchmarks/comparisons/setup_mobile.sh b/build_tools/benchmarks/comparisons/setup_mobile.sh
index a426b37..5f81003 100644
--- a/build_tools/benchmarks/comparisons/setup_mobile.sh
+++ b/build_tools/benchmarks/comparisons/setup_mobile.sh
@@ -6,7 +6,7 @@
 
 # Run commands below on the workstation that the phone is attached to.
 # Prerequisites:
-#   Manual installations of the Android NDK and ADB are needed. See https://google.github.io/iree/building-from-source/android/#install-android-ndk-and-adb for instructions.
+#   Manual installations of the Android NDK and ADB are needed. See https://iree-org.github.io/iree/building-from-source/android/#install-android-ndk-and-adb for instructions.
 #   Manual installations of the Termux App and python are needed on the Android device. See README.md for instructions.
 
 #!/bin/bash
@@ -41,7 +41,7 @@
 mkdir "${SOURCE_DIR}"
 cd "${SOURCE_DIR}"
 
-git clone https://github.com/google/iree.git
+git clone https://github.com/iree-org/iree.git
 
 cd iree
 cp "${SOURCE_DIR}/iree/build_tools/benchmarks/set_adreno_gpu_scaling_policy.sh" "${ROOT_DIR}/setup/"
diff --git a/build_tools/benchmarks/post_benchmarks_as_pr_comment.py b/build_tools/benchmarks/post_benchmarks_as_pr_comment.py
index fc02184..32dfaea 100755
--- a/build_tools/benchmarks/post_benchmarks_as_pr_comment.py
+++ b/build_tools/benchmarks/post_benchmarks_as_pr_comment.py
@@ -47,8 +47,8 @@
 
 ABBR_PR_COMMENT_TITLE = "Abbreviated Benchmark Summary"
 GITHUB_GIST_API_PREFIX = "https://api.github.com/gists"
-GITHUB_IREE_API_PREFIX = "https://api.github.com/repos/google/iree"
-GITHUB_IREE_REPO_PREFIX = "https://github.com/google/iree"
+GITHUB_IREE_API_PREFIX = "https://api.github.com/repos/iree-org/iree"
+GITHUB_IREE_REPO_PREFIX = "https://github.com/iree-org/iree"
 GITHUB_USER = "iree-github-actions-bot"
 IREE_PROJECT_ID = 'IREE'
 # The maximal numbers of trials when querying base commit benchmark results.
diff --git a/build_tools/benchmarks/upload_benchmarks_to_dashboard.py b/build_tools/benchmarks/upload_benchmarks_to_dashboard.py
index 9a9d753..52f200e 100755
--- a/build_tools/benchmarks/upload_benchmarks_to_dashboard.py
+++ b/build_tools/benchmarks/upload_benchmarks_to_dashboard.py
@@ -30,7 +30,7 @@
                                          execute_cmd_and_get_output)
 from common.benchmark_thresholds import BENCHMARK_THRESHOLDS
 
-IREE_GITHUB_COMMIT_URL_PREFIX = 'https://github.com/google/iree/commit'
+IREE_GITHUB_COMMIT_URL_PREFIX = 'https://github.com/iree-org/iree/commit'
 IREE_PROJECT_ID = 'IREE'
 THIS_DIRECTORY = os.path.dirname(os.path.realpath(__file__))
 
@@ -39,8 +39,8 @@
 For the graph, the x axis is the Git commit index, and the y axis is the
 measured latency in milliseconds.
 <br>
-See <a href="https://github.com/google/iree/tree/main/benchmarks/dashboard.md">
-https://github.com/google/iree/tree/main/benchmarks/dashboard.md
+See <a href="https://github.com/iree-org/iree/tree/main/benchmarks/dashboard.md">
+https://github.com/iree-org/iree/tree/main/benchmarks/dashboard.md
 </a> for benchmark philosophy, specification, and definitions.
 """
 
diff --git a/build_tools/buildkite/pipelines/fragment/cla-failure.yml b/build_tools/buildkite/pipelines/fragment/cla-failure.yml
index 8a3b008..1d036f8 100644
--- a/build_tools/buildkite/pipelines/fragment/cla-failure.yml
+++ b/build_tools/buildkite/pipelines/fragment/cla-failure.yml
@@ -13,5 +13,5 @@
       Manual authorization required
     prompt: >
       Is it safe to run presubmits on
-      https://github.com/google/iree/commit/${BUILDKITE_COMMIT} from
-      https://github.com/google/iree/pull/${BUILDKITE_PULL_REQUEST}?
+      https://github.com/iree-org/iree/commit/${BUILDKITE_COMMIT} from
+      https://github.com/iree-org/iree/pull/${BUILDKITE_PULL_REQUEST}?
diff --git a/build_tools/buildkite/samples.yml b/build_tools/buildkite/samples.yml
index 881a813..585dbae 100644
--- a/build_tools/buildkite/samples.yml
+++ b/build_tools/buildkite/samples.yml
@@ -24,7 +24,7 @@
   # A host tools install is required for the WebAssembly build.
   # We _could_ instead download from the latest GitHub release, but all of our
   # releases are marked as "draft"s, so the convenient
-  # https://github.com/google/iree/releases/latest does not actually work :/
+  # https://github.com/iree-org/iree/releases/latest does not actually work :/
   - label: ":linux: Build host install"
     key: "build-host-install"
     commands:
diff --git a/build_tools/buildkite/scripts/set_buildkite_env.sh b/build_tools/buildkite/scripts/set_buildkite_env.sh
index 58a0fb3..6b5b67f 100644
--- a/build_tools/buildkite/scripts/set_buildkite_env.sh
+++ b/build_tools/buildkite/scripts/set_buildkite_env.sh
@@ -10,7 +10,7 @@
 export BUILDKITE_ORGANIZATION_SLUG="iree"
 export BUILDKITE_PIPELINE_SLUG="local-test"
 export BUILDKITE_BUILD_NUMBER=1
-export BUILDKITE_REPO="https://github.com/google/iree"
+export BUILDKITE_REPO="https://github.com/iree-org/iree"
 export BUILDKITE_COMMIT="$(git rev-parse HEAD)"
 export BUILDKITE_BRANCH="$(git rev-parse --abbrev-ref HEAD)"
 export BUILDKITE_BUILD_AUTHOR="$(git log -n 1 --pretty="format:%aN" HEAD)"
diff --git a/build_tools/buildkite/scripts/update_pipeline_configurations.py b/build_tools/buildkite/scripts/update_pipeline_configurations.py
index 2997b2c..eab53f9 100755
--- a/build_tools/buildkite/scripts/update_pipeline_configurations.py
+++ b/build_tools/buildkite/scripts/update_pipeline_configurations.py
@@ -31,7 +31,7 @@
 
 from common.buildkite_utils import get_pipeline
 
-GIT_REPO = "https://github.com/google/iree"
+GIT_REPO = "https://github.com/iree-org/iree"
 PIPELINE_ROOT_PATH = "build_tools/buildkite/pipelines"
 TRUSTED_BOOTSTRAP_PIPELINE_PATH = os.path.join(PIPELINE_ROOT_PATH, "fragment",
                                                "bootstrap-trusted.yml")
@@ -66,11 +66,11 @@
 
   new_build_url = f"https://buildkite.com/{organization}/{running_pipeline}/builds/{running_build_number}"
   script_relpath = os.path.relpath(__file__)
-  new_script_url = f"https://github.com/google/iree/blob/{short_running_commit}/{script_relpath}"
+  new_script_url = f"https://github.com/iree-org/iree/blob/{short_running_commit}/{script_relpath}"
 
   bootstrap_pipeline_path = TRUSTED_BOOTSTRAP_PIPELINE_PATH if trusted else UNTRUSTED_BOOTSTRAP_PIPELINE_PATH
 
-  new_pipeline_file_url = f"https://github.com/google/iree/blob/{short_running_commit}/{bootstrap_pipeline_path}"
+  new_pipeline_file_url = f"https://github.com/iree-org/iree/blob/{short_running_commit}/{bootstrap_pipeline_path}"
 
   header = UPDATE_INFO_HEADER.format(build_url=new_build_url,
                                      script_url=new_script_url,
diff --git a/build_tools/cmake/build_and_test_tsan.sh b/build_tools/cmake/build_and_test_tsan.sh
index 96d7b6e..e499529 100755
--- a/build_tools/cmake/build_and_test_tsan.sh
+++ b/build_tools/cmake/build_and_test_tsan.sh
@@ -71,7 +71,7 @@
 # Disable actually running GPU tests. This tends to yield TSan reports that are
 # specific to one's particular GPU driver and therefore hard to reproduce across
 # machines and often un-actionable anyway.
-# See e.g. https://github.com/google/iree/issues/9393
+# See e.g. https://github.com/iree-org/iree/issues/9393
 export IREE_VULKAN_DISABLE=1
 export IREE_CUDA_DISABLE=1
 
diff --git a/build_tools/cmake/build_tracing.sh b/build_tools/cmake/build_tracing.sh
index 6f261ea..0ae8f1a 100755
--- a/build_tools/cmake/build_tracing.sh
+++ b/build_tools/cmake/build_tracing.sh
@@ -28,7 +28,7 @@
 fi
 cd build-tracing
 
-# Note: https://github.com/google/iree/issues/6404 prevents us from building
+# Note: https://github.com/iree-org/iree/issues/6404 prevents us from building
 # tests with these other settings. Many tests invoke the compiler tools with
 # MLIR threading enabled, which crashes with compiler tracing enabled.
 "${CMAKE_BIN?}" -G Ninja .. \
diff --git a/build_tools/cmake/iree_copts.cmake b/build_tools/cmake/iree_copts.cmake
index ee0feea..240c141 100644
--- a/build_tools/cmake/iree_copts.cmake
+++ b/build_tools/cmake/iree_copts.cmake
@@ -306,7 +306,7 @@
 # compatible solution.
 #
 # See also:
-#   https://github.com/google/iree/issues/4665.
+#   https://github.com/iree-org/iree/issues/4665.
 #   https://discourse.cmake.org/t/how-to-fix-build-warning-d9025-overriding-gr-with-gr/878
 #   https://gitlab.kitware.com/cmake/cmake/-/issues/20610
 if(CMAKE_CXX_FLAGS AND "${CMAKE_CXX_COMPILER_ID}" STREQUAL "MSVC")
diff --git a/build_tools/cmake/iree_python.cmake b/build_tools/cmake/iree_python.cmake
index d35b36f..d487313 100644
--- a/build_tools/cmake/iree_python.cmake
+++ b/build_tools/cmake/iree_python.cmake
@@ -166,7 +166,7 @@
   foreach(_SRC_FILE ${_RULE_SRCS})
     # _SRC_FILE could have other path components in it, so we need to make a
     # directory for it. Ninja does this automatically, but make doesn't. See
-    # https://github.com/google/iree/issues/6801
+    # https://github.com/iree-org/iree/issues/6801
     set(_SRC_BIN_PATH "${CMAKE_CURRENT_BINARY_DIR}/${_SRC_FILE}")
     get_filename_component(_SRC_BIN_DIR "${_SRC_BIN_PATH}" DIRECTORY)
     add_custom_command(
diff --git a/build_tools/kokoro/gcp_ubuntu/simulate_kokoro.sh b/build_tools/kokoro/gcp_ubuntu/simulate_kokoro.sh
index 6095462..0d45d44 100755
--- a/build_tools/kokoro/gcp_ubuntu/simulate_kokoro.sh
+++ b/build_tools/kokoro/gcp_ubuntu/simulate_kokoro.sh
@@ -25,7 +25,7 @@
 set -o pipefail
 
 RELATIVE_KOKORO_BUILD_SCRIPT="${1?}"
-REPO_TO_CLONE="${2:-git@github.com:google/iree.git}"
+REPO_TO_CLONE="${2:-git@github.com:iree-org/iree.git}"
 
 # Set up the temporary Kokoro directories
 export KOKORO_ROOT="$(mktemp --directory --tmpdir kokoro-root-XXXXXX)"
diff --git a/build_tools/python_deploy/fetch_github_release_files.py b/build_tools/python_deploy/fetch_github_release_files.py
index 08cee8c..174b175 100755
--- a/build_tools/python_deploy/fetch_github_release_files.py
+++ b/build_tools/python_deploy/fetch_github_release_files.py
@@ -38,7 +38,7 @@
     print("No github user set. Recommend setting GITHUB_USER=user:token")
   print("Fetching release from tag:", tag)
   release_resp = requests.get(
-      f"https://api.github.com/repos/google/iree/releases/tags/{tag}",
+      f"https://api.github.com/repos/iree-org/iree/releases/tags/{tag}",
       headers={"Accept": "application/vnd.github.v3+json"},
       auth=github_auth)
   release_resp.raise_for_status()
diff --git a/build_tools/scripts/get_e2e_artifacts.py b/build_tools/scripts/get_e2e_artifacts.py
index bdac038..81889b6 100755
--- a/build_tools/scripts/get_e2e_artifacts.py
+++ b/build_tools/scripts/get_e2e_artifacts.py
@@ -96,7 +96,7 @@
   """Check that we aren't overwriting files unless we expect to."""
   # Note: We can't use a check that the files have identical contents because
   # tf_input.mlir can have random numbers appended to its function names.
-  # See https://github.com/google/iree/issues/3375
+  # See https://github.com/iree-org/iree/issues/3375
 
   expected_collision = any([name in filename for name in EXPECTED_COLLISIONS])
   if filename in written_paths and not expected_collision:
@@ -150,8 +150,8 @@
   print(
       "The bazel integrations build and tests are deprecated. This script "
       "may be reworked in the future. For the time being refer to "
-      "https://google.github.io/iree/building-from-source/python-bindings-and-importers/ "
-      "and https://github.com/google/iree/blob/main/docs/developers/developing_iree/e2e_benchmarking.md "
+      "https://iree-org.github.io/iree/building-from-source/python-bindings-and-importers/ "
+      "and https://github.com/iree-org/iree/blob/main/docs/developers/developing_iree/e2e_benchmarking.md "
       "for information on how to run TensorFlow benchmarks.")
   exit(1)
 
diff --git a/compiler/src/iree/compiler/API/python/test/tools/compiler_core_test.py b/compiler/src/iree/compiler/API/python/test/tools/compiler_core_test.py
index 0dcd090..68dba25 100644
--- a/compiler/src/iree/compiler/API/python/test/tools/compiler_core_test.py
+++ b/compiler/src/iree/compiler/API/python/test/tools/compiler_core_test.py
@@ -42,7 +42,7 @@
 
   # Compiling the string form means that the compiler does not have a valid
   # source file name, which can cause issues on the AOT side. Verify
-  # specifically. See: https://github.com/google/iree/issues/4439
+  # specifically. See: https://github.com/iree-org/iree/issues/4439
   def testCompileStrLLVMAOT(self):
     binary = iree.compiler.tools.compile_str(SIMPLE_MUL_ASM,
                                              input_type="mhlo",
@@ -52,7 +52,7 @@
 
   # Verifies that multiple target_backends are accepted. Which two are not
   # load bearing.
-  # See: https://github.com/google/iree/issues/4436
+  # See: https://github.com/iree-org/iree/issues/4436
   def testCompileMultipleBackends(self):
     binary = iree.compiler.tools.compile_str(
         SIMPLE_MUL_ASM,
diff --git a/compiler/src/iree/compiler/API/test/compile-mhlo-test-main.c b/compiler/src/iree/compiler/API/test/compile-mhlo-test-main.c
index 0355b72..91ff2d4 100644
--- a/compiler/src/iree/compiler/API/test/compile-mhlo-test-main.c
+++ b/compiler/src/iree/compiler/API/test/compile-mhlo-test-main.c
@@ -10,7 +10,7 @@
 //
 // Originally contributed due to the work of edubart who figured out how to
 // be the first user of the combined MLIR+IREE CAPI:
-// https://github.com/google/iree/pull/8582
+// https://github.com/iree-org/iree/pull/8582
 
 #include <stdio.h>
 
diff --git a/compiler/src/iree/compiler/Codegen/Common/ConvertToDestinationPassingStylePass.cpp b/compiler/src/iree/compiler/Codegen/Common/ConvertToDestinationPassingStylePass.cpp
index 82d35f1..f703948 100644
--- a/compiler/src/iree/compiler/Codegen/Common/ConvertToDestinationPassingStylePass.cpp
+++ b/compiler/src/iree/compiler/Codegen/Common/ConvertToDestinationPassingStylePass.cpp
@@ -327,7 +327,7 @@
 namespace {
 /// Adapts Linalg ops input operand to output operand. This is required for not
 /// creating extra alloca ops. For more details, see
-/// https://github.com/google/iree/issues/8303
+/// https://github.com/iree-org/iree/issues/8303
 struct AdaptLinalgInputOperandToOutputOperand
     : public OpRewritePattern<linalg::GenericOp> {
   using OpRewritePattern<linalg::GenericOp>::OpRewritePattern;
diff --git a/compiler/src/iree/compiler/Dialect/Flow/Conversion/TensorToFlow/ConvertTensorToFlow.cpp b/compiler/src/iree/compiler/Dialect/Flow/Conversion/TensorToFlow/ConvertTensorToFlow.cpp
index de7aaa1..6aa0052 100644
--- a/compiler/src/iree/compiler/Dialect/Flow/Conversion/TensorToFlow/ConvertTensorToFlow.cpp
+++ b/compiler/src/iree/compiler/Dialect/Flow/Conversion/TensorToFlow/ConvertTensorToFlow.cpp
@@ -292,7 +292,7 @@
     }
 
     // TODO: Decide if this needs to be replaced with a flow.tensor.cast
-    // See https://github.com/google/iree/issues/6418
+    // See https://github.com/iree-org/iree/issues/6418
     rewriter.replaceOpWithNewOp<IREE::Flow::TensorReshapeOp>(
         op, resultType, input, sourceDynamicDims, targetDynamicDims);
 
@@ -306,8 +306,8 @@
   LogicalResult matchAndRewrite(tensor::FromElementsOp op,
                                 PatternRewriter &rewriter) const override {
     // TODO: This pattern was mainly added to iron out some kinks specific to
-    // detensoring (see: https://github.com/google/iree/issues/1159). Do we need
-    // to expand this check for other uses?
+    // detensoring (see: https://github.com/iree-org/iree/issues/1159). Do we
+    // need to expand this check for other uses?
     if (op->getParentOfType<Flow::DispatchWorkgroupsOp>()) {
       return failure();
     }
diff --git a/compiler/src/iree/compiler/Dialect/HAL/Target/LLVM/LLVMAOTTarget.cpp b/compiler/src/iree/compiler/Dialect/HAL/Target/LLVM/LLVMAOTTarget.cpp
index 6c02b70..c1b6601 100644
--- a/compiler/src/iree/compiler/Dialect/HAL/Target/LLVM/LLVMAOTTarget.cpp
+++ b/compiler/src/iree/compiler/Dialect/HAL/Target/LLVM/LLVMAOTTarget.cpp
@@ -259,7 +259,7 @@
       // Tracy. In principle this could also be achieved by enabling unwind
       // tables, but we tried that and that didn't work in Tracy (which uses
       // libbacktrace), while enabling frame pointers worked.
-      // https://github.com/google/iree/issues/3957
+      // https://github.com/iree-org/iree/issues/3957
       func.addFnAttr("frame-pointer", "all");
 
       // -ffreestanding-like behavior.
diff --git a/docs/api_docs/python/requirements.txt b/docs/api_docs/python/requirements.txt
index 34af6a1..13bbfcd 100644
--- a/docs/api_docs/python/requirements.txt
+++ b/docs/api_docs/python/requirements.txt
@@ -6,6 +6,6 @@
 sphinx_toolbox==2.15.0
 
 # IREE Python API
--f https://github.com/google/iree/releases
+-f https://github.com/iree-org/iree/releases
 iree-compiler
 iree-runtime
diff --git a/docs/developers/best_practices.md b/docs/developers/best_practices.md
index a73f8ea..1c2d9a7 100644
--- a/docs/developers/best_practices.md
+++ b/docs/developers/best_practices.md
@@ -29,7 +29,7 @@
 arguments.
 
 See the
-[variables and state](https://github.com/google/iree/tree/main/samples/variables_and_state)
+[variables and state](https://github.com/iree-org/iree/tree/main/samples/variables_and_state)
 sample for further guidance on tracking and using state.
 
 ### Limit uses of dynamic shapes
@@ -40,7 +40,7 @@
 varying dimensions like the x/y/channel dimensions of images.
 
 See the
-[dynamic shapes](https://github.com/google/iree/tree/main/samples/dynamic_shapes)
+[dynamic shapes](https://github.com/iree-org/iree/tree/main/samples/dynamic_shapes)
 sample for further guidance on using dynamic shapes.
 
 ## Practices for compilation settings
@@ -52,7 +52,7 @@
 ### Tuning compilation heuristics
 
 IREE runs its own suite of benchmarks continuously using the definitions at
-https://github.com/google/iree/tree/main/benchmarks. The flags set for these
+https://github.com/iree-org/iree/tree/main/benchmarks. The flags set for these
 benchmarks represent the latest manually tuned values for workloads we track
 closely and referencing them may help with your own search for peak performance.
 You can use these flags in your own explorations, but note that as compiler
@@ -66,7 +66,7 @@
 ### Tuning runtime settings
 
 When running on the CPU, the task system flags specified in
-[iree/task/api.c](https://github.com/google/iree/blob/main/iree/task/api.c)
+[iree/task/api.c](https://github.com/iree-org/iree/blob/main/iree/task/api.c)
 give control over how worker threads will be created. For example, the
 `--task_topology_group_count=3` flag can be set to explicitly run on three
 workers rather than rely on heuristic selection that defaults to one worker
diff --git a/docs/developers/debugging/integration_correctness_issue_breakdown.md b/docs/developers/debugging/integration_correctness_issue_breakdown.md
index 72aa41b..79bd97f 100644
--- a/docs/developers/debugging/integration_correctness_issue_breakdown.md
+++ b/docs/developers/debugging/integration_correctness_issue_breakdown.md
@@ -9,12 +9,12 @@
 See [instructions for reproducing failures in TF/TFLite integration tests](https://github.com/hanhanW/iree/blob/main/docs/developers/debugging/tf_integrations_test_repro.md).
 
 For input data, they are not dumped within the flagfile. You can construct the
-function inputs by looking into `log.txt`. There is an [issue](https://github.com/google/iree/issues/8658)
+function inputs by looking into `log.txt`. There is an [issue](https://github.com/iree-org/iree/issues/8658)
 for tracking this.
 
 ## iree-samples
 
-Follow [README](https://github.com/google/iree-samples#readme) to run the model.
+Follow [README](https://github.com/iree-org/iree-samples#readme) to run the model.
 The MLIR files will be generated. You'll find the saved file from log. E.g.,
 
 ```
@@ -28,14 +28,14 @@
     --save-temp-iree-input=/tmp/iree-samples/tflitehub/tmp/mobilenet_v2_int8_test.py/tosa.mlir
 ```
 
-Unfortunately, the artifacts are not dumped in the runs. There is an [issue](https://github.com/google/iree/issues/8756)
+Unfortunately, the artifacts are not dumped in the runs. There is an [issue](https://github.com/iree-org/iree/issues/8756)
 for tracking this. A workaround can be found in the issue.
 
 # Narrow down the repro
 
 The model itself is big. IREE breaks a model into dispatches and launches the
 kernels. The inputs and outputs could be diverged starting from one of
-launches. To get a smaller reproduce, you can use [-iree-flow-trace-dispatch-tensors](https://github.com/google/iree/blob/main/docs/developers/developing_iree/developer_overview.md#iree-flow-trace-dispatch-tensors).
+launches. To get a smaller reproduce, you can use [-iree-flow-trace-dispatch-tensors](https://github.com/iree-org/iree/blob/main/docs/developers/developing_iree/developer_overview.md#iree-flow-trace-dispatch-tensors).
 You can compare the logs between builds/backends, and get the idea about which
 dispatch results in wrong outputs. The dumped inputs can be reused in a
 flagfile.
diff --git a/docs/developers/debugging/releases.md b/docs/developers/debugging/releases.md
index e8f705a..e9ef4c3 100644
--- a/docs/developers/debugging/releases.md
+++ b/docs/developers/debugging/releases.md
@@ -93,16 +93,16 @@
 branch.
 
 To run
-[`schedule_snapshot_release.yml`](https://github.com/google/iree/blob/main/.github/workflows/schedule_snapshot_release.yml),
+[`schedule_snapshot_release.yml`](https://github.com/iree-org/iree/blob/main/.github/workflows/schedule_snapshot_release.yml),
 comment out
-[this line](https://github.com/google/iree/blob/392449e986493bf710e3da637ebf807715da9ffe/.github/workflows/schedule_snapshot_release.yml#L14):
+[this line](https://github.com/iree-org/iree/blob/392449e986493bf710e3da637ebf807715da9ffe/.github/workflows/schedule_snapshot_release.yml#L14):
 ```yaml
 # Don't run this in everyone's forks.
-if: github.repository == 'google/iree'
+if: github.repository == 'iree-org/iree'
 ```
 
 And change the branch from 'main' to the branch you are developing on
-[here](https://github.com/google/iree/blob/392449e986493bf710e3da637ebf807715da9ffe/.github/workflows/schedule_snapshot_release.yml#L37):
+[here](https://github.com/iree-org/iree/blob/392449e986493bf710e3da637ebf807715da9ffe/.github/workflows/schedule_snapshot_release.yml#L37):
 ```yaml
 - name: Pushing changes
   uses: ad-m/github-push-action@40bf560936a8022e68a3c00e7d2abefaf01305a6  # v0.6.0
@@ -113,21 +113,21 @@
 ```
 
 To speed up
-[`build_package.yml`](https://github.com/google/iree/blob/main/.github/workflows/build_package.yml),
+[`build_package.yml`](https://github.com/iree-org/iree/blob/main/.github/workflows/build_package.yml),
 you may want to comment out some of the builds
-[here](https://github.com/google/iree/blob/392449e986493bf710e3da637ebf807715da9ffe/.github/workflows/build_package.yml#L34-L87).
+[here](https://github.com/iree-org/iree/blob/392449e986493bf710e3da637ebf807715da9ffe/.github/workflows/build_package.yml#L34-L87).
 The
-[`py-pure-pkgs`](https://github.com/google/iree/blob/392449e986493bf710e3da637ebf807715da9ffe/.github/workflows/build_package.yml#L52)
+[`py-pure-pkgs`](https://github.com/iree-org/iree/blob/392449e986493bf710e3da637ebf807715da9ffe/.github/workflows/build_package.yml#L52)
 build takes only ~2 minutes and the
-[`py-runtime-pkg`](https://github.com/google/iree/blob/392449e986493bf710e3da637ebf807715da9ffe/.github/workflows/build_package.yml#L39)
+[`py-runtime-pkg`](https://github.com/iree-org/iree/blob/392449e986493bf710e3da637ebf807715da9ffe/.github/workflows/build_package.yml#L39)
 build takes ~5, while the others can take several hours.
 
 From your development branch, you can manually run the
-[Schedule Snapshot Release](https://github.com/google/iree/actions/workflows/schedule_snapshot_release.yml)
+[Schedule Snapshot Release](https://github.com/iree-org/iree/actions/workflows/schedule_snapshot_release.yml)
 action, which invokes the
-[Build Native Release Packages](https://github.com/google/iree/actions/workflows/build_package.yml)
+[Build Native Release Packages](https://github.com/iree-org/iree/actions/workflows/build_package.yml)
 action, which finally invokes the
-[Validate and Publish Release](https://github.com/google/iree/actions/workflows/validate_and_publish_release.yml)
+[Validate and Publish Release](https://github.com/iree-org/iree/actions/workflows/validate_and_publish_release.yml)
 action.  If you already have a draft release and know the release id, package
 version, and run ID from a previous Build Native Release Packages run, you can
 also manually run just the Validate and Publish Release action.
diff --git a/docs/developers/debugging/tf_integrations_test_repro.md b/docs/developers/debugging/tf_integrations_test_repro.md
index f8d617b..14871ae 100644
--- a/docs/developers/debugging/tf_integrations_test_repro.md
+++ b/docs/developers/debugging/tf_integrations_test_repro.md
@@ -3,7 +3,7 @@
 These are steps to reproduce/address failures in TF/TFLite integration tests. All steps here
 assume starting from the IREE root directory.
 
-1. First setup the python environment as described [here](https://google.github.io/iree/building-from-source/python-bindings-and-importers/#environment-setup).
+1. First setup the python environment as described [here](https://iree-org.github.io/iree/building-from-source/python-bindings-and-importers/#environment-setup).
 
 ```
 python -m venv iree.venv
@@ -13,7 +13,7 @@
 2. Install latest IREE release binaries. The importers are not expected to change much, so using the release binaries should work for most cases
 
 ```
-python -m pip install iree-compiler iree-runtime iree-tools-tf iree-tools-tflite -f https://github.com/google/iree/releases/latest
+python -m pip install iree-compiler iree-runtime iree-tools-tf iree-tools-tflite -f https://github.com/iree-org/iree/releases/latest
 ```
 
 3. Install TF nightly
diff --git a/docs/developers/design_docs/codegen_passes.md b/docs/developers/design_docs/codegen_passes.md
index d4d76d1..c52daa2 100644
--- a/docs/developers/design_docs/codegen_passes.md
+++ b/docs/developers/design_docs/codegen_passes.md
@@ -639,8 +639,8 @@
 Once applied the resulting IR is in SPIR-V dialect that can be serialized to a
 SPIR-V binary.
 
-[ConvertToGPU]: https://github.com/google/iree/blob/main/iree/compiler/Conversion/LinalgToSPIRV/ConvertToGPUPass.cpp
-[ConvertToSPIRV]: https://github.com/google/iree/blob/main/iree/compiler/Conversion/LinalgToSPIRV/ConvertToSPIRVPass.cpp
+[ConvertToGPU]: https://github.com/iree-org/iree/blob/main/iree/compiler/Conversion/LinalgToSPIRV/ConvertToGPUPass.cpp
+[ConvertToSPIRV]: https://github.com/iree-org/iree/blob/main/iree/compiler/Conversion/LinalgToSPIRV/ConvertToSPIRVPass.cpp
 [DotAfterAll]: https://gist.github.com/MaheshRavishankar/9e2d406296f469515c4a79bf1e7eef44
 [GPUToSPIRV]: https://github.com/llvm/llvm-project/blob/master/mlir/include/mlir/Conversion/GPUToSPIRV/ConvertGPUToSPIRV.h
 [HLOToLinalgPass]: https://github.com/tensorflow/tensorflow/blob/75c40f6bff2faa3d90a375dfa4025b2e6e2d7a3d/tensorflow/compiler/mlir/xla/transforms/passes.h#L67
@@ -649,7 +649,7 @@
 [LinalgFusionOfTensorOps]: https://github.com/llvm/llvm-project/blob/80cb25cbd555f9634836b766c86aead435b60eaa/mlir/include/mlir/Dialect/Linalg/Passes.td#L30
 [LinalgPromotionPatterns]: https://github.com/llvm/llvm-project/blob/303a7f7a26e2aae1cb85f49dccbc0b5d14e0b2e0/mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h#L358
 [LinalgRationale]: https://mlir.llvm.org/docs/Rationale/RationaleLinalgDialect/
-[LinalgTileAndFuse]: https://github.com/google/iree/blob/main/iree/compiler/Conversion/LinalgToSPIRV/LinalgTileAndFusePass.cpp
+[LinalgTileAndFuse]: https://github.com/iree-org/iree/blob/main/iree/compiler/Conversion/LinalgToSPIRV/LinalgTileAndFusePass.cpp
 [LinalgTiling]: https://mlir.llvm.org/docs/Dialects/Linalg/#set-of-key-transformationsa-namekey_transformationsa
 [LinalgTilingPatterns]: https://github.com/llvm/llvm-project/blob/master/mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
 [NVVMAddressSpace]: https://docs.nvidia.com/cuda/nvvm-ir-spec/index.html#address-space
diff --git a/docs/developers/design_docs/cuda_backend.md b/docs/developers/design_docs/cuda_backend.md
index 6dbc93e..de4c82a 100644
--- a/docs/developers/design_docs/cuda_backend.md
+++ b/docs/developers/design_docs/cuda_backend.md
@@ -94,10 +94,10 @@
 4xf32=3 4 5 6
 ```
 
-[iree-cuda]: https://github.com/google/iree/tree/main/iree/hal/drivers/cuda/
-[cuda-symbols]: https://github.com/google/iree/blob/main/iree/hal/drivers/cuda/dynamic_symbols_tables.h
+[iree-cuda]: https://github.com/iree-org/iree/tree/main/iree/hal/drivers/cuda/
+[cuda-symbols]: https://github.com/iree-org/iree/blob/main/iree/hal/drivers/cuda/dynamic_symbols_tables.h
 [cuda-driver]: https://docs.nvidia.com/cuda/cuda-driver-api/index.html
 [cuda-graph]: https://developer.nvidia.com/blog/cuda-graphs/
 [vulkan-semaphore]: https://www.khronos.org/blog/vulkan-timeline-semaphores
-[semaphore-issue]: https://github.com/google/iree/issues/4727
-[codegen-passes]: https://github.com/google/iree/blob/main/docs/design_docs/codegen_passes.md
+[semaphore-issue]: https://github.com/iree-org/iree/issues/4727
+[codegen-passes]: https://github.com/iree-org/iree/blob/main/docs/design_docs/codegen_passes.md
diff --git a/docs/developers/design_docs/function_abi.md b/docs/developers/design_docs/function_abi.md
index d936bda..457ed58 100644
--- a/docs/developers/design_docs/function_abi.md
+++ b/docs/developers/design_docs/function_abi.md
@@ -50,26 +50,26 @@
 -   ValueType:
 
     -   Runtime:
-        [`iree_vm_value`](https://github.com/google/iree/blob/main/iree/vm/value.h)
+        [`iree_vm_value`](https://github.com/iree-org/iree/blob/main/iree/vm/value.h)
     -   Compile Time: primitive MLIR integer/floating point types
 
 -   Simple ND-Array Buffer:
 
     -   Runtime:
-        [`iree_hal_buffer_view`](https://github.com/google/iree/blob/main/iree/hal/buffer_view.h)
+        [`iree_hal_buffer_view`](https://github.com/iree-org/iree/blob/main/iree/hal/buffer_view.h)
     -   Compile Time: `tensor<>`
 
 -   String:
 
     -   Runtime:
-        [`iree_vm_list`](https://github.com/google/iree/blob/main/iree/vm/list.h)
+        [`iree_vm_list`](https://github.com/iree-org/iree/blob/main/iree/vm/list.h)
         containing `i8`
     -   Compile Time: `!util.list<i8>`
 
 -   Tuple:
 
     -   Runtime:
-        [`iree_vm_list`](https://github.com/google/iree/blob/main/iree/vm/list.h)
+        [`iree_vm_list`](https://github.com/iree-org/iree/blob/main/iree/vm/list.h)
         of variant
     -   Compile Time: `!util.list<?>`
     -   Note that these are statically type erased at the boundary.
@@ -77,7 +77,7 @@
 -   TypedList (homogenous):
 
     -   Runtime:
-        [`iree_vm_list`](https://github.com/google/iree/blob/main/iree/vm/list.h)
+        [`iree_vm_list`](https://github.com/iree-org/iree/blob/main/iree/vm/list.h)
         of `T`
     -   Compile Time: `!util.list<T>`
 
diff --git a/docs/developers/design_docs/hal_driver_features.md b/docs/developers/design_docs/hal_driver_features.md
index 7ef7cbb..f5dcbdc 100644
--- a/docs/developers/design_docs/hal_driver_features.md
+++ b/docs/developers/design_docs/hal_driver_features.md
@@ -101,20 +101,20 @@
    `VK_CapabilitiesAttr` to the attribute added to `SPV_ResourceLimitsAttr`.
 
 [d89364]: https://reviews.llvm.org/D89364
-[iree-hal]: https://github.com/google/iree/tree/main/iree/hal
-[iree-hal-c-api]: https://github.com/google/iree/blob/main/iree/hal/api.h
-[iree-hal-dialect]: https://github.com/google/iree/tree/main/iree/compiler/Dialect/HAL
-[iree-vulkan-dialect]: https://github.com/google/iree/tree/main/iree/compiler/Dialect/Vulkan
-[iree-vulkan-base-td]: https://github.com/google/iree/blob/main/iree/compiler/Dialect/Vulkan/IR/VulkanBase.td
-[iree-vulkan-cap-td]: https://github.com/google/iree/blob/main/iree/compiler/Dialect/Vulkan/IR/VulkanAttributes.td
-[iree-vulkan-target-env]: https://github.com/google/iree/blob/b4739d704de15029cd671e53e7d7e743f4ca2e35/iree/compiler/Dialect/HAL/Target/VulkanSPIRV/VulkanSPIRVTarget.cpp#L66-L70
-[iree-vulkan-target-triple]: https://github.com/google/iree/blob/main/iree/compiler/Dialect/Vulkan/Utils/TargetEnvUtils.cpp
-[iree-vulkan-target-conv]: https://github.com/google/iree/blob/b4739d704de15029cd671e53e7d7e743f4ca2e35/iree/compiler/Dialect/Vulkan/Utils/TargetEnvUtils.h#L29-L42
-[iree-spirv-target-attach]: https://github.com/google/iree/blob/b4739d704de15029cd671e53e7d7e743f4ca2e35/iree/compiler/Dialect/HAL/Target/VulkanSPIRV/VulkanSPIRVTarget.cpp#L228-L240
+[iree-hal]: https://github.com/iree-org/iree/tree/main/iree/hal
+[iree-hal-c-api]: https://github.com/iree-org/iree/blob/main/iree/hal/api.h
+[iree-hal-dialect]: https://github.com/iree-org/iree/tree/main/iree/compiler/Dialect/HAL
+[iree-vulkan-dialect]: https://github.com/iree-org/iree/tree/main/iree/compiler/Dialect/Vulkan
+[iree-vulkan-base-td]: https://github.com/iree-org/iree/blob/main/iree/compiler/Dialect/Vulkan/IR/VulkanBase.td
+[iree-vulkan-cap-td]: https://github.com/iree-org/iree/blob/main/iree/compiler/Dialect/Vulkan/IR/VulkanAttributes.td
+[iree-vulkan-target-env]: https://github.com/iree-org/iree/blob/b4739d704de15029cd671e53e7d7e743f4ca2e35/iree/compiler/Dialect/HAL/Target/VulkanSPIRV/VulkanSPIRVTarget.cpp#L66-L70
+[iree-vulkan-target-triple]: https://github.com/iree-org/iree/blob/main/iree/compiler/Dialect/Vulkan/Utils/TargetEnvUtils.cpp
+[iree-vulkan-target-conv]: https://github.com/iree-org/iree/blob/b4739d704de15029cd671e53e7d7e743f4ca2e35/iree/compiler/Dialect/Vulkan/Utils/TargetEnvUtils.h#L29-L42
+[iree-spirv-target-attach]: https://github.com/iree-org/iree/blob/b4739d704de15029cd671e53e7d7e743f4ca2e35/iree/compiler/Dialect/HAL/Target/VulkanSPIRV/VulkanSPIRVTarget.cpp#L228-L240
 [mlir-spirv-extensions-attr]: https://github.com/llvm/llvm-project/blob/076305568cd6c7c02ceb9cfc35e1543153406d19/mlir/include/mlir/Dialect/SPIRV/SPIRVBase.td#L314
 [mlir-spirv-target]: https://mlir.llvm.org/docs/Dialects/SPIR-V/#target-environment
 [mlir-spirv-attr]: https://github.com/llvm/llvm-project/blob/076305568cd6c7c02ceb9cfc35e1543153406d19/mlir/include/mlir/Dialect/SPIRV/SPIRVAttributes.h
 [mlir-spirv-target-td]: https://github.com/llvm/llvm-project/blob/076305568cd6c7c02ceb9cfc35e1543153406d19/mlir/include/mlir/Dialect/SPIRV/TargetAndABI.td
-[pr-3469]: https://github.com/google/iree/pull/3469
+[pr-3469]: https://github.com/iree-org/iree/pull/3469
 [vk-coop-mat-ext]: khronos.org/registry/vulkan/specs/1.2-extensions/man/html/VK_NV_cooperative_matrix.html
 [vulkaninfo]: https://vulkan.lunarg.com/doc/view/latest/linux/vulkaninfo.html
diff --git a/docs/developers/developing_iree/benchmarking.md b/docs/developers/developing_iree/benchmarking.md
index 7191d3c..c7bef0a 100644
--- a/docs/developers/developing_iree/benchmarking.md
+++ b/docs/developers/developing_iree/benchmarking.md
@@ -156,7 +156,7 @@
 introduce as little overhead as possible and have several benchmark binaries
 dedicated for evaluating the VM's performance. These benchmark binaries are
 named as `*_benchmark` in the
-[`iree/vm/`](https://github.com/google/iree/tree/main/iree/vm) directory. They
+[`iree/vm/`](https://github.com/iree-org/iree/tree/main/iree/vm) directory. They
 also use the Google Benchmark library as the above.
 
 ## CPU Configuration
diff --git a/docs/developers/developing_iree/contributor_tips.md b/docs/developers/developing_iree/contributor_tips.md
index c8a170f..20f179b 100644
--- a/docs/developers/developing_iree/contributor_tips.md
+++ b/docs/developers/developing_iree/contributor_tips.md
@@ -13,7 +13,7 @@
 We tend to use the "triangular" or "forking" workflow. Develop primarily on a
 clone of the repository on your development machine. Any local branches named
 the same as persistent branches from the
-[main repository](https://github.com/google/iree) (currently `main`, `google`,
+[main repository](https://github.com/iree-org/iree) (currently `main`, `google`,
 and `stable`) are pristine (though potentially stale) copies. You only
 fastforward these to match upstream and otherwise do development on other
 branches. When sending PRs, you push to a different branch on your public fork
@@ -42,7 +42,7 @@
     # From whatever directory under which you want to nest your repo
     $ git clone git@github.com:<github_username>/iree.git
     $ cd iree
-    $ git remote add upstream git@github.com:google/iree.git
+    $ git remote add upstream git@github.com:iree-org/iree.git
     ```
 
     This is especially important for maintainers who have write access (so can
@@ -55,7 +55,7 @@
     URL explicitly before pushing.
 
 3.  Use a script like
-    [git_update.sh](https://github.com/google/iree/blob/main/scripts/git/git_update.sh)
+    [git_update.sh](https://github.com/iree-org/iree/blob/main/scripts/git/git_update.sh)
     to easily synchronize `main` with `upstream`. Submodules make this is a
     little trickier than it should be. You can also add this as a git alias.
 
diff --git a/docs/developers/developing_iree/developer_overview.md b/docs/developers/developing_iree/developer_overview.md
index dc9004b..2b4fd99 100644
--- a/docs/developers/developing_iree/developer_overview.md
+++ b/docs/developers/developing_iree/developer_overview.md
@@ -4,59 +4,59 @@
 developers.
 
 ** Note: project layout is evolving at the moment, see
-   https://github.com/google/iree/issues/8955 **
+   https://github.com/iree-org/iree/issues/8955 **
 
 ## Project Code Layout
 
-[iree/](https://github.com/google/iree/blob/main/iree/)
+[iree/](https://github.com/iree-org/iree/blob/main/iree/)
 
 *   Core IREE project
 
-[integrations/](https://github.com/google/iree/blob/main/integrations/)
+[integrations/](https://github.com/iree-org/iree/blob/main/integrations/)
 
 *   Integrations between IREE and other frameworks, such as TensorFlow
 
-[runtime/](https://github.com/google/iree/tree/main/runtime/)
+[runtime/](https://github.com/iree-org/iree/tree/main/runtime/)
 
 *   IREE runtime code, with no dependencies on the compiler
 
-[bindings/](https://github.com/google/iree/blob/main/bindings/)
+[bindings/](https://github.com/iree-org/iree/blob/main/bindings/)
 
 *   Language and platform bindings, such as Python
-*   Also see [runtime/bindings/](https://github.com/google/iree/tree/main/runtime/bindings)
+*   Also see [runtime/bindings/](https://github.com/iree-org/iree/tree/main/runtime/bindings)
 
-[samples/](https://github.com/google/iree/blob/main/samples/)
+[samples/](https://github.com/iree-org/iree/blob/main/samples/)
 
 *   Samples built using IREE's runtime and compiler
-*   Also see the separate https://github.com/google/iree-samples repository
+*   Also see the separate https://github.com/iree-org/iree-samples repository
 
 ## IREE Compiler Code Layout
 
-[iree/compiler/](https://github.com/google/iree/blob/main/iree/compiler/)
+[iree/compiler/](https://github.com/iree-org/iree/blob/main/iree/compiler/)
 
 *   IREE's MLIR dialects, LLVM compiler passes, module translation code, etc.
 
 ## IREE Runtime Code Layout
 
-[iree/base/](https://github.com/google/iree/blob/main/runtime/src/iree/base/)
+[iree/base/](https://github.com/iree-org/iree/blob/main/runtime/src/iree/base/)
 
 *   Common types and utilities used throughout the runtime
 
-[iree/hal/](https://github.com/google/iree/blob/main/runtime/src/iree/hal/)
+[iree/hal/](https://github.com/iree-org/iree/blob/main/runtime/src/iree/hal/)
 
 *   **H**ardware **A**bstraction **L**ayer for IREE's runtime, with
     implementations for hardware and software backends
 
-[iree/schemas/](https://github.com/google/iree/blob/main/runtime/src/iree/schemas/)
+[iree/schemas/](https://github.com/iree-org/iree/blob/main/runtime/src/iree/schemas/)
 
 *   Shared data storage format definitions, primarily using
     [FlatBuffers](https://google.github.io/flatbuffers/)
 
-[tools/](https://github.com/google/iree/blob/main/tools/)
+[tools/](https://github.com/iree-org/iree/blob/main/tools/)
 
 *   Assorted tools used to optimize, translate, and evaluate IREE
 
-[iree/vm/](https://github.com/google/iree/blob/main/runtime/src/iree/vm/)
+[iree/vm/](https://github.com/iree-org/iree/blob/main/runtime/src/iree/vm/)
 
 *   Bytecode **V**irtual **M**achine used to work with IREE modules and invoke
     IREE functions
@@ -87,7 +87,7 @@
 `FileCheck` should be used to test the generated output.
 
 Here's an example of a small compiler pass running on a
-[test file](https://github.com/google/iree/blob/main/iree/compiler/Dialect/Util/Transforms/test/drop_compiler_hints.mlir):
+[test file](https://github.com/iree-org/iree/blob/main/iree/compiler/Dialect/Util/Transforms/test/drop_compiler_hints.mlir):
 
 ```shell
 $ ../iree-build/tools/iree-opt \
@@ -99,7 +99,7 @@
 
 For a more complex example, here's how to run IREE's complete transformation
 pipeline targeting the VMVX backend on the
-[fullyconnected.mlir](https://github.com/google/iree/blob/main/tests/e2e/models/fullyconnected.mlir)
+[fullyconnected.mlir](https://github.com/iree-org/iree/blob/main/tests/e2e/models/fullyconnected.mlir)
 model file:
 
 ```shell
@@ -110,7 +110,7 @@
 ```
 
 Custom passes may also be layered on top of `iree-opt`, see
-[samples/custom_modules/dialect](https://github.com/google/iree/blob/main/samples/custom_modules/dialect)
+[samples/custom_modules/dialect](https://github.com/iree-org/iree/blob/main/samples/custom_modules/dialect)
 for a sample.
 
 ### iree-compile
@@ -151,7 +151,7 @@
 and executes it as a series of
 [googletest](https://github.com/google/googletest) tests. This is the test
 runner for the IREE
-[check framework](https://github.com/google/iree/tree/main/docs/developing_iree/testing_guide.md#end-to-end-tests).
+[check framework](https://github.com/iree-org/iree/tree/main/docs/developing_iree/testing_guide.md#end-to-end-tests).
 
 ```shell
 $ ../iree-build/tools/iree-compile \
@@ -177,7 +177,7 @@
 function as exported by default and running all of them.
 
 For example, to execute the contents of
-[samples/models/simple_abs.mlir](https://github.com/google/iree/blob/main/samples/models/simple_abs.mlir):
+[samples/models/simple_abs.mlir](https://github.com/iree-org/iree/blob/main/samples/models/simple_abs.mlir):
 
 ```shell
 $ ../iree-build/tools/iree-run-mlir \
@@ -226,4 +226,4 @@
 ### Useful Vulkan driver flags
 
 For IREE's Vulkan runtime driver, there are a few useful flags defined in
-[driver_module.cc](https://github.com/google/iree/blob/main/iree/hal/drivers/vulkan/registration/driver_module.cc):
+[driver_module.cc](https://github.com/iree-org/iree/blob/main/iree/hal/drivers/vulkan/registration/driver_module.cc):
diff --git a/docs/developers/developing_iree/e2e_benchmarking.md b/docs/developers/developing_iree/e2e_benchmarking.md
index 84c9fae..d808abc 100644
--- a/docs/developers/developing_iree/e2e_benchmarking.md
+++ b/docs/developers/developing_iree/e2e_benchmarking.md
@@ -5,7 +5,7 @@
 > Note:<br>
 > &nbsp;&nbsp;&nbsp;&nbsp;The TensorFlow integrations are currently being
   refactored. The `bazel` build is deprecated. Refer to
-  https://google.github.io/iree/get-started/getting-started-python for a general
+  https://iree-org.github.io/iree/get-started/getting-started-python for a general
   overview of how to build and execute the e2e tests.
 
 We use our end-to-end TensorFlow integration tests to test compilation and
@@ -14,7 +14,7 @@
 to, and to run them using valid inputs for each model.
 
 This guide assumes that you can run the tensorflow integration tests. See
-[this doc](https://google.github.io/iree/building-from-source/python-bindings-and-importers/)
+[this doc](https://iree-org.github.io/iree/building-from-source/python-bindings-and-importers/)
 for more information. That doc also covers writing new tests, which you'll need
 to do if you'd like to benchmark a new TensorFlow model.
 
@@ -187,7 +187,7 @@
 
 IREE only supports compiling to Android with CMake. Documentation on setting up
 your environment to cross-compile to Android can be found
-[here](https://google.github.io/iree/building-from-source/android/).
+[here](https://iree-org.github.io/iree/building-from-source/android/).
 
 ```shell
 # After following the instructions above up to 'Build all targets', the
diff --git a/docs/developers/developing_iree/llvm_version_bump.md b/docs/developers/developing_iree/llvm_version_bump.md
index 0190d38..8786087 100644
--- a/docs/developers/developing_iree/llvm_version_bump.md
+++ b/docs/developers/developing_iree/llvm_version_bump.md
@@ -5,7 +5,7 @@
 is just to get us by until we get it better scripted/automated.
 
 Note that scripts referenced in this guide are temporarily hosted in the
-[iree-samples repository](https://github.com/google/iree-samples/tree/main/scripts/integrate).
+[iree-samples repository](https://github.com/iree-org/iree-samples/tree/main/scripts/integrate).
 This is because it is very non-user friendly to have branch and submodule
 management scripts in the repository being managed, and we don't have an
 immediately better place. In this guide, we reference this location as
@@ -16,14 +16,14 @@
 The IREE team maintains fork repositories for both llvm-project and mlir-hlo,
 allowing them to be patched out of band. These repositories are:
 
-* https://github.com/google/iree-llvm-fork (`main` branch)
-* https://github.com/google/iree-mhlo-fork (`master` branch)
-* https://github.com/google/iree-tf-fork (`master` branch)
+* https://github.com/iree-org/iree-llvm-fork (`main` branch)
+* https://github.com/iree-org/iree-mhlo-fork (`master` branch)
+* https://github.com/iree-org/iree-tf-fork (`master` branch)
 
 By the time you read this, they may be on a cron to advance automatically, but
 even so, it is a good idea to advance them prior to any integrate activities
 so that you have freshest commits available. Iree repository has an
-action named [Advance Upstream Forks](https://github.com/google/iree/actions/workflows/advance_upstream_forks.yml)
+action named [Advance Upstream Forks](https://github.com/iree-org/iree/actions/workflows/advance_upstream_forks.yml)
 to update the forks. Just select `Run Workflow` on that action and give it a
 minute. You should see the fork repository mainline branch move forward.
 
@@ -110,8 +110,8 @@
 
 ```
 cd ~/src
-git clone --branch master https://github.com/google/iree-tf-fork.git
-git clone --branch master https://github.com/google/iree-mhlo-fork.git
+git clone --branch master https://github.com/iree-org/iree-tf-fork.git
+git clone --branch master https://github.com/iree-org/iree-mhlo-fork.git
 ```
 
 Get MHLO's published version:
@@ -208,7 +208,7 @@
 Either fix any issues or get people to do so and land patches until the
 PR is green.
 
-A script from [iree-samples](https://github.com/google/iree-samples/blob/main/scripts/integrate/bump_llvm.py)
+A script from [iree-samples](https://github.com/iree-org/iree-samples/blob/main/scripts/integrate/bump_llvm.py)
 repository can help with bumping the LLVM version and creating a PR.
 To use the script the steps are
 
@@ -324,7 +324,7 @@
   ..
 ```
 
-To repro failures in CI `bazel_linux_x86-swiftshader_core`, we can follow the [doc](https://github.com/google/iree/blob/main/docs/developers/get_started/building_with_bazel_linux.md) to build IREE using bazel. E.g.,
+To repro failures in CI `bazel_linux_x86-swiftshader_core`, we can follow the [doc](https://github.com/iree-org/iree/blob/main/docs/developers/get_started/building_with_bazel_linux.md) to build IREE using bazel. E.g.,
 
 ```bash
 export CC=clang
diff --git a/docs/developers/developing_iree/profiling_vulkan_gpu.md b/docs/developers/developing_iree/profiling_vulkan_gpu.md
index 0368d1a..294c69b 100644
--- a/docs/developers/developing_iree/profiling_vulkan_gpu.md
+++ b/docs/developers/developing_iree/profiling_vulkan_gpu.md
@@ -21,7 +21,7 @@
 app. In IREE we have a simple Android native app wrapper to help package
 IREE core libraries together with a specific VM bytecode invocation into an
 Android app. The wrapper and its documentation are placed at
-[`tools/android/run_module_app/`](https://github.com/google/iree/tree/main/tools/android/run_module_app).
+[`tools/android/run_module_app/`](https://github.com/iree-org/iree/tree/main/tools/android/run_module_app).
 
 For example, to package a module compiled from the following `mhlo-dot.mlir` as
 an Android app:
diff --git a/docs/developers/developing_iree/profiling_with_tracy.md b/docs/developers/developing_iree/profiling_with_tracy.md
index 497107c..b9eca8e 100644
--- a/docs/developers/developing_iree/profiling_with_tracy.md
+++ b/docs/developers/developing_iree/profiling_with_tracy.md
@@ -160,7 +160,7 @@
 
 For tracing the compiler, additionally set `IREE_ENABLE_COMPILER_TRACING` to
 `ON`. Compiler tracing is less stable, particularly on Linux with MLIR threading
-enabled (https://github.com/google/iree/issues/6404).
+enabled (https://github.com/iree-org/iree/issues/6404).
 
 Once done configuring CMake, proceed to rebuild, e.g.
 
@@ -374,6 +374,6 @@
 ## Configuring Tracy instrumentation
 
 Set IREE's `IREE_TRACING_MODE` value (defined in
-[iree/base/tracing.h](https://github.com/google/iree/blob/main/iree/base/tracing.h))
+[iree/base/tracing.h](https://github.com/iree-org/iree/blob/main/iree/base/tracing.h))
 to adjust which tracing features, such as allocation tracking and callstacks,
 are enabled.
diff --git a/docs/developers/developing_iree/sanitizers.md b/docs/developers/developing_iree/sanitizers.md
index 8112b7e..045a5f7 100644
--- a/docs/developers/developing_iree/sanitizers.md
+++ b/docs/developers/developing_iree/sanitizers.md
@@ -59,7 +59,7 @@
 etc. (anything that internally uses the CMake `iree_bytecode_module` macro).
 
 The CMake option `IREE_BUILD_SAMPLES=OFF` is needed because samples [currently
-assume](https://github.com/google/iree/pull/8893) that the embedded linker is
+assume](https://github.com/iree-org/iree/pull/8893) that the embedded linker is
 used, so they are incompatible with
 `IREE_BYTECODE_MODULE_FORCE_SYSTEM_DYLIB_LINKER=ON`.
 
@@ -70,7 +70,7 @@
 If you know what you're doing (i.e. if you are not building targets that
 internally involve a LLVM/CPU `iree_bytecode_module`), feel free to locally comment out
 the CMake error and only set `IREE_ENABLE_TSAN`. Also see a
-[past attempt]((https://github.com/google/iree/pull/8966) to relax that CMake
+[past attempt]((https://github.com/iree-org/iree/pull/8966) to relax that CMake
 validation.
 
 ### MSan (MemorySanitizer)
diff --git a/docs/developers/developing_iree/testing_guide.md b/docs/developers/developing_iree/testing_guide.md
index 2c3464d..2b1c3e0 100644
--- a/docs/developers/developing_iree/testing_guide.md
+++ b/docs/developers/developing_iree/testing_guide.md
@@ -85,7 +85,7 @@
 
 We have created a corresponding CMake function `iree_cc_test` that mirrors the
 Bazel rule's behavior. Our
-[Bazel to CMake converter](https://github.com/google/iree/tree/main/build_tools/bazel_to_cmake/bazel_to_cmake.py)
+[Bazel to CMake converter](https://github.com/iree-org/iree/tree/main/build_tools/bazel_to_cmake/bazel_to_cmake.py)
 should generally derive the `CMakeLists.txt` file from the BUILD file:
 
 ```cmake
@@ -113,7 +113,7 @@
 ### Running a Test
 
 For the test
-https://github.com/google/iree/blob/main/iree/compiler/Dialect/VM/Conversion/MathToVM/test/arithmetic_ops.mlir
+https://github.com/iree-org/iree/blob/main/iree/compiler/Dialect/VM/Conversion/MathToVM/test/arithmetic_ops.mlir
 
 With CMake, run this from the build directory:
 
@@ -158,7 +158,7 @@
 ```
 
 There is a corresponding CMake function, calls to which will be generated by our
-[Bazel to CMake Converter](https://github.com/google/iree/tree/main/build_tools/bazel_to_cmake/bazel_to_cmake.py).
+[Bazel to CMake Converter](https://github.com/iree-org/iree/tree/main/build_tools/bazel_to_cmake/bazel_to_cmake.py).
 
 ```cmake
 iree_lit_test_suite(
@@ -211,7 +211,7 @@
 ### Running a Test
 
 For the test
-https://github.com/google/iree/tree/main/tests/e2e/xla_ops/floor.mlir
+https://github.com/iree-org/iree/tree/main/tests/e2e/xla_ops/floor.mlir
 compiled for the VMVX target backend and running on the VMVX driver (here they
 match exactly, but in principle there's a many-to-many mapping from backends to
 drivers).
@@ -277,7 +277,7 @@
 Next we use this input constant to exercise the runtime feature under test (in
 this case, just a single floor operation). Finally, we use a check dialect
 operation to make an assertion about the output. There are a few different
-[assertion operations](https://github.com/google/iree/tree/main/iree/compiler/Dialect/Modules/Check).
+[assertion operations](https://github.com/iree-org/iree/tree/main/iree/compiler/Dialect/Modules/Check).
 Here we use the `expect_almost_eq_const` op: *almost* because we are comparing
 floats and want to allow for floating-point imprecision, and *const* because we
 want to compare it to a constant value. This last part is just syntactic sugar
@@ -374,7 +374,7 @@
 
 The CMake functions follow a similar pattern. The calls to them are generated in
 our `CMakeLists.txt` file by
-[bazel_to_cmake](https://github.com/google/iree/tree/main/build_tools/bazel_to_cmake/bazel_to_cmake.py).
+[bazel_to_cmake](https://github.com/iree-org/iree/tree/main/build_tools/bazel_to_cmake/bazel_to_cmake.py).
 
 ## Binding Tests
 
diff --git a/docs/developers/get_started/README.md b/docs/developers/get_started/README.md
index 8bcf918..c5b160a 100644
--- a/docs/developers/get_started/README.md
+++ b/docs/developers/get_started/README.md
@@ -3,7 +3,7 @@
 ---
 
 The primary guides are located at
-https://google.github.io/iree/building-from-source/ (source in
+https://iree-org.github.io/iree/building-from-source/ (source in
 [the website/ folder](../../website/docs/building-from-source/) )
 
 ---
diff --git a/docs/developers/get_started/building_with_bazel_linux.md b/docs/developers/get_started/building_with_bazel_linux.md
index 5fd18da..0715b84 100644
--- a/docs/developers/get_started/building_with_bazel_linux.md
+++ b/docs/developers/get_started/building_with_bazel_linux.md
@@ -12,7 +12,7 @@
 ### Install Bazel
 
 Install Bazel, matching IREE's
-[`.bazelversion`](https://github.com/google/iree/blob/main/.bazelversion) by
+[`.bazelversion`](https://github.com/iree-org/iree/blob/main/.bazelversion) by
 following the
 [official docs](https://docs.bazel.build/versions/master/install.html).
 
@@ -44,7 +44,7 @@
 Clone the repository, initialize its submodules and configure:
 
 ```shell
-$ git clone https://github.com/google/iree.git
+$ git clone https://github.com/iree-org/iree.git
 $ cd iree
 $ git submodule update --init
 $ python3 configure_bazel.py
@@ -105,7 +105,7 @@
 ```
 
 Translate a
-[MLIR file](https://github.com/google/iree/blob/main/samples/models/simple_abs.mlir)
+[MLIR file](https://github.com/iree-org/iree/blob/main/samples/models/simple_abs.mlir)
 and execute a function in the compiled module:
 
 ```shell
diff --git a/docs/developers/get_started/building_with_bazel_macos.md b/docs/developers/get_started/building_with_bazel_macos.md
index 00149b1..75b7e6f 100644
--- a/docs/developers/get_started/building_with_bazel_macos.md
+++ b/docs/developers/get_started/building_with_bazel_macos.md
@@ -45,7 +45,7 @@
 Clone the repository, initialize its submodules and configure:
 
 ```shell
-$ git clone https://github.com/google/iree.git
+$ git clone https://github.com/iree-org/iree.git
 $ cd iree
 $ git submodule update --init
 $ python3 configure_bazel.py
@@ -108,7 +108,7 @@
 ```
 
 Translate a
-[MLIR file](https://github.com/google/iree/blob/main/samples/models/simple_abs.mlir)
+[MLIR file](https://github.com/iree-org/iree/blob/main/samples/models/simple_abs.mlir)
 and execute a function in the compiled module:
 
 ```shell
diff --git a/docs/developers/get_started/building_with_bazel_windows.md b/docs/developers/get_started/building_with_bazel_windows.md
index b90f6a4..38bf367 100644
--- a/docs/developers/get_started/building_with_bazel_windows.md
+++ b/docs/developers/get_started/building_with_bazel_windows.md
@@ -18,7 +18,7 @@
 ### Install Bazel
 
 Install Bazel version > 2.0.0 (see
-[`.bazelversion`](https://github.com/google/iree/blob/main/.bazelversion) for
+[`.bazelversion`](https://github.com/iree-org/iree/blob/main/.bazelversion) for
 the specific version IREE uses) by following the
 [official docs](https://docs.bazel.build/versions/master/install-windows.html).
 
@@ -52,7 +52,7 @@
 clone the repository, initialize its submodules, and configure:
 
 ```powershell
-> git clone https://github.com/google/iree.git
+> git clone https://github.com/iree-org/iree.git
 > cd iree
 > git submodule update --init
 > python configure_bazel.py
@@ -111,7 +111,7 @@
 ```
 
 Translate a
-[MLIR file](https://github.com/google/iree/blob/main/samples/models/simple_abs.mlir)
+[MLIR file](https://github.com/iree-org/iree/blob/main/samples/models/simple_abs.mlir)
 and execute a function in the compiled module:
 
 ```powershell
diff --git a/docs/developers/objectives.md b/docs/developers/objectives.md
index d6e865d..cb07621 100644
--- a/docs/developers/objectives.md
+++ b/docs/developers/objectives.md
@@ -68,7 +68,7 @@
 ### P1 O: [Perf burndown] Bring up and execute perf burndown process for MobileBERT workload
 
 +   P1 KR: IREE CPU codegen achieves near peak throughput on all of the MobileBERT matmul shapes
-    + At least the ones with > 4% weight in the above profile. 
+    + At least the ones with > 4% weight in the above profile.
 +   P1 KR: Ensure that IREE CPU codegen achieves decent performance on softmax. That means matching TFLite performance on the softmax layers in the MobileBERT model.
 +   P1 KR: Able to benchmark and profile the whole MobileBERT workload in both TFLite and IREE and compare results.
 +   P1 KR: Performed at least 2 cycles of: (a) assess whole workload performance (b) identify key source of performance delta between TFLite and IREE (c) fix the issue (d) repeat.
@@ -87,14 +87,14 @@
 
 +   P1 KR: Able to perform IREE profiling using Tracy
 
-    + See [https://github.com/google/iree/issues/1886](https://github.com/google/iree/issues/1886), [https://github.com/wolfpld/tracy](https://github.com/wolfpld/tracy)
+    + See [https://github.com/iree-org/iree/issues/1886](https://github.com/iree-org/iree/issues/1886), [https://github.com/wolfpld/tracy](https://github.com/wolfpld/tracy)
 
 +   P1 KR: Able to map time spent in execution to back to source using Tracy
-    +   See https://github.com/google/iree/issues/1199
+    +   See https://github.com/iree-org/iree/issues/1199
     +   Source layer (source python, HLO, HAL, etc) is configurable at compile time.
 
 +   P1 KR: Able to track compile-time performance-related statistics
-    + See [https://github.com/google/iree/issues/1409](https://github.com/google/iree/issues/1409)
+    + See [https://github.com/iree-org/iree/issues/1409](https://github.com/iree-org/iree/issues/1409)
     + Initial stats to track: number of executables, the serialized size of constant data, the serialized size of the executables, the number of host readbacks (flow.tensor.load), backend specific stats like the number of split dispatches in the SPIR-V backend, dynamic shape info like the number of tensors with dynamic shapes that survive after shape propagation
 
 +   P1 KR: Internal and external contributors able to confidently assess performance impact of a change.
@@ -105,7 +105,7 @@
 
 ### P1 O: [Perf burndown] Identify target workload for IREE perf credibility burndown
 
-Note: This is in preparation for a 2021Q1 objective: Establish IREE's credibility at delivering competitive production levels of performance on a realistic use case.  
+Note: This is in preparation for a 2021Q1 objective: Establish IREE's credibility at delivering competitive production levels of performance on a realistic use case.
 
 +   P1 KR: Defined criteria for evaluating workloads for the burndown.
 +   P1 KR: Evaluated criteria (including performance analysis) for all candidate workloads.
@@ -144,7 +144,7 @@
 
 ### P1 O: [User-facing] Prepare to support real-world use cases
 
-Notes: Keep a pulse on deployment user journeys, continue to gather requirements from interested users, set ourselves on a path to production use on at least one platform.  
+Notes: Keep a pulse on deployment user journeys, continue to gather requirements from interested users, set ourselves on a path to production use on at least one platform.
 
 +   P1 KR: A new sample application showing high-level IREE behavior
 
@@ -186,11 +186,11 @@
 
 +   P1 KR: Support Linalg fusion on buffers using stack allocations.
 +   P1 KR: Improve AOT linking and support automatic toolchain discovery
-    
+
     +   Link all executables in a single dylib.
     +   Support exporting/loading dylib to standalone binary.
 
-### P1 O: [MLIR codegen] Retargetable codegeneration (Vector dialect-based approach) 
+### P1 O: [MLIR codegen] Retargetable codegeneration (Vector dialect-based approach)
 
 +   P1 KR: Develop mechanisms to distribute vector operation at workgroup level to vector operation at subgroup level / work item level
 +   P1 KR: Handle distribution of producer-consumer vector operations to implement fusion
diff --git a/docs/developers/tensorflow_coverage/language_and_speech_coverage.md b/docs/developers/tensorflow_coverage/language_and_speech_coverage.md
index d6d20cc..2c6ab58 100644
--- a/docs/developers/tensorflow_coverage/language_and_speech_coverage.md
+++ b/docs/developers/tensorflow_coverage/language_and_speech_coverage.md
@@ -3,7 +3,7 @@
 Tests of MobileBert and streamable Keyword Spotting models.
 
 IREE has three main backend
-[targets](https://github.com/google/iree/tree/main/iree/compiler/Dialect/HAL/Target):
+[targets](https://github.com/iree-org/iree/tree/main/iree/compiler/Dialect/HAL/Target):
 `vmvx` , `llvm` and `vulkan-spirv`. We also test TFLite in our infrastructure
 for benchmarking purposes.
 
@@ -13,40 +13,40 @@
 
 |                                                          target                                                          |                    tflite                    |                     vmvx                     |                 vulkan-spirv                 |
 |:------------------------------------------------------------------------------------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|
-| [mobile_bert_squad_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/mobile_bert_squad_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+| [mobile_bert_squad_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/mobile_bert_squad_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
 
 ## End to end tests of [Keyword Spotting Streaming](https://github.com/google-research/google-research/tree/master/kws_streaming) models
 
 |                                                               target                                                              |                    tflite                    |                     vmvx                     |                 vulkan-spirv                 |
 |:---------------------------------------------------------------------------------------------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|
-|    [att_mh_rnn](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)    | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|      [att_rnn](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)     | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [cnn](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [cnn_stride](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|       [crnn](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)       | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> |
-|        [dnn](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|      [ds_cnn](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|   [ds_tc_resnet](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)   | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|        [gru](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)       | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> |
-|     [inception](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-| [inception_resnet](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|       [lstm](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)       | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> |
-|     [mobilenet](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|   [mobilenet_v2](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)   | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|       [svdf](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [svdf_resnet](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)   | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [tc_resnet](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [xception](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [att_mh_rnn](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)    | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|      [att_rnn](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)     | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [cnn](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [cnn_stride](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [crnn](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)       | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> |
+|        [dnn](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|      [ds_cnn](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|   [ds_tc_resnet](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)   | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|        [gru](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)       | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> |
+|     [inception](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+| [inception_resnet](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [lstm](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)       | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> |
+|     [mobilenet](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|   [mobilenet_v2](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)   | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [svdf](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [svdf_resnet](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)   | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [tc_resnet](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [xception](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
 
 ## End to end tests of [Keyword Spotting Streaming](https://github.com/google-research/google-research/tree/master/kws_streaming) models in internal streaming mode
 
 |                                                             target                                                            |                    tflite                    |                     vmvx                     |                 vulkan-spirv                 |
 |:-----------------------------------------------------------------------------------------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|
-|      [cnn](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|  [cnn_stride](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)  | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [crnn](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|      [dnn](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-| [ds_tc_resnet](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py) | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|      [gru](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [lstm](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [svdf](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|      [cnn](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|  [cnn_stride](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)  | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [crnn](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|      [dnn](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+| [ds_tc_resnet](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py) | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|      [gru](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [lstm](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [svdf](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/keyword_spotting_streaming_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
diff --git a/docs/developers/tensorflow_coverage/tf_base_coverage.md b/docs/developers/tensorflow_coverage/tf_base_coverage.md
index 932d7c7..b252c2b 100644
--- a/docs/developers/tensorflow_coverage/tf_base_coverage.md
+++ b/docs/developers/tensorflow_coverage/tf_base_coverage.md
@@ -3,7 +3,7 @@
 Tests of the `tf` , `tf.math` , `tf.nn` , `tf.signal` and `tf.strings` APIs.
 
 IREE has three main backend
-[targets](https://github.com/google/iree/tree/main/iree/compiler/Dialect/HAL/Target):
+[targets](https://github.com/iree-org/iree/tree/main/iree/compiler/Dialect/HAL/Target):
 `vmvx` , `llvm` and `vulkan-spirv`. We also test TFLite in our infrastructure
 for benchmarking purposes.
 
@@ -13,40 +13,40 @@
 
 |                                                           target                                                           |                    tflite                    |                     vmvx                     |                 vulkan-spirv                 |
 |:--------------------------------------------------------------------------------------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|
-|         [batch_norm_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/batch_norm_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|  [batch_to_space_nd_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/batch_to_space_nd_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|       [broadcast_to_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/broadcast_to_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|       [broadcasting_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/broadcasting_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|             [concat_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/concat_test.py)             | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|       [control_flow_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/control_flow_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|               [conv_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/conv_test.py)               | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [conv_transpose_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/conv_transpose_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|         [depth_conv_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/depth_conv_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|   [dynamic_mlp_relu_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/dynamic_mlp_relu_test.py)   | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|        [dynamic_mlp_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/dynamic_mlp_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|     [einsum_dynamic_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/einsum_dynamic_test.py)     | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|      [einsum_static_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/einsum_static_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|      [einsum_vector_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/einsum_vector_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|                [fft_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/fft_test.py)                | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|               [fill_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/fill_test.py)               | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|             [gather_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/gather_test.py)             | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|       [image_resize_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/image_resize_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|           [linspace_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/linspace_test.py)           | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|         [mandelbrot_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/mandelbrot_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-| [matrix_ops_dynamic_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/matrix_ops_dynamic_test.py) | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|  [matrix_ops_static_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/matrix_ops_static_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|   [quantization_dyn_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/quantization_dyn_test.py)   | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|       [quantization_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/quantization_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|              [range_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/range_test.py)              | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|       [resource_ops_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/resource_ops_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|        [ring_buffer_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/ring_buffer_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|     [scatter_update_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/scatter_update_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|  [simple_arithmetic_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/simple_arithmetic_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [simple_stateful_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/simple_stateful_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [sliding_window_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/sliding_window_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|  [space_to_batch_nd_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/space_to_batch_nd_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|            [strings_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/strings_test.py)            | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [tensorlist_test](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/tensorlist_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [batch_norm_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/batch_norm_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|  [batch_to_space_nd_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/batch_to_space_nd_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [broadcast_to_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/broadcast_to_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|       [broadcasting_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/broadcasting_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|             [concat_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/concat_test.py)             | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [control_flow_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/control_flow_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|               [conv_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/conv_test.py)               | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [conv_transpose_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/conv_transpose_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|         [depth_conv_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/depth_conv_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|   [dynamic_mlp_relu_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/dynamic_mlp_relu_test.py)   | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|        [dynamic_mlp_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/dynamic_mlp_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|     [einsum_dynamic_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/einsum_dynamic_test.py)     | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|      [einsum_static_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/einsum_static_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|      [einsum_vector_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/einsum_vector_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|                [fft_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/fft_test.py)                | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|               [fill_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/fill_test.py)               | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|             [gather_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/gather_test.py)             | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [image_resize_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/image_resize_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|           [linspace_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/linspace_test.py)           | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|         [mandelbrot_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/mandelbrot_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+| [matrix_ops_dynamic_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/matrix_ops_dynamic_test.py) | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|  [matrix_ops_static_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/matrix_ops_static_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|   [quantization_dyn_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/quantization_dyn_test.py)   | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|       [quantization_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/quantization_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|              [range_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/range_test.py)              | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|       [resource_ops_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/resource_ops_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|        [ring_buffer_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/ring_buffer_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|     [scatter_update_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/scatter_update_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|  [simple_arithmetic_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/simple_arithmetic_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [simple_stateful_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/simple_stateful_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [sliding_window_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/sliding_window_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|  [space_to_batch_nd_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/space_to_batch_nd_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|            [strings_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/strings_test.py)            | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [tensorlist_test](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/tensorlist_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
 
 ## End to end tests of tf.math functions with static dimensions
 
@@ -57,308 +57,308 @@
 
 |                                                       target                                                      |                    tflite                    |                     vmvx                     |                 vulkan-spirv                 |
 |:-----------------------------------------------------------------------------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|
-|           [abs](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|       [accumulate_n](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|           [acos](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|          [acosh](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [add](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [add_n](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [angle](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [argmax](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [argmin](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [asin](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [asinh](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [atan](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|          [atan2](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [atanh](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [bessel_i0](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [bessel_i0e](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [bessel_i1](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [bessel_i1e](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [betainc](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [bincount](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [ceil](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [confusion_matrix](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [cos](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|           [cosh](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|      [count_nonzero](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [cumprod](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [cumsum](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|   [cumulative_logsumexp](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [digamma](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [divide](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|      [divide_no_nan](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [equal](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|           [erf](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [erfc](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [erfinv](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [exp](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [expm1](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [floor](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [floordiv](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [floormod](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [greater](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|      [greater_equal](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [igamma](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [igammac](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [imag](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [in_top_k](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|    [invert_permutation](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [is_finite](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [is_inf](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [is_nan](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|    [is_non_decreasing](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|  [is_strictly_increasing](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py) | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [lbeta](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|           [less](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|        [less_equal](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [lgamma](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|           [log](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [log1p](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|       [log_sigmoid](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|       [log_softmax](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|       [logical_and](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|       [logical_not](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|        [logical_or](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|       [logical_xor](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|         [maximum](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [minimum](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|           [mod](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|         [multiply](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [multiply_no_nan](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [ndtri](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [negative](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|        [nextafter](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [not_equal](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|        [polygamma](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [polyval](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|           [pow](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [real](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|        [reciprocal](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [reciprocal_no_nan](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|        [reduce_all](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|        [reduce_any](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|  [reduce_euclidean_norm](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|     [reduce_logsumexp](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|        [reduce_max](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|       [reduce_mean](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|        [reduce_min](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|       [reduce_prod](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [reduce_std](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|        [reduce_sum](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|     [reduce_variance](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|           [rint](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [round](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [rsqrt](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|        [scalar_mul](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|       [segment_max](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [segment_mean](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [segment_min](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [segment_prod](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [segment_sum](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [sigmoid](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|           [sign](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [sin](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|           [sinh](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|       [sobol_sample](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [softmax](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [softplus](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [softsign](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [sqrt](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [square](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [squared_difference](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)   | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [subtract](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|           [tan](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|           [tanh](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [truediv](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|   [unsorted_segment_max](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|  [unsorted_segment_mean](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|   [unsorted_segment_min](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|  [unsorted_segment_prod](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-| [unsorted_segment_sqrt_n](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py) | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|   [unsorted_segment_sum](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [xdivy](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [xlog1py](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [xlogy](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|      [zero_fraction](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|           [zeta](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [abs](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [accumulate_n](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|           [acos](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|          [acosh](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [add](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [add_n](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [angle](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [argmax](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [argmin](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [asin](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [asinh](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [atan](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|          [atan2](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [atanh](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [bessel_i0](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [bessel_i0e](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [bessel_i1](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [bessel_i1e](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [betainc](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [bincount](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [ceil](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [confusion_matrix](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [cos](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|           [cosh](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|      [count_nonzero](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [cumprod](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [cumsum](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|   [cumulative_logsumexp](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [digamma](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [divide](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|      [divide_no_nan](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [equal](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|           [erf](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [erfc](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [erfinv](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [exp](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [expm1](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [floor](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [floordiv](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [floormod](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [greater](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|      [greater_equal](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [igamma](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [igammac](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [imag](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [in_top_k](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|    [invert_permutation](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [is_finite](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [is_inf](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [is_nan](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|    [is_non_decreasing](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|  [is_strictly_increasing](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py) | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [lbeta](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|           [less](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|        [less_equal](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [lgamma](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|           [log](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [log1p](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [log_sigmoid](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [log_softmax](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [logical_and](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|       [logical_not](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|        [logical_or](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|       [logical_xor](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|         [maximum](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [minimum](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|           [mod](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|         [multiply](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [multiply_no_nan](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [ndtri](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [negative](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|        [nextafter](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [not_equal](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|        [polygamma](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [polyval](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|           [pow](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [real](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|        [reciprocal](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [reciprocal_no_nan](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|        [reduce_all](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|        [reduce_any](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|  [reduce_euclidean_norm](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|     [reduce_logsumexp](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|        [reduce_max](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|       [reduce_mean](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|        [reduce_min](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|       [reduce_prod](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [reduce_std](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|        [reduce_sum](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|     [reduce_variance](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|           [rint](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [round](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [rsqrt](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|        [scalar_mul](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [segment_max](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [segment_mean](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [segment_min](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [segment_prod](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [segment_sum](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [sigmoid](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|           [sign](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [sin](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|           [sinh](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [sobol_sample](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [softmax](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [softplus](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [softsign](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [sqrt](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [square](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [squared_difference](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)   | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [subtract](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|           [tan](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|           [tanh](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [truediv](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|   [unsorted_segment_max](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|  [unsorted_segment_mean](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|   [unsorted_segment_min](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|  [unsorted_segment_prod](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+| [unsorted_segment_sqrt_n](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py) | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|   [unsorted_segment_sum](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [xdivy](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [xlog1py](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [xlogy](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|      [zero_fraction](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|           [zeta](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
 
 ## End to end tests of tf.math functions with dynamic dimensions
 
 |                                                       target                                                      |                    tflite                    |                     vmvx                     |                 vulkan-spirv                 |
 |:-----------------------------------------------------------------------------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|
-|           [abs](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|       [accumulate_n](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|           [acos](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|          [acosh](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [add](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|          [add_n](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|          [angle](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [argmax](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [argmin](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [asin](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [asinh](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [atan](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|          [atan2](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [atanh](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [bessel_i0](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [bessel_i0e](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [bessel_i1](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [bessel_i1e](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [betainc](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [bincount](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [ceil](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|     [confusion_matrix](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [cos](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|           [cosh](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|      [count_nonzero](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [cumprod](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [cumsum](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|   [cumulative_logsumexp](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [digamma](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [divide](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|      [divide_no_nan](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [equal](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|           [erf](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [erfc](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [erfinv](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [exp](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|          [expm1](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [floor](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|         [floordiv](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [floormod](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [greater](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|      [greater_equal](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|          [igamma](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [igammac](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [imag](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|         [in_top_k](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|    [invert_permutation](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [is_finite](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|          [is_inf](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|          [is_nan](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|    [is_non_decreasing](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|  [is_strictly_increasing](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py) | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [lbeta](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [less](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|        [less_equal](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|          [lgamma](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [log](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|          [log1p](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [log_sigmoid](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [log_softmax](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|       [logical_and](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [logical_not](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [logical_or](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [logical_xor](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [maximum](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|         [minimum](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|           [mod](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [multiply](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|     [multiply_no_nan](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)     | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [ndtri](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [negative](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|        [nextafter](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [not_equal](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|        [polygamma](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [polyval](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|           [pow](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [real](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|        [reciprocal](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|    [reciprocal_no_nan](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [reduce_all](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [reduce_any](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|  [reduce_euclidean_norm](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|     [reduce_logsumexp](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [reduce_max](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [reduce_mean](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [reduce_min](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [reduce_prod](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [reduce_std](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [reduce_sum](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|     [reduce_variance](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)     | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [rint](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [round](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [rsqrt](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|        [scalar_mul](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|       [segment_max](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [segment_mean](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [segment_min](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [segment_prod](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [segment_sum](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [sigmoid](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|           [sign](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [sin](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|           [sinh](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|       [sobol_sample](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [softmax](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|         [softplus](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [softsign](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [sqrt](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|          [square](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|    [squared_difference](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)   | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|         [subtract](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|           [tan](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|           [tanh](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|         [truediv](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|   [unsorted_segment_max](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|  [unsorted_segment_mean](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|   [unsorted_segment_min](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|  [unsorted_segment_prod](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-| [unsorted_segment_sqrt_n](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py) | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|   [unsorted_segment_sum](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [xdivy](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [xlog1py](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [xlogy](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|      [zero_fraction](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [zeta](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [abs](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|       [accumulate_n](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|           [acos](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|          [acosh](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [add](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|          [add_n](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|          [angle](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [argmax](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [argmin](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [asin](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [asinh](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [atan](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|          [atan2](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [atanh](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [bessel_i0](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [bessel_i0e](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [bessel_i1](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [bessel_i1e](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [betainc](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [bincount](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [ceil](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|     [confusion_matrix](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [cos](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|           [cosh](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|      [count_nonzero](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [cumprod](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [cumsum](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|   [cumulative_logsumexp](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [digamma](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [divide](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|      [divide_no_nan](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [equal](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|           [erf](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [erfc](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [erfinv](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [exp](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|          [expm1](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [floor](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|         [floordiv](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [floormod](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [greater](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|      [greater_equal](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|          [igamma](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [igammac](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [imag](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|         [in_top_k](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|    [invert_permutation](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [is_finite](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|          [is_inf](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|          [is_nan](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|    [is_non_decreasing](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|  [is_strictly_increasing](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py) | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [lbeta](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [less](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|        [less_equal](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|          [lgamma](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [log](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|          [log1p](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [log_sigmoid](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [log_softmax](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|       [logical_and](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [logical_not](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [logical_or](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [logical_xor](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [maximum](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|         [minimum](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|           [mod](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [multiply](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|     [multiply_no_nan](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)     | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [ndtri](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [negative](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|        [nextafter](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [not_equal](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|        [polygamma](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [polyval](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|           [pow](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [real](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|        [reciprocal](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|    [reciprocal_no_nan](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [reduce_all](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [reduce_any](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|  [reduce_euclidean_norm](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|     [reduce_logsumexp](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [reduce_max](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [reduce_mean](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [reduce_min](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [reduce_prod](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [reduce_std](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [reduce_sum](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|     [reduce_variance](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)     | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [rint](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [round](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [rsqrt](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|        [scalar_mul](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|       [segment_max](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [segment_mean](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [segment_min](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [segment_prod](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [segment_sum](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [sigmoid](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|           [sign](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [sin](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|           [sinh](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|       [sobol_sample](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [softmax](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|         [softplus](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [softsign](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [sqrt](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|          [square](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|    [squared_difference](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)   | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|         [subtract](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|           [tan](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)           | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|           [tanh](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|         [truediv](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|   [unsorted_segment_max](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|  [unsorted_segment_mean](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|   [unsorted_segment_min](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|  [unsorted_segment_prod](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+| [unsorted_segment_sqrt_n](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py) | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|   [unsorted_segment_sum](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [xdivy](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [xlog1py](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [xlogy](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|      [zero_fraction](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [zeta](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
 
 ## End to end tests of tf.math functions with complex numbers
 
 |                                                      target                                                     |                    tflite                    |                     vmvx                     |                 vulkan-spirv                 |
 |:---------------------------------------------------------------------------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|
-|          [abs](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [add](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [angle](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [asinh](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [atanh](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [conj](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [cos](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [cosh](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|     [count_nonzero](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)     | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [cumprod](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [cumsum](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [divide](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|     [divide_no_nan](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)     | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [exp](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [expm1](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [imag](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|      [l2_normalize](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|          [log](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [log1p](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [multiply](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [multiply_no_nan](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [negative](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [pow](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [real](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|       [reciprocal](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|   [reciprocal_no_nan](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-| [reduce_euclidean_norm](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py) | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [reduce_std](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|    [reduce_variance](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [rsqrt](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [sigmoid](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [sign](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [sin](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [sinh](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [sqrt](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [square](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|   [squared_difference](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|        [subtract](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [tan](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [tanh](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [truediv](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [xdivy](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [xlog1py](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [xlogy](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|     [zero_fraction](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)     | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [abs](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [add](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [angle](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [asinh](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [atanh](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [conj](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [cos](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [cosh](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|     [count_nonzero](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)     | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [cumprod](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [cumsum](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [divide](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|     [divide_no_nan](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)     | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [exp](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [expm1](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [imag](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|      [l2_normalize](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|          [log](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [log1p](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [multiply](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [multiply_no_nan](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [negative](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [pow](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [real](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [reciprocal](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|   [reciprocal_no_nan](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+| [reduce_euclidean_norm](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py) | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [reduce_std](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|    [reduce_variance](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [rsqrt](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [sigmoid](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [sign](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [sin](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [sinh](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [sqrt](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [square](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|   [squared_difference](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)  | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|        [subtract](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [tan](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [tanh](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [truediv](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [xdivy](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [xlog1py](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [xlogy](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|     [zero_fraction](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/math/math_test.py)     | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
diff --git a/docs/developers/tensorflow_coverage/tf_keras_coverage.md b/docs/developers/tensorflow_coverage/tf_keras_coverage.md
index 3af5b57..babf7a8 100644
--- a/docs/developers/tensorflow_coverage/tf_keras_coverage.md
+++ b/docs/developers/tensorflow_coverage/tf_keras_coverage.md
@@ -3,7 +3,7 @@
 Tests of `tf.keras.layers` compiled with static shapes, dynamic shapes and training enabled.
 
 IREE has three main backend
-[targets](https://github.com/google/iree/tree/main/iree/compiler/Dialect/HAL/Target):
+[targets](https://github.com/iree-org/iree/tree/main/iree/compiler/Dialect/HAL/Target):
 `vmvx` , `llvm` and `vulkan-spirv`. We also test TFLite in our infrastructure
 for benchmarking purposes.
 
@@ -20,168 +20,168 @@
 
 |                                                           target                                                           |                    tflite                    |                     vmvx                     |                 vulkan-spirv                 |
 |:--------------------------------------------------------------------------------------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|
-|       [Activation](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-| [ActivityRegularization](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|           [Add](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [AdditiveAttention](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|      [AlphaDropout](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|        [Attention](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|         [Average](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [AveragePooling1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|    [AveragePooling2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|    [AveragePooling3D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|   [BatchNormalization](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|       [Concatenate](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [Conv1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [Conv1DTranspose](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [Conv2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [Conv2DTranspose](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [Conv3D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> |
-|     [Conv3DTranspose](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> |
-|       [Cropping1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|       [Cropping2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|       [Cropping3D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [Dense](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [DepthwiseConv2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|           [Dot](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [Dropout](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|           [ELU](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|        [Embedding](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [Flatten](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|           [GRU](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|     [GaussianDropout](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|      [GaussianNoise](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-| [GlobalAveragePooling1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-| [GlobalAveragePooling2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-| [GlobalAveragePooling3D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [GlobalMaxPool1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [GlobalMaxPool2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [GlobalMaxPool3D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|       [InputLayer](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [LSTM](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [Lambda](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|   [LayerNormalization](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [LeakyReLU](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|   [LocallyConnected1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|   [LocallyConnected2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [Masking](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|        [MaxPool1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|        [MaxPool2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|        [MaxPool3D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|         [Maximum](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [Minimum](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|   [MultiHeadAttention](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [Multiply](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [PReLU](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [Permute](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [ReLU](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|      [RepeatVector](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [Reshape](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [SeparableConv1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [SeparableConv2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|         [Softmax](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [SpatialDropout1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [SpatialDropout2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [SpatialDropout3D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|        [Subtract](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [ThresholdedReLU](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|      [UpSampling1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|      [UpSampling2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|      [UpSampling3D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|      [ZeroPadding1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|      [ZeroPadding2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|      [ZeroPadding3D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [Activation](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+| [ActivityRegularization](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|           [Add](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [AdditiveAttention](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|      [AlphaDropout](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|        [Attention](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|         [Average](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [AveragePooling1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|    [AveragePooling2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|    [AveragePooling3D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|   [BatchNormalization](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [Concatenate](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [Conv1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [Conv1DTranspose](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [Conv2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [Conv2DTranspose](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [Conv3D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> |
+|     [Conv3DTranspose](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> |
+|       [Cropping1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [Cropping2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [Cropping3D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [Dense](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [DepthwiseConv2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|           [Dot](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [Dropout](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|           [ELU](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|        [Embedding](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [Flatten](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|           [GRU](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|     [GaussianDropout](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|      [GaussianNoise](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+| [GlobalAveragePooling1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+| [GlobalAveragePooling2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+| [GlobalAveragePooling3D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [GlobalMaxPool1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [GlobalMaxPool2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [GlobalMaxPool3D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [InputLayer](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [LSTM](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [Lambda](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|   [LayerNormalization](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [LeakyReLU](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|   [LocallyConnected1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|   [LocallyConnected2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [Masking](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|        [MaxPool1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|        [MaxPool2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|        [MaxPool3D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|         [Maximum](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [Minimum](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|   [MultiHeadAttention](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [Multiply](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [PReLU](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [Permute](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [ReLU](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|      [RepeatVector](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [Reshape](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [SeparableConv1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [SeparableConv2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|         [Softmax](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [SpatialDropout1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [SpatialDropout2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [SpatialDropout3D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|        [Subtract](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [ThresholdedReLU](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|      [UpSampling1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|      [UpSampling2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|      [UpSampling3D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|      [ZeroPadding1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|      [ZeroPadding2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|      [ZeroPadding3D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
 
 ## End to end tests of tf.keras layers with dynamic dimensions (with default configuration in inference mode)
 
 |                                                           target                                                           |                    tflite                    |                     vmvx                     |                 vulkan-spirv                 |
 |:--------------------------------------------------------------------------------------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|
-|       [Activation](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-| [ActivityRegularization](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|           [Add](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|    [AdditiveAttention](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|      [AlphaDropout](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|        [Attention](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|         [Average](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|    [AveragePooling1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|    [AveragePooling2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|    [AveragePooling3D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|   [BatchNormalization](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [Concatenate](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [Conv1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|     [Conv1DTranspose](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [Conv2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|     [Conv2DTranspose](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [Conv3D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|     [Conv3DTranspose](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [Cropping1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [Cropping2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [Cropping3D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|          [Dense](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|     [DepthwiseConv2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [Dot](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [Dropout](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|           [ELU](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [Embedding](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|         [Flatten](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|           [GRU](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|     [GaussianDropout](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|      [GaussianNoise](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-| [GlobalAveragePooling1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-| [GlobalAveragePooling2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-| [GlobalAveragePooling3D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|     [GlobalMaxPool1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [GlobalMaxPool2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [GlobalMaxPool3D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|       [InputLayer](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [LSTM](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [Lambda](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|   [LayerNormalization](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [LeakyReLU](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|   [LocallyConnected1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|   [LocallyConnected2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [Masking](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [MaxPool1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [MaxPool2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [MaxPool3D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [Maximum](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|         [Minimum](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|   [MultiHeadAttention](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [Multiply](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|          [PReLU](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|         [Permute](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|          [ReLU](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|      [RepeatVector](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [Reshape](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|     [SeparableConv1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|     [SeparableConv2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [Softmax](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|    [SpatialDropout1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [SpatialDropout2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [SpatialDropout3D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|        [Subtract](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|     [ThresholdedReLU](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|      [UpSampling1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|      [UpSampling2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|      [UpSampling3D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|      [ZeroPadding1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)     | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|      [ZeroPadding2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)     | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|      [ZeroPadding3D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)     | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [Activation](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+| [ActivityRegularization](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|           [Add](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|    [AdditiveAttention](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|      [AlphaDropout](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|        [Attention](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|         [Average](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|    [AveragePooling1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|    [AveragePooling2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|    [AveragePooling3D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|   [BatchNormalization](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [Concatenate](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [Conv1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|     [Conv1DTranspose](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [Conv2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|     [Conv2DTranspose](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [Conv3D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|     [Conv3DTranspose](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [Cropping1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [Cropping2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [Cropping3D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|          [Dense](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|     [DepthwiseConv2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [Dot](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [Dropout](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|           [ELU](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [Embedding](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|         [Flatten](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|           [GRU](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|     [GaussianDropout](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|      [GaussianNoise](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)     | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+| [GlobalAveragePooling1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+| [GlobalAveragePooling2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+| [GlobalAveragePooling3D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|     [GlobalMaxPool1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [GlobalMaxPool2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [GlobalMaxPool3D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [InputLayer](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [LSTM](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [Lambda](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|   [LayerNormalization](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [LeakyReLU](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|   [LocallyConnected1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|   [LocallyConnected2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [Masking](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [MaxPool1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [MaxPool2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [MaxPool3D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)       | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [Maximum](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|         [Minimum](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|   [MultiHeadAttention](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [Multiply](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|          [PReLU](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)         | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|         [Permute](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|          [ReLU](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)          | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|      [RepeatVector](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [Reshape](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|     [SeparableConv1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|     [SeparableConv2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [Softmax](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|    [SpatialDropout1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [SpatialDropout2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [SpatialDropout3D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|        [Subtract](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|     [ThresholdedReLU](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|      [UpSampling1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|      [UpSampling2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|      [UpSampling3D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|      [ZeroPadding1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)     | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|      [ZeroPadding2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)     | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|      [ZeroPadding3D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)     | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
 
 ## End to end tests of tf.keras layers in training mode (with default configuration and static dimensions)
 
 |                                                         target                                                         |                    tflite                    |                     vmvx                     |                 vulkan-spirv                 |
 |:----------------------------------------------------------------------------------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|
-|  [AdditiveAttention](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|    [AlphaDropout](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|      [Attention](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)     | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-| [BatchNormalization](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|       [Dropout](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|         [GRU](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|   [GaussianDropout](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|    [GaussianNoise](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|        [LSTM](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-| [MultiHeadAttention](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|  [SpatialDropout1D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)  | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|  [SpatialDropout2D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)  | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|  [SpatialDropout3D](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)  | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|  [AdditiveAttention](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|    [AlphaDropout](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)    | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|      [Attention](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)     | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+| [BatchNormalization](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|       [Dropout](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)      | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|         [GRU](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|   [GaussianDropout](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)  | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|    [GaussianNoise](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)   | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|        [LSTM](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)        | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+| [MultiHeadAttention](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py) | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|  [SpatialDropout1D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)  | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|  [SpatialDropout2D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)  | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|  [SpatialDropout3D](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/layers/layers_test.py)  | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
diff --git a/docs/developers/tensorflow_coverage/vision_coverage.md b/docs/developers/tensorflow_coverage/vision_coverage.md
index f27b3b0..ac12d7e 100644
--- a/docs/developers/tensorflow_coverage/vision_coverage.md
+++ b/docs/developers/tensorflow_coverage/vision_coverage.md
@@ -3,7 +3,7 @@
 Tests of Keras and Slim vision models.
 
 IREE has three main backend
-[targets](https://github.com/google/iree/tree/main/iree/compiler/Dialect/HAL/Target):
+[targets](https://github.com/iree-org/iree/tree/main/iree/compiler/Dialect/HAL/Target):
 `vmvx` , `llvm` and `vulkan-spirv`. We also test TFLite in our infrastructure
 for benchmarking purposes.
 
@@ -13,48 +13,48 @@
 
 |                                                               target                                                              |                    tflite                    |                     vmvx                     |                 vulkan-spirv                 |
 |:---------------------------------------------------------------------------------------------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|
-|    [DenseNet121](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [DenseNet169](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [DenseNet201](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|   [EfficientNetB0](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|   [EfficientNetB1](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|   [EfficientNetB2](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|   [EfficientNetB3](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|   [EfficientNetB4](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|   [EfficientNetB5](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|   [EfficientNetB6](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|   [EfficientNetB7](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-| [InceptionResNetV2](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|    [InceptionV3](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|     [MobileNet](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [MobileNetV2](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|  [MobileNetV3Large](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|  [MobileNetV3Small](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [NASNetLarge](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|    [NASNetMobile](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)   | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|     [ResNet101](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [ResNet101V2](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|     [ResNet152](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [ResNet152V2](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|      [ResNet50](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [ResNet50V2](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|       [VGG16](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|       [VGG19](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [DenseNet121](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [DenseNet169](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [DenseNet201](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|   [EfficientNetB0](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|   [EfficientNetB1](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|   [EfficientNetB2](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|   [EfficientNetB3](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|   [EfficientNetB4](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|   [EfficientNetB5](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|   [EfficientNetB6](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|   [EfficientNetB7](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)  | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+| [InceptionResNetV2](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|    [InceptionV3](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|     [MobileNet](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [MobileNetV2](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|  [MobileNetV3Large](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|  [MobileNetV3Small](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [NASNetLarge](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|    [NASNetMobile](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)   | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|     [ResNet101](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [ResNet101V2](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|     [ResNet152](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [ResNet152V2](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|      [ResNet50](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)     | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [ResNet50V2](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|       [VGG16](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|       [VGG19](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/keras/applications/vision_model_test.py)       | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
 
 ## End to end tests of TensorFlow slim vision models
 
 |                                                                  target                                                                  |                    tflite                    |                     vmvx                     |                 vulkan-spirv                 |
 |:----------------------------------------------------------------------------------------------------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|:--------------------------------------------:|
-| [inception_resnet_v2](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|     [inception_v1](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|     [inception_v2](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|     [inception_v3](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|     [nasnet_large](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|    [nasnet_mobile](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
-|    [pnasnet_large](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|    [resnet_v1_101](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [resnet_v1_152](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|     [resnet_v1_50](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
-|    [resnet_v2_101](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|    [resnet_v2_152](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
-|     [resnet_v2_50](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+| [inception_resnet_v2](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py) | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|     [inception_v1](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|     [inception_v2](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|     [inception_v3](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|     [nasnet_large](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|    [nasnet_mobile](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> | <span class="failure-table-element">✗</span> |
+|    [pnasnet_large](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|    [resnet_v1_101](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [resnet_v1_152](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|     [resnet_v1_50](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> |
+|    [resnet_v2_101](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|    [resnet_v2_152](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
+|     [resnet_v2_50](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py)    | <span class="success-table-element">✓</span> | <span class="success-table-element">✓</span> | <span class="failure-table-element">✗</span> |
diff --git a/docs/website/docs/bindings/c-api.md b/docs/website/docs/bindings/c-api.md
index 8811d56..cd3e515 100644
--- a/docs/website/docs/bindings/c-api.md
+++ b/docs/website/docs/bindings/c-api.md
@@ -7,11 +7,11 @@
 
 | Component header file                                                       | Overview                                                                  |
 |-----------------------------------------------------------------------------|---------------------------------------------------------------------------|
-| [iree/base/api.h](https://github.com/google/iree/blob/main/runtime/src/iree/base/api.h) | Core API, type definitions, ownership policies, utilities                 |
-| [iree/vm/api.h](https://github.com/google/iree/blob/main/runtime/src/iree/vm/api.h)     | VM APIs: loading modules, I/O, calling functions                          |
-| [iree/hal/api.h](https://github.com/google/iree/blob/main/runtime/src/iree/hal/api.h)   | HAL APIs: device management, synchronization, accessing hardware features |
+| [iree/base/api.h](https://github.com/iree-org/iree/blob/main/runtime/src/iree/base/api.h) | Core API, type definitions, ownership policies, utilities                 |
+| [iree/vm/api.h](https://github.com/iree-org/iree/blob/main/runtime/src/iree/vm/api.h)     | VM APIs: loading modules, I/O, calling functions                          |
+| [iree/hal/api.h](https://github.com/iree-org/iree/blob/main/runtime/src/iree/hal/api.h)   | HAL APIs: device management, synchronization, accessing hardware features |
 
-The [samples/](https://github.com/google/iree/tree/main/samples)
+The [samples/](https://github.com/iree-org/iree/tree/main/samples)
 directory demonstrates several ways to use IREE's C API.
 
 ## Prerequisites
@@ -151,7 +151,7 @@
 
 !!! note
     Many IREE samples use
-    [`c_embed_data`](https://github.com/google/iree/tree/main/build_tools/embed_data)
+    [`c_embed_data`](https://github.com/iree-org/iree/tree/main/build_tools/embed_data)
     to embed vmfb files as C code to avoid file I/O and ease portability.
     Applications should use what makes sense for their platforms and deployment
     configurations.
@@ -215,11 +215,11 @@
 
 [^1]:
   We are exploring adding a C API for IREE's compiler, see
-  [this GitHub issue](https://github.com/google/iree/issues/3817)
+  [this GitHub issue](https://github.com/iree-org/iree/issues/3817)
 
 [^2]:
   We plan on deploying via [vcpkg](https://github.com/microsoft/vcpkg) in the
   future too, see
-  [this GitHub project](https://github.com/google/iree/projects/18)
+  [this GitHub project](https://github.com/iree-org/iree/projects/18)
 
 *[vmfb]: VM FlatBuffer
diff --git a/docs/website/docs/bindings/python.md b/docs/website/docs/bindings/python.md
index 9fc16b7..f4c6df5 100644
--- a/docs/website/docs/bindings/python.md
+++ b/docs/website/docs/bindings/python.md
@@ -25,7 +25,7 @@
 !!! warning
     The TensorFlow, TensorFlow Lite, and XLA packages are currently only
     available on Linux and macOS. They are not available on Windows yet (see
-    [this issue](https://github.com/google/iree/issues/6417)).
+    [this issue](https://github.com/iree-org/iree/issues/6417)).
 
 ## Prerequisites
 
@@ -97,11 +97,11 @@
 !!! Tip
 
     Nightly packages are also published on
-    [GitHub releases](https://github.com/google/iree/releases). To use these,
+    [GitHub releases](https://github.com/iree-org/iree/releases). To use these,
     run `pip install` with this extra option:
 
     ```
-    --find-links https://github.com/google/iree/releases
+    --find-links https://github.com/iree-org/iree/releases
     ```
 
 ### Building from source
@@ -115,8 +115,8 @@
 [readthedocs](https://iree-python-api.readthedocs.io/en/latest/).
 
 Check out the samples in IREE's
-[samples/colab/ directory](https://github.com/google/iree/tree/main/samples/colab)
-and the [iree-samples repository](https://github.com/google/iree-samples) for
+[samples/colab/ directory](https://github.com/iree-org/iree/tree/main/samples/colab)
+and the [iree-samples repository](https://github.com/iree-org/iree-samples) for
 examples using the Python APIs.
 
 <!-- ## Troubleshooting -->
diff --git a/docs/website/docs/bindings/tensorflow-lite.md b/docs/website/docs/bindings/tensorflow-lite.md
index db4c4bf..cd309d5 100644
--- a/docs/website/docs/bindings/tensorflow-lite.md
+++ b/docs/website/docs/bindings/tensorflow-lite.md
@@ -1,7 +1,7 @@
 # TensorFlow Lite bindings
 
 !!! todo
-    [Issue#5462](https://github.com/google/iree/issues/5462): write this documentation
+    [Issue#5462](https://github.com/iree-org/iree/issues/5462): write this documentation
 
 <!-- TODO(??): overview, advantages/disadvantages to using TFLite bindings -->
 
diff --git a/docs/website/docs/blog/2021-07-19-tflite-tosa.md b/docs/website/docs/blog/2021-07-19-tflite-tosa.md
index a97ae10..b64c9d6 100644
--- a/docs/website/docs/blog/2021-07-19-tflite-tosa.md
+++ b/docs/website/docs/blog/2021-07-19-tflite-tosa.md
@@ -34,7 +34,7 @@
 ## Examples
 
 TFLite with IREE is available in Python and Java.  We have a
-[colab notebook](https://colab.research.google.com/github/google/iree/blob/main/samples/colab/tflite_text_classification.ipynb)
+[colab notebook](https://colab.research.google.com/github/iree-org/iree/blob/main/samples/colab/tflite_text_classification.ipynb)
 that shows how to use IREE’s python bindings and TFLite compiler tools to
 compile a pre-trained TFLite model from a FlatBuffer and run using IREE.  We
 also have an
diff --git a/docs/website/docs/blog/2021-10-15-cuda-backend.md b/docs/website/docs/blog/2021-10-15-cuda-backend.md
index b70bf88..7396e94 100644
--- a/docs/website/docs/blog/2021-10-15-cuda-backend.md
+++ b/docs/website/docs/blog/2021-10-15-cuda-backend.md
@@ -7,7 +7,7 @@
 IREE is being designed with re-targetability as a core goal: it should be
 possible to use IREE to target a broad spectrum of power regimes, from embedded
 systems to distributed clusters; and it should be possible to extend IREE to
-target new back-ends without having to reinvent the wheel each time. 
+target new back-ends without having to reinvent the wheel each time.
 
 To explore this, we recently branched out from our initial focus on low-latency
 mobile deployments with a goal of using IREE to target data center workloads on
@@ -19,7 +19,7 @@
 
 ### HAL support
 
-IREE has a [HAL API](https://github.com/google/iree/blob/main/docs/developers/design_roadmap.md#hal-hardware-abstraction-layer-and-multi-architecture-executables)
+IREE has a [HAL API](https://github.com/iree-org/iree/blob/main/docs/developers/design_roadmap.md#hal-hardware-abstraction-layer-and-multi-architecture-executables)
 that abstract all the targets behind a common interface. The first step to
 supporting a CUDA target was to map the HAL API onto CUDA. We use the CUDA
 driver API to reduce dependencies and be closer to the hardware. The HAL API is
@@ -32,7 +32,7 @@
 
 HAL exposes an API that can be tested independently, even if we are not able to
 create CUDA kernels yet we can test a large portion of the CUDA driver using
-[CTS tests](https://github.com/google/iree/blob/main/iree/hal/cts/README.md).
+[CTS tests](https://github.com/iree-org/iree/blob/main/iree/hal/cts/README.md).
 Those can be run to make sure a system has the required CUDA support.
 
  ![Compilation flow](./2021-10-15-cuda-compiler-flow.png){ align=left }
@@ -86,7 +86,7 @@
 ![Compilation diagram](./2021-10-15-cuda-bring_up.png)
 
 The steps to reproduce running a simple op end to end through CUDA backend are
-described [here](https://github.com/google/iree/blob/main/docs/developers/design_docs/cuda_backend.md#example).
+described [here](https://github.com/iree-org/iree/blob/main/docs/developers/design_docs/cuda_backend.md#example).
 
 ## Performance
 
@@ -104,7 +104,7 @@
 
 At the beginning of the code generation we look at the dispatch region and
 decide on the tile size for a workgroup. For CUDA we also decide the number of
-threads per block. 
+threads per block.
 We will then have a pass tiling the ops in the dispatch region a second time to
 distribute the work onto threads within the block.
 
@@ -119,8 +119,8 @@
         indexing_maps = [affine_map<(d0) -> (d0)>,
                          affine_map<(d0) -> (d0)>,
                          affine_map<(d0) -> (d0)>],
-        iterator_types = ["parallel"]} 
-      ins(%10, %11 : 
+        iterator_types = ["parallel"]}
+      ins(%10, %11 :
           memref<4xf32, affine_map<(d0)[s0] -> (d0 + s0)>>,
           memref<4xf32, affine_map<(d0)[s0] -> (d0 + s0)>>)
       outs(%12 : memref<4xf32, affine_map<(d0)[s0] -> (d0 + s0)>>) {
@@ -158,7 +158,7 @@
 cases where we may be memory bound and have the potential to re-use memory
 reads.
 
-For operations like GEMM using shared memory gives us a significant speed up. 
+For operations like GEMM using shared memory gives us a significant speed up.
 We leverage memory promotion, vector distribution and software pipelining
 transformations from MLIR to generate efficient copies from global to shared
 memory that can be interleaved with the compute work.
@@ -176,7 +176,7 @@
 ### GEMM
 
 We compare the performance of a single GEMM operation to highly optimized
-library cuBLAS using [mmperf framework](https://github.com/mmperf/mmperf). 
+library cuBLAS using [mmperf framework](https://github.com/mmperf/mmperf).
 
 ![Compilation diagram](./2021-10-15-cuda-memperf.png)
 
diff --git a/docs/website/docs/building-from-source/getting-started.md b/docs/website/docs/building-from-source/getting-started.md
index 4756132..17b9174 100644
--- a/docs/website/docs/building-from-source/getting-started.md
+++ b/docs/website/docs/building-from-source/getting-started.md
@@ -51,7 +51,7 @@
 submodules:
 
 ``` shell
-git clone https://github.com/google/iree.git
+git clone https://github.com/iree-org/iree.git
 cd iree
 git submodule update --init
 ```
diff --git a/docs/website/docs/building-from-source/index.md b/docs/website/docs/building-from-source/index.md
index eb919ea..fbf91d8 100644
--- a/docs/website/docs/building-from-source/index.md
+++ b/docs/website/docs/building-from-source/index.md
@@ -1,7 +1,7 @@
 # Building IREE from source
 
 While IREE does offer
-[binary distributions](https://github.com/google/iree/releases) for its
+[binary distributions](https://github.com/iree-org/iree/releases) for its
 compiler tools and [Python bindings](../bindings/python.md), building from
 source is still useful when using IREE's runtime or when making changes to the
 compiler or import tools themselves.
diff --git a/docs/website/docs/building-from-source/python-bindings-and-importers.md b/docs/website/docs/building-from-source/python-bindings-and-importers.md
index 0147959..c48d374 100644
--- a/docs/website/docs/building-from-source/python-bindings-and-importers.md
+++ b/docs/website/docs/building-from-source/python-bindings-and-importers.md
@@ -28,7 +28,7 @@
 * A relatively recent Python3 installation >=3.7 (we aim to support
   [non-eol Python versions](https://endoflife.date/python)).
 * Installation of python dependencies as specified in
-  [`runtime/bindings/python/iree/runtime/build_requirements.txt`](https://github.com/google/iree/blob/main/runtime/bindings/python/iree/runtime/build_requirements.txt).
+  [`runtime/bindings/python/iree/runtime/build_requirements.txt`](https://github.com/iree-org/iree/blob/main/runtime/bindings/python/iree/runtime/build_requirements.txt).
 
 **CMake Variables:**
 
@@ -158,7 +158,7 @@
 !!! Warning
 
     This section is under construction. Refer to the
-    [source documentation](https://github.com/google/iree/tree/main/integrations/tensorflow#readme)
+    [source documentation](https://github.com/iree-org/iree/tree/main/integrations/tensorflow#readme)
     for the latest building from source instructions.
 
 ???+ Note
diff --git a/docs/website/docs/building-from-source/riscv.md b/docs/website/docs/building-from-source/riscv.md
index 66c5e94..1db06de 100644
--- a/docs/website/docs/building-from-source/riscv.md
+++ b/docs/website/docs/building-from-source/riscv.md
@@ -68,7 +68,7 @@
 
 The following instruction shows how to build for a RISC-V 64-bit Linux machine.
 For other RISC-V targets, please refer to
-[riscv.toolchain.cmake](https://github.com/google/iree/blob/main/build_tools/cmake/riscv.toolchain.cmake)
+[riscv.toolchain.cmake](https://github.com/iree-org/iree/blob/main/build_tools/cmake/riscv.toolchain.cmake)
 as a reference of how to set up the cmake configuration.
 
 #### RISC-V 64-bit Linux target
@@ -89,7 +89,7 @@
 !!! note
     The following instructions are meant for the RISC-V 64-bit Linux
     target. For the bare-metal target, please refer to
-    [simple_embedding](https://github.com/google/iree/blob/main/samples/simple_embedding)
+    [simple_embedding](https://github.com/iree-org/iree/blob/main/samples/simple_embedding)
     to see how to build a ML workload for a bare-metal machine.
 
 Set the path to qemu-riscv64 Linux emulator binary in the `QEMU_BIN` environment
diff --git a/docs/website/docs/community/index.md b/docs/website/docs/community/index.md
index d8954e9..a5d7c96 100644
--- a/docs/website/docs/community/index.md
+++ b/docs/website/docs/community/index.md
@@ -12,7 +12,7 @@
     The project demonstrates the usage of runtime support and how to use a
     custom dialect alongside with the runtime.
 
-*   The [IREE LLVM Sandbox](https://github.com/google/iree-llvm-sandbox)
+*   The [IREE LLVM Sandbox](https://github.com/iree-org/iree-llvm-sandbox)
     contains experimental work by the IREE team closely related to LLVM and
     MLIR, usually with the aim of contributing back to those upstream projects
     in some form.
diff --git a/docs/website/docs/deployment-configurations/bare-metal.md b/docs/website/docs/deployment-configurations/bare-metal.md
index c4896af..b58e985 100644
--- a/docs/website/docs/deployment-configurations/bare-metal.md
+++ b/docs/website/docs/deployment-configurations/bare-metal.md
@@ -46,10 +46,10 @@
     library CPU HAL driver
 *   `iree-llvm-target-triple`: Use the `<arch>-pc-linux-elf` LLVM target triple
     so the artifact has a fixed ABI to be rendered by the
-    [elf_module library](https://github.com/google/iree/tree/main/iree/hal/local/elf)
+    [elf_module library](https://github.com/iree-org/iree/tree/main/iree/hal/local/elf)
 *   `iree-llvm-debug-symbols=false`: To reduce the artifact size
 
-See [generate.sh](https://github.com/google/iree/blob/main/iree/hal/local/elf/testdata/generate.sh)
+See [generate.sh](https://github.com/iree-org/iree/blob/main/iree/hal/local/elf/testdata/generate.sh)
 for example command-line instructions of some common architectures
 
 You can replace the MLIR file with the other MLIR model files, following the
@@ -57,7 +57,7 @@
 
 ### Compiling the bare-metal model for static-library support
 
-See the [static_library](https://github.com/google/iree/tree/main/samples/static_library)
+See the [static_library](https://github.com/iree-org/iree/tree/main/samples/static_library)
 demo sample for an example and instructions on running a model with IREE's
 `static_library_loader`.
 
@@ -87,11 +87,11 @@
 * `set(IREE_BUILD_TESTS OFF)`: Disable tests until IREE supports running them on
 bare-metal platforms
 * `set(IREE_BUILD_SAMPLES ON)`: Build
-[simple_embedding](https://github.com/google/iree/tree/main/samples/simple_embedding)
+[simple_embedding](https://github.com/iree-org/iree/tree/main/samples/simple_embedding)
 example
 
 !!! todo
-    Clean the list up after [#6353](https://github.com/google/iree/issues/6353)
+    Clean the list up after [#6353](https://github.com/iree-org/iree/issues/6353)
     is fixed.
 
 Also, set the toolchain-specific cmake file to match the tool path, target
@@ -113,13 +113,13 @@
 
 Examples of how to setup the CMakeLists.txt and .cmake file:
 
-* [IREE RISC-V toolchain cmake](https://github.com/google/iree/blob/main/build_tools/cmake/riscv.toolchain.cmake)
+* [IREE RISC-V toolchain cmake](https://github.com/iree-org/iree/blob/main/build_tools/cmake/riscv.toolchain.cmake)
 * [IREE Bare-Metal Arm Sample](https://github.com/iml130/iree-bare-metal-arm)
 * [IREE Bare-Metal RV32 Sample](https://github.com/AmbiML/iree-rv32-springbok)
 
 ## Bare-metal execution example
 
 See
-[simple_embedding for generic platform](https://github.com/google/iree/blob/main/samples/simple_embedding/README.md#generic-platform-support)
+[simple_embedding for generic platform](https://github.com/iree-org/iree/blob/main/samples/simple_embedding/README.md#generic-platform-support)
 to see how to use the IREE runtime library to build/run the IREE model for the
 bare-metal target.
diff --git a/docs/website/docs/extensions/index.md b/docs/website/docs/extensions/index.md
index 2f04f26..9f08dcb 100644
--- a/docs/website/docs/extensions/index.md
+++ b/docs/website/docs/extensions/index.md
@@ -235,11 +235,11 @@
 
 The runtime portion requires that the code be exported to the VM system by way
 of an `iree_vm_module_t` interface. A low-level native interface exists with
-minimal overhead and is used for example [by the IREE HAL itself](https://github.com/google/iree/tree/main/iree/modules/hal).
+minimal overhead and is used for example [by the IREE HAL itself](https://github.com/iree-org/iree/tree/main/iree/modules/hal).
 There is also a C++ wrapper that is significantly easier to work with however it
 needs some performance improvements.
 
-A full end-to-end example can be found under [`samples/custom_modules/`](https://github.com/google/iree/tree/main/samples/custom_modules),
+A full end-to-end example can be found under [`samples/custom_modules/`](https://github.com/iree-org/iree/tree/main/samples/custom_modules),
 though it should not currently be considered representative of best practices.
 
 ## 3. Extend target-specific device conversion patterns
@@ -483,6 +483,6 @@
 marshal arguments and results.
 
 The compiler-side needs some additional work but an example is included here:
-[Issue 7504](https://github.com/google/iree/issues/7504).
+[Issue 7504](https://github.com/iree-org/iree/issues/7504).
 The runtime-side is complete and resolution is performed by a user-supplied
 `iree_hal_executable_import_provider_t`.
diff --git a/docs/website/docs/getting-started/index.md b/docs/website/docs/getting-started/index.md
index 203f70a..31b2280 100644
--- a/docs/website/docs/getting-started/index.md
+++ b/docs/website/docs/getting-started/index.md
@@ -28,8 +28,8 @@
 ## Samples
 
 Check out the samples in IREE's
-[samples/colab/ directory](https://github.com/google/iree/tree/main/colab),
-as well as the [iree-samples repository](https://github.com/google/iree-samples),
+[samples/colab/ directory](https://github.com/iree-org/iree/tree/main/colab),
+as well as the [iree-samples repository](https://github.com/iree-org/iree-samples),
 which contains workflow comparisons across frameworks.
 
 ## Import
diff --git a/docs/website/docs/getting-started/jax.md b/docs/website/docs/getting-started/jax.md
index d0ec737..af1a8b0 100644
--- a/docs/website/docs/getting-started/jax.md
+++ b/docs/website/docs/getting-started/jax.md
@@ -1,7 +1,7 @@
 # JAX Integration
 
 !!! todo
-    [Issue#5454](https://github.com/google/iree/issues/5454): write this documentation
+    [Issue#5454](https://github.com/iree-org/iree/issues/5454): write this documentation
 
 <!-- TODO(??): overview (mention XLA? numpy?) -->
 
diff --git a/docs/website/docs/getting-started/tensorflow.md b/docs/website/docs/getting-started/tensorflow.md
index 3a0e118..8ab15d0 100644
--- a/docs/website/docs/getting-started/tensorflow.md
+++ b/docs/website/docs/getting-started/tensorflow.md
@@ -29,7 +29,7 @@
 !!! warning
     The TensorFlow package is currently only available on Linux and macOS. It
     is not available on Windows yet (see
-    [this issue](https://github.com/google/iree/issues/6417)).
+    [this issue](https://github.com/iree-org/iree/issues/6417)).
 
 ## Importing models
 
@@ -97,13 +97,13 @@
 
 | Colab notebooks |  |
 | -- | -- |
-Training an MNIST digits classifier | [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/iree/blob/main/samples/colab/mnist_training.ipynb)
-Edge detection module | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/iree/blob/main/samples/colab/edge_detection.ipynb)
-Pretrained ResNet50 inference | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/iree/blob/main/samples/colab/resnet.ipynb)
-TensorFlow Hub Import | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/iree/blob/main/samples/colab/tensorflow_hub_import.ipynb)
+Training an MNIST digits classifier | [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/iree-org/iree/blob/main/samples/colab/mnist_training.ipynb)
+Edge detection module | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/iree-org/iree/blob/main/samples/colab/edge_detection.ipynb)
+Pretrained ResNet50 inference | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/iree-org/iree/blob/main/samples/colab/resnet.ipynb)
+TensorFlow Hub Import | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/iree-org/iree/blob/main/samples/colab/tensorflow_hub_import.ipynb)
 
 End-to-end execution tests can be found in IREE's
-[integrations/tensorflow/e2e/](https://github.com/google/iree/tree/main/integrations/tensorflow/e2e)
+[integrations/tensorflow/e2e/](https://github.com/iree-org/iree/tree/main/integrations/tensorflow/e2e)
 directory.
 
 ## Troubleshooting
diff --git a/docs/website/docs/getting-started/tflite.md b/docs/website/docs/getting-started/tflite.md
index 919ac67..5ab8424 100644
--- a/docs/website/docs/getting-started/tflite.md
+++ b/docs/website/docs/getting-started/tflite.md
@@ -133,30 +133,30 @@
 TensorFlow Lite's operations to TOSA, the intermediate representation used by
 IREE. Many TensorFlow Lite operations are not fully supported, particularly
 those than use dynamic shapes. File an issue to IREE's TFLite model support
-[project](https://github.com/google/iree/projects/42).
+[project](https://github.com/iree-org/iree/projects/42).
 
 
 ## Additional Samples
 
 * The
-[tflitehub folder](https://github.com/google/iree-samples/tree/main/tflitehub)
-in the [iree-samples repository](https://github.com/google/iree-samples)
+[tflitehub folder](https://github.com/iree-org/iree-samples/tree/main/tflitehub)
+in the [iree-samples repository](https://github.com/iree-org/iree-samples)
 contains test scripts to compile, run, and compare various TensorFlow Lite
 models sourced from [TensorFlow Hub](https://tfhub.dev/).
 
 * An example smoke test of the
-[TensorFlow Lite C API](https://github.com/google/iree/tree/main/runtime/bindings/tflite)
+[TensorFlow Lite C API](https://github.com/iree-org/iree/tree/main/runtime/bindings/tflite)
 is available
-[here](https://github.com/google/iree/blob/main/runtime/bindings/tflite/smoke_test.cc).
+[here](https://github.com/iree-org/iree/blob/main/runtime/bindings/tflite/smoke_test.cc).
 
 | Colab notebooks |  |
 | -- | -- |
-Text classification with TFLite and IREE | [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/iree/blob/main/samples/colab/tflite_text_classification.ipynb)
+Text classification with TFLite and IREE | [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/iree-org/iree/blob/main/samples/colab/tflite_text_classification.ipynb)
 
 !!! todo
 
-    [Issue#3954](https://github.com/google/iree/issues/3954): Add documentation
+    [Issue#3954](https://github.com/iree-org/iree/issues/3954): Add documentation
     for an Android demo using the
-    [Java TFLite bindings](https://github.com/google/iree/tree/main/runtime/bindings/tflite/java),
+    [Java TFLite bindings](https://github.com/iree-org/iree/tree/main/runtime/bindings/tflite/java),
     once it is complete at
     [not-jenni/iree-android-tflite-demo](https://github.com/not-jenni/iree-android-tflite-demo).
diff --git a/docs/website/docs/index.md b/docs/website/docs/index.md
index b530da0..b8e8566 100644
--- a/docs/website/docs/index.md
+++ b/docs/website/docs/index.md
@@ -136,7 +136,7 @@
 ## Communication channels
 
 *   :fontawesome-brands-github:
-    [GitHub issues](https://github.com/google/iree/issues): Feature requests,
+    [GitHub issues](https://github.com/iree-org/iree/issues): Feature requests,
     bugs, and other work tracking
 *   :fontawesome-brands-discord:
     [IREE Discord server](https://discord.gg/26P4xW4): Daily development
@@ -148,17 +148,17 @@
 
 IREE is in the early stages of development and is not yet ready for broad
 adoption. Check out the
-[long-term design roadmap](https://github.com/google/iree/blob/main/docs/developers/design_roadmap.md)
+[long-term design roadmap](https://github.com/iree-org/iree/blob/main/docs/developers/design_roadmap.md)
 to get a sense of where we're headed.
 
 We plan on a quarterly basis using [OKRs](https://en.wikipedia.org/wiki/OKR).
 Review our latest
-[objectives](https://github.com/google/iree/blob/main/docs/developers/objectives.md) to
+[objectives](https://github.com/iree-org/iree/blob/main/docs/developers/objectives.md) to
 see what we're up to.
 
-We use [GitHub Projects](https://github.com/google/iree/projects) to track
+We use [GitHub Projects](https://github.com/iree-org/iree/projects) to track
 progress on IREE components and specific efforts and
-[GitHub Milestones](https://github.com/google/iree/milestones) to track the
+[GitHub Milestones](https://github.com/iree-org/iree/milestones) to track the
 work associated with plans for each quarter.
 
 [^1]:
diff --git a/docs/website/mkdocs.yml b/docs/website/mkdocs.yml
index a8fa9bd..9094147 100644
--- a/docs/website/mkdocs.yml
+++ b/docs/website/mkdocs.yml
@@ -1,5 +1,5 @@
 site_name: IREE
-site_url: https://google.github.io/iree/
+site_url: https://iree-org.github.io/iree/
 
 theme:
   name: material
@@ -45,14 +45,14 @@
         icon: material/toggle-switch
         name: Switch to light mode
 
-repo_url: https://github.com/google/iree
-repo_name: google/iree
+repo_url: https://github.com/iree-org/iree
+repo_name: iree-org/iree
 edit_uri: ""  # Omit edit button (linking to GitHub) from each document
 
 extra:
   social:
     - icon: fontawesome/brands/github
-      link: https://github.com/google/iree
+      link: https://github.com/iree-org/iree
       name: IREE on GitHub
     - icon: fontawesome/brands/discord
       link: https://discord.gg/26P4xW4
diff --git a/docs/website/overrides/404.html b/docs/website/overrides/404.html
index bcd1da4..47ed00f 100644
--- a/docs/website/overrides/404.html
+++ b/docs/website/overrides/404.html
@@ -1,14 +1,14 @@
 {% extends "main.html" %}
 {% block content %}
 <h1>
-  <img alt="👻" class="twemoji" src="https://google.github.io/iree/ghost.svg" title=":ghost:">
+  <img alt="👻" class="twemoji" src="https://iree-org.github.io/iree/ghost.svg" title=":ghost:">
   404 - Not found
-  <img alt="👻" class="twemoji" src="https://google.github.io/iree/ghost.svg" title=":ghost:">
+  <img alt="👻" class="twemoji" src="https://iree-org.github.io/iree/ghost.svg" title=":ghost:">
 </h1>
 
 <p>Sorry, we couldn't find that page.</p>
 
-<p>The <a href="https://github.com/google/iree/tree/main/docs/developers"><code>docs/developers/</code></a> directory on GitHub might be helpful.
+<p>The <a href="https://github.com/iree-org/iree/tree/main/docs/developers"><code>docs/developers/</code></a> directory on GitHub might be helpful.
 
 <p>Click <a href="{{ config.site_url }}">here</a> to go back to the home page.</p>
 
diff --git a/experimental/web/generate_web_metrics.sh b/experimental/web/generate_web_metrics.sh
index 880c2a4..6c73a01 100644
--- a/experimental/web/generate_web_metrics.sh
+++ b/experimental/web/generate_web_metrics.sh
@@ -66,7 +66,7 @@
 # specific version when iterating on metrics is useful, and fetching is slow.
 
 python -m pip install --upgrade \
-  --find-links https://github.com/google/iree/releases \
+  --find-links https://github.com/iree-org/iree/releases \
   iree-compiler iree-tools-tflite iree-tools-xla
 
 ###############################################################################
diff --git a/experimental/web/sample_dynamic/index.html b/experimental/web/sample_dynamic/index.html
index 76a68fb..fef4d47 100644
--- a/experimental/web/sample_dynamic/index.html
+++ b/experimental/web/sample_dynamic/index.html
@@ -52,8 +52,8 @@
 
     <p>
       This tool works similarly to
-      <a href="https://github.com/google/iree/blob/main/tools/iree-run-module-main.cc"><code>iree-run-module</code></a>
-      (<a href="https://github.com/google/iree/blob/main/docs/developers/developing_iree/developer_overview.md#iree-run-module">docs</a>).
+      <a href="https://github.com/iree-org/iree/blob/main/tools/iree-run-module-main.cc"><code>iree-run-module</code></a>
+      (<a href="https://github.com/iree-org/iree/blob/main/docs/developers/developing_iree/developer_overview.md#iree-run-module">docs</a>).
       <br>It loads a compiled IREE program then lets you call exported functions.
       <br><b>Note:</b> Some outputs are logged to the console.</p>
     </p>
@@ -122,7 +122,7 @@
       <div class="row" style="padding:4px">
         <div class="col-sm">
           simple_abs
-          (<a href="https://github.com/google/iree/blob/main/iree/samples/models/simple_abs.mlir">source</a>)
+          (<a href="https://github.com/iree-org/iree/blob/main/iree/samples/models/simple_abs.mlir">source</a>)
         </div>
         <div class="col-sm-auto">
           <button class="btn btn-secondary" onclick="loadSample('simple_abs')">Load sample</button>
@@ -131,7 +131,7 @@
       <div class="row" style="padding:4px">
         <div class="col-sm">
           fullyconnected
-          (<a href="https://github.com/google/iree/blob/main/tests/e2e/models/fullyconnected.mlir">source</a>)
+          (<a href="https://github.com/iree-org/iree/blob/main/tests/e2e/models/fullyconnected.mlir">source</a>)
         </div>
         <div class="col-sm-auto">
           <button class="btn btn-secondary" onclick="loadSample('fullyconnected')">Load sample</button>
@@ -140,7 +140,7 @@
       <div class="row" style="padding:4px">
         <div class="col-sm">
           collatz
-          (<a href="https://github.com/google/iree/blob/main/tests/e2e/models/collatz.mlir">source</a>)
+          (<a href="https://github.com/iree-org/iree/blob/main/tests/e2e/models/collatz.mlir">source</a>)
         </div>
         <div class="col-sm-auto">
           <button class="btn btn-secondary" onclick="loadSample('collatz')">Load sample</button>
diff --git a/integrations/tensorflow/WORKSPACE b/integrations/tensorflow/WORKSPACE
index a3919d2..aff2fc2 100644
--- a/integrations/tensorflow/WORKSPACE
+++ b/integrations/tensorflow/WORKSPACE
@@ -12,7 +12,7 @@
 git_repository(
     name = "org_tensorflow",
     commit = TENSORFLOW_COMMIT,
-    remote = "https://github.com/google/iree-tf-fork.git",
+    remote = "https://github.com/iree-org/iree-tf-fork.git",
 )
 
 # Import all of the tensorflow dependencies.
diff --git a/integrations/tensorflow/iree_tf_compiler/TFL/Passes.cpp b/integrations/tensorflow/iree_tf_compiler/TFL/Passes.cpp
index 804d2e0..c0b7c7f 100644
--- a/integrations/tensorflow/iree_tf_compiler/TFL/Passes.cpp
+++ b/integrations/tensorflow/iree_tf_compiler/TFL/Passes.cpp
@@ -55,7 +55,7 @@
   pm.addPass(createLowerGlobalTensorsPass());
 
   mlir::tosa::TOSATFTFLLegalizationPipelineOptions tosaOptions;
-  // Temporary work-around for https://github.com/google/iree/issues/8974
+  // Temporary work-around for https://github.com/iree-org/iree/issues/8974
   tosaOptions.dequantize_tfl_softmax = true;
   mlir::tosa::createTFTFLtoTOSALegalizationPipeline(pm, tosaOptions);
 
diff --git a/integrations/tensorflow/python_projects/iree_tf/setup.py b/integrations/tensorflow/python_projects/iree_tf/setup.py
index 45024e5..3a33691 100644
--- a/integrations/tensorflow/python_projects/iree_tf/setup.py
+++ b/integrations/tensorflow/python_projects/iree_tf/setup.py
@@ -89,7 +89,7 @@
     description="IREE TensorFlow Compiler Tools",
     long_description=README,
     long_description_content_type="text/markdown",
-    url="https://github.com/google/iree",
+    url="https://github.com/iree-org/iree",
     classifiers=[
         "Development Status :: 3 - Alpha",
         "License :: OSI Approved :: Apache Software License",
diff --git a/integrations/tensorflow/python_projects/iree_tflite/setup.py b/integrations/tensorflow/python_projects/iree_tflite/setup.py
index 6e76112..3173ad7 100644
--- a/integrations/tensorflow/python_projects/iree_tflite/setup.py
+++ b/integrations/tensorflow/python_projects/iree_tflite/setup.py
@@ -89,7 +89,7 @@
     description="IREE TFLite Compiler Tools",
     long_description=README,
     long_description_content_type="text/markdown",
-    url="https://github.com/google/iree",
+    url="https://github.com/iree-org/iree",
     classifiers=[
         "Development Status :: 3 - Alpha",
         "License :: OSI Approved :: Apache Software License",
diff --git a/integrations/tensorflow/python_projects/iree_xla/setup.py b/integrations/tensorflow/python_projects/iree_xla/setup.py
index b328db9..22a1a8d 100644
--- a/integrations/tensorflow/python_projects/iree_xla/setup.py
+++ b/integrations/tensorflow/python_projects/iree_xla/setup.py
@@ -89,7 +89,7 @@
     description="IREE XLA Compiler Tools",
     long_description=README,
     long_description_content_type="text/markdown",
-    url="https://github.com/google/iree",
+    url="https://github.com/iree-org/iree",
     classifiers=[
         "Development Status :: 3 - Alpha",
         "License :: OSI Approved :: Apache Software License",
diff --git a/integrations/tensorflow/test/python/iree_tfl_tests/mobilebert_tf2_quant_test.py b/integrations/tensorflow/test/python/iree_tfl_tests/mobilebert_tf2_quant_test.py
index 259aa1f..a8876ad 100644
--- a/integrations/tensorflow/test/python/iree_tfl_tests/mobilebert_tf2_quant_test.py
+++ b/integrations/tensorflow/test/python/iree_tfl_tests/mobilebert_tf2_quant_test.py
@@ -37,7 +37,7 @@
                                                 details)
     # We have confirmed in large scale accuracy tests that differences as large
     # as 5.0 is acceptable. We later further relaxed from 5.0 to 7.0 in
-    # https://github.com/google/iree/pull/9337 when quantized Softmax got
+    # https://github.com/iree-org/iree/pull/9337 when quantized Softmax got
     # de-quantized, which should be numerically correct albeit not bit-exact.
     # The actual observed max error was ~ 6.36. The value 7.0 is that rounded up
     # to the next integer.
diff --git a/runtime/bindings/python/CMakeLists.txt b/runtime/bindings/python/CMakeLists.txt
index 383daf8..f3b37b6 100644
--- a/runtime/bindings/python/CMakeLists.txt
+++ b/runtime/bindings/python/CMakeLists.txt
@@ -155,7 +155,7 @@
 )
 
 # TODO: Enable this once the CI bots are updated to install the python3-venv
-# apt package. https://github.com/google/iree/issues/9080
+# apt package. https://github.com/iree-org/iree/issues/9080
 # iree_py_test(
 #   NAME
 #     package_test
diff --git a/runtime/bindings/tflite/README.md b/runtime/bindings/tflite/README.md
index a9fc9da..917ad41 100644
--- a/runtime/bindings/tflite/README.md
+++ b/runtime/bindings/tflite/README.md
@@ -1,7 +1,7 @@
 # IREE TFLite C API Compatibility Shim
 
 **EXPERIMENTAL**: we are working towards making this a stable API but it has a
-ways to go still. Progress is being tracked in https://github.com/google/iree/projects/17.
+ways to go still. Progress is being tracked in https://github.com/iree-org/iree/projects/17.
 
 Provides a (mostly) tflite-compatible API that allows loading compiled IREE
 modules, managing tensors, and invoking functions with the same conventions as
@@ -174,7 +174,7 @@
 
 Custom ops in tflite map to functions imported into compiled IREE modules.
 The IREE tflite API shim could provide a wrapper implemented as an
-[iree_vm_module_t](https://github.com/google/iree/blob/main/iree/vm/module.h)
+[iree_vm_module_t](https://github.com/iree-org/iree/blob/main/iree/vm/module.h)
 that resolves and executes the functions as they are called by the VM. Having
 real IREE modules, though, provides significant benefits in representation
 such as the ability to have asynchronous custom behavior that interacts well
diff --git a/runtime/setup.py b/runtime/setup.py
index ae7f308..3411657 100644
--- a/runtime/setup.py
+++ b/runtime/setup.py
@@ -375,7 +375,7 @@
         "Programming Language :: Python :: 3.8",
         "Programming Language :: Python :: 3.9",
     ],
-    url="https://github.com/google/iree",
+    url="https://github.com/iree-org/iree",
     python_requires=">=3.7",
     ext_modules=[
         CMakeExtension("iree._runtime"),
diff --git a/runtime/src/iree/hal/drivers/vulkan/descriptor_set_arena.cc b/runtime/src/iree/hal/drivers/vulkan/descriptor_set_arena.cc
index 0eebe07..6590029 100644
--- a/runtime/src/iree/hal/drivers/vulkan/descriptor_set_arena.cc
+++ b/runtime/src/iree/hal/drivers/vulkan/descriptor_set_arena.cc
@@ -61,7 +61,7 @@
       // to match the ABI and provide the buffer as 32-bit aligned, otherwise
       // the whole read by the shader is considered as out of bounds per the
       // Vulkan spec. See
-      // https://github.com/google/iree/issues/2022#issuecomment-640617234 for
+      // https://github.com/iree-org/iree/issues/2022#issuecomment-640617234 for
       // more details.
       buffer_info.range = iree_device_align(
           std::min(binding.length, iree_hal_buffer_byte_length(binding.buffer) -
diff --git a/runtime/src/iree/tooling/vm_util.h b/runtime/src/iree/tooling/vm_util.h
index c58bacd..b2304f8 100644
--- a/runtime/src/iree/tooling/vm_util.h
+++ b/runtime/src/iree/tooling/vm_util.h
@@ -41,7 +41,7 @@
 // Prints buffers in the IREE standard shaped buffer format:
 //   [shape]xtype=[value]
 // described in
-// https://github.com/google/iree/tree/main/iree/hal/api.h
+// https://github.com/iree-org/iree/tree/main/iree/hal/api.h
 Status PrintVariantList(iree_vm_list_t* variant_list, size_t max_element_count,
                         std::ostream* os);
 inline Status PrintVariantList(iree_vm_list_t* variant_list, std::ostream* os) {
diff --git a/samples/README.md b/samples/README.md
index 1053293..eb8fc22 100644
--- a/samples/README.md
+++ b/samples/README.md
@@ -1,3 +1,3 @@
 # IREE Samples
 
-Also see the [iree-samples](https://github.com/google/iree-samples) repository.
+Also see the [iree-samples](https://github.com/iree-org/iree-samples) repository.
diff --git a/samples/colab/README.md b/samples/colab/README.md
index 5c7a53d..7e7d7a7 100644
--- a/samples/colab/README.md
+++ b/samples/colab/README.md
@@ -7,19 +7,19 @@
 Constructs a TF module for performing image edge detection and runs it using
 IREE
 
-[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/iree/blob/main/samples/colab/edge_detection.ipynb)
+[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/iree-org/iree/blob/main/samples/colab/edge_detection.ipynb)
 
 ### [low_level_invoke_function\.ipynb](low_level_invoke_function.ipynb)
 
 Shows off some concepts of the low level IREE python bindings
 
-[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/iree/blob/main/samples/colab/low_level_invoke_function.ipynb)
+[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/iree-org/iree/blob/main/samples/colab/low_level_invoke_function.ipynb)
 
 ### [mnist_training\.ipynb](mnist_training.ipynb)
 
 Compile, train and execute a TensorFlow Keras neural network with IREE
 
-[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/iree/blob/main/samples/colab/mnist_training.ipynb)
+[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/iree-org/iree/blob/main/samples/colab/mnist_training.ipynb)
 
 ### [resnet\.ipynb](resnet.ipynb)
 
@@ -27,7 +27,7 @@
 [ResNet50](https://www.tensorflow.org/api_docs/python/tf/keras/applications/ResNet50)
 model and runs it using IREE
 
-[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/iree/blob/main/samples/colab/resnet.ipynb)
+[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/iree-org/iree/blob/main/samples/colab/resnet.ipynb)
 
 ### [tensorflow_hub_import\.ipynb](tensorflow_hub_import.ipynb)
 
@@ -35,7 +35,7 @@
 [MobileNet V2](https://tfhub.dev/google/tf2-preview/mobilenet_v2/classification)
 model, pre-processes it for import, then compiles it using IREE
 
-[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/iree/blob/main/samples/colab/tensorflow_hub_import.ipynb)
+[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/iree-org/iree/blob/main/samples/colab/tensorflow_hub_import.ipynb)
 
 ### [tflite_text_classification\.ipynb](tflite_text_classification.ipynb)
 
@@ -43,7 +43,7 @@
 [TFLite text classification](https://www.tensorflow.org/lite/examples/text_classification/overview)
 model, and runs it using TFLite and IREE
 
-[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/iree/blob/main/samples/colab/tflite_text_classification.ipynb)
+[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/iree-org/iree/blob/main/samples/colab/tflite_text_classification.ipynb)
 
 ## Working with GitHub
 
diff --git a/samples/colab/edge_detection.ipynb b/samples/colab/edge_detection.ipynb
index b82c3a4..c8e9bcc 100644
--- a/samples/colab/edge_detection.ipynb
+++ b/samples/colab/edge_detection.ipynb
@@ -66,7 +66,7 @@
         "outputId": "bb1e5d0b-ef67-4f25-d4a1-61bc2b7d8356"
       },
       "source": [
-        "!python -m pip install iree-compiler iree-runtime iree-tools-tf -f https://github.com/google/iree/releases"
+        "!python -m pip install iree-compiler iree-runtime iree-tools-tf -f https://github.com/iree-org/iree/releases"
       ],
       "execution_count": 2,
       "outputs": [
@@ -74,15 +74,15 @@
           "output_type": "stream",
           "name": "stdout",
           "text": [
-            "Looking in links: https://github.com/google/iree/releases\n",
+            "Looking in links: https://github.com/iree-org/iree/releases\n",
             "Collecting iree-compiler\n",
-            "  Downloading https://github.com/google/iree/releases/download/candidate-20220119.22/iree_compiler-20220119.22-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (42.6 MB)\n",
+            "  Downloading https://github.com/iree-org/iree/releases/download/candidate-20220119.22/iree_compiler-20220119.22-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (42.6 MB)\n",
             "\u001b[K     |████████████████████████████████| 42.6 MB 1.3 MB/s \n",
             "\u001b[?25hCollecting iree-runtime\n",
-            "  Downloading https://github.com/google/iree/releases/download/candidate-20220119.22/iree_runtime-20220119.22-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.7 MB)\n",
+            "  Downloading https://github.com/iree-org/iree/releases/download/candidate-20220119.22/iree_runtime-20220119.22-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.7 MB)\n",
             "\u001b[K     |████████████████████████████████| 1.7 MB 45.2 MB/s \n",
             "\u001b[?25hCollecting iree-tools-tf\n",
-            "  Downloading https://github.com/google/iree/releases/download/candidate-20220119.22/iree_tools_tf-20220119.22-py3-none-linux_x86_64.whl (49.1 MB)\n",
+            "  Downloading https://github.com/iree-org/iree/releases/download/candidate-20220119.22/iree_tools_tf-20220119.22-py3-none-linux_x86_64.whl (49.1 MB)\n",
             "\u001b[K     |████████████████████████████████| 49.1 MB 94 kB/s \n",
             "\u001b[?25hRequirement already satisfied: PyYAML in /usr/local/lib/python3.7/dist-packages (from iree-compiler) (3.13)\n",
             "Requirement already satisfied: numpy in /usr/local/lib/python3.7/dist-packages (from iree-compiler) (1.19.5)\n",
@@ -448,4 +448,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}
diff --git a/samples/colab/low_level_invoke_function.ipynb b/samples/colab/low_level_invoke_function.ipynb
index ed28e7b..95ba289 100644
--- a/samples/colab/low_level_invoke_function.ipynb
+++ b/samples/colab/low_level_invoke_function.ipynb
@@ -66,7 +66,7 @@
         "outputId": "ec9dbc63-c510-41d8-98fb-54a2151b4815"
       },
       "source": [
-        "!python -m pip install iree-compiler iree-runtime -f https://github.com/google/iree/releases"
+        "!python -m pip install iree-compiler iree-runtime -f https://github.com/iree-org/iree/releases"
       ],
       "execution_count": 2,
       "outputs": [
@@ -75,12 +75,12 @@
           "name": "stdout",
           "text": [
             "Looking in indexes: https://pypi.org/simple, https://us-python.pkg.dev/colab-wheels/public/simple/\n",
-            "Looking in links: https://github.com/google/iree/releases\n",
+            "Looking in links: https://github.com/iree-org/iree/releases\n",
             "Collecting iree-compiler\n",
-            "  Downloading https://github.com/google/iree/releases/download/candidate-20220608.163/iree_compiler-20220608.163-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (49.5 MB)\n",
+            "  Downloading https://github.com/iree-org/iree/releases/download/candidate-20220608.163/iree_compiler-20220608.163-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (49.5 MB)\n",
             "\u001b[K     |████████████████████████████████| 49.5 MB 151 kB/s \n",
             "\u001b[?25hCollecting iree-runtime\n",
-            "  Downloading https://github.com/google/iree/releases/download/candidate-20220608.163/iree_runtime-20220608.163-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (2.1 MB)\n",
+            "  Downloading https://github.com/iree-org/iree/releases/download/candidate-20220608.163/iree_runtime-20220608.163-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (2.1 MB)\n",
             "\u001b[K     |████████████████████████████████| 2.1 MB 53.6 MB/s \n",
             "\u001b[?25hRequirement already satisfied: PyYAML in /usr/local/lib/python3.7/dist-packages (from iree-compiler) (3.13)\n",
             "Requirement already satisfied: numpy in /usr/local/lib/python3.7/dist-packages (from iree-compiler) (1.21.6)\n",
@@ -164,4 +164,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}
diff --git a/samples/colab/mnist_training.ipynb b/samples/colab/mnist_training.ipynb
index ecf94b8..3251d34 100644
--- a/samples/colab/mnist_training.ipynb
+++ b/samples/colab/mnist_training.ipynb
@@ -64,7 +64,7 @@
       },
       "source": [
         "%%capture\n",
-        "!python -m pip install iree-compiler iree-runtime iree-tools-tf -f https://github.com/google/iree/releases"
+        "!python -m pip install iree-compiler iree-runtime iree-tools-tf -f https://github.com/iree-org/iree/releases"
       ],
       "execution_count": 1,
       "outputs": []
@@ -348,7 +348,7 @@
       },
       "source": [
         "# Compile the TrainableDNN module\n",
-        "# Note: extra flags are needed to i64 demotion, see https://github.com/google/iree/issues/8644\n",
+        "# Note: extra flags are needed to i64 demotion, see https://github.com/iree-org/iree/issues/8644\n",
         "vm_flatbuffer = iree.compiler.tf.compile_module(\n",
         "    TrainableDNN(),\n",
         "    target_backends=[backend_choice],\n",
@@ -606,4 +606,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}
diff --git a/samples/colab/resnet.ipynb b/samples/colab/resnet.ipynb
index e4690bc..0bbf9ff 100644
--- a/samples/colab/resnet.ipynb
+++ b/samples/colab/resnet.ipynb
@@ -65,22 +65,22 @@
         "outputId": "d266854b-414f-4add-9f4c-9119b1d91ef6"
       },
       "source": [
-        "!python -m pip install iree-compiler iree-runtime iree-tools-tf -f https://github.com/google/iree/releases"
+        "!python -m pip install iree-compiler iree-runtime iree-tools-tf -f https://github.com/iree-org/iree/releases"
       ],
       "execution_count": 2,
       "outputs": [
         {
           "output_type": "stream",
           "text": [
-            "Looking in links: https://github.com/google/iree/releases\n",
+            "Looking in links: https://github.com/iree-org/iree/releases\n",
             "Collecting iree-compiler\n",
-            "\u001b[?25l  Downloading https://github.com/google/iree/releases/download/snapshot-20210608.328/iree_compiler_snapshot-20210608.328-py3-none-manylinux2014_x86_64.whl (33.6MB)\n",
+            "\u001b[?25l  Downloading https://github.com/iree-org/iree/releases/download/snapshot-20210608.328/iree_compiler_snapshot-20210608.328-py3-none-manylinux2014_x86_64.whl (33.6MB)\n",
             "\u001b[K     |████████████████████████████████| 33.6MB 122kB/s \n",
             "\u001b[?25hCollecting iree-runtime\n",
-            "\u001b[?25l  Downloading https://github.com/google/iree/releases/download/snapshot-20210608.328/iree_runtime_snapshot-20210608.328-cp37-cp37m-manylinux2014_x86_64.whl (521kB)\n",
+            "\u001b[?25l  Downloading https://github.com/iree-org/iree/releases/download/snapshot-20210608.328/iree_runtime_snapshot-20210608.328-cp37-cp37m-manylinux2014_x86_64.whl (521kB)\n",
             "\u001b[K     |████████████████████████████████| 522kB 36.5MB/s \n",
             "\u001b[?25hCollecting iree-tools-tf\n",
-            "\u001b[?25l  Downloading https://github.com/google/iree/releases/download/snapshot-20210608.328/iree_tools_tf_snapshot-20210608.328-py3-none-manylinux2014_x86_64.whl (53.6MB)\n",
+            "\u001b[?25l  Downloading https://github.com/iree-org/iree/releases/download/snapshot-20210608.328/iree_tools_tf_snapshot-20210608.328-py3-none-manylinux2014_x86_64.whl (53.6MB)\n",
             "\u001b[K     |████████████████████████████████| 53.6MB 76kB/s \n",
             "\u001b[?25hInstalling collected packages: iree-compiler, iree-runtime, iree-tools-tf\n",
             "Successfully installed iree-compiler-20210608.328 iree-runtime-20210608.328 iree-tools-tf-20210608.328\n"
@@ -357,4 +357,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}
diff --git a/samples/colab/tensorflow_hub_import.ipynb b/samples/colab/tensorflow_hub_import.ipynb
index cd63df3..9ffbf7a 100644
--- a/samples/colab/tensorflow_hub_import.ipynb
+++ b/samples/colab/tensorflow_hub_import.ipynb
@@ -58,7 +58,7 @@
         "\n",
         "At the end of the notebook, the compilation artifacts are compressed into a .zip file for you to download and use in an application.\n",
         "\n",
-        "See also https://google.github.io/iree/ml-frameworks/tensorflow/."
+        "See also https://iree-org.github.io/iree/ml-frameworks/tensorflow/."
       ]
     },
     {
@@ -77,7 +77,7 @@
       },
       "source": [
         "%%capture\n",
-        "!python -m pip install iree-compiler iree-runtime iree-tools-tf -f https://github.com/google/iree/releases"
+        "!python -m pip install iree-compiler iree-runtime iree-tools-tf -f https://github.com/iree-org/iree/releases"
       ],
       "execution_count": 2,
       "outputs": []
@@ -502,4 +502,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}
diff --git a/samples/colab/tflite_text_classification.ipynb b/samples/colab/tflite_text_classification.ipynb
index b485d06..a17a7ec 100644
--- a/samples/colab/tflite_text_classification.ipynb
+++ b/samples/colab/tflite_text_classification.ipynb
@@ -44,7 +44,7 @@
       "outputs": [],
       "source": [
         "%%capture\n",
-        "!python -m pip install iree-compiler iree-runtime iree-tools-tflite -f https://github.com/google/iree/releases/latest\n",
+        "!python -m pip install iree-compiler iree-runtime iree-tools-tflite -f https://github.com/iree-org/iree/releases/latest\n",
         "!pip3 install --extra-index-url https://google-coral.github.io/py-repo/ tflite_runtime"
       ]
     },
diff --git a/samples/dynamic_shapes/README.md b/samples/dynamic_shapes/README.md
index b1aeff1..107db9b 100644
--- a/samples/dynamic_shapes/README.md
+++ b/samples/dynamic_shapes/README.md
@@ -13,7 +13,7 @@
 [`dynamic_shapes.ipynb`](./dynamic_shapes.ipynb)
 [Colab](https://research.google.com/colaboratory/) notebook:
 
-[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/iree/blob/main/samples/dynamic_shapes/dynamic_shapes.ipynb)
+[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/iree-org/iree/blob/main/samples/dynamic_shapes/dynamic_shapes.ipynb)
 
 Step 3 should be performed on your development host machine
 
@@ -68,7 +68,7 @@
     generates
 
 2. Build the `iree-compile` tool (see
-    [here](https://google.github.io/iree/building-from-source/getting-started/)
+    [here](https://iree-org.github.io/iree/building-from-source/getting-started/)
     for general instructions on building using CMake)
 
     ```
@@ -77,7 +77,7 @@
     ```
 
 3. Compile the `dynamic_shapes.mlir` file using `iree-compile`. The
-    [CPU configuration](https://google.github.io/iree/deployment-configurations/cpu/)
+    [CPU configuration](https://iree-org.github.io/iree/deployment-configurations/cpu/)
     has the best support for dynamic shapes:
 
     ```
diff --git a/samples/dynamic_shapes/dynamic_shapes.ipynb b/samples/dynamic_shapes/dynamic_shapes.ipynb
index 4e25a74..1a5532a 100644
--- a/samples/dynamic_shapes/dynamic_shapes.ipynb
+++ b/samples/dynamic_shapes/dynamic_shapes.ipynb
@@ -141,7 +141,7 @@
       },
       "source": [
         "%%capture\n",
-        "!python -m pip install iree-compiler iree-tools-tf -f https://github.com/google/iree/releases"
+        "!python -m pip install iree-compiler iree-tools-tf -f https://github.com/iree-org/iree/releases"
       ],
       "execution_count": 4,
       "outputs": []
@@ -241,7 +241,7 @@
         "\n",
         "_Note: you can stop after each step and use intermediate outputs with other tools outside of Colab._\n",
         "\n",
-        "_See the [README](https://github.com/google/iree/tree/main/iree/samples/dynamic_shapes#instructions) for more details and example command line instructions._\n",
+        "_See the [README](https://github.com/iree-org/iree/tree/main/iree/samples/dynamic_shapes#instructions) for more details and example command line instructions._\n",
         "\n",
         "* _The \"imported MLIR\" can be used by IREE's generic compiler tools_\n",
         "* _The \"flatbuffer blob\" can be saved and used by runtime applications_\n",
@@ -256,7 +256,7 @@
       },
       "source": [
         "%%capture\n",
-        "!python -m pip install iree-compiler -f https://github.com/google/iree/releases"
+        "!python -m pip install iree-compiler -f https://github.com/iree-org/iree/releases"
       ],
       "execution_count": 6,
       "outputs": []
@@ -304,7 +304,7 @@
       },
       "source": [
         "%%capture\n",
-        "!python -m pip install iree-runtime -f https://github.com/google/iree/releases"
+        "!python -m pip install iree-runtime -f https://github.com/iree-org/iree/releases"
       ],
       "execution_count": 8,
       "outputs": []
@@ -473,4 +473,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}
diff --git a/samples/models/mnist.mlir b/samples/models/mnist.mlir
index b367abc..0da4f4c 100644
--- a/samples/models/mnist.mlir
+++ b/samples/models/mnist.mlir
@@ -1,5 +1,5 @@
 // Trained MNIST model generated by
-// https://github.com/google/iree/blob/main/samples/colab/mnist_training.ipynb.
+// https://github.com/iree-org/iree/blob/main/samples/colab/mnist_training.ipynb.
 //
 // Model structure is from tf.keras:
 //
diff --git a/samples/simple_embedding/README.md b/samples/simple_embedding/README.md
index d9081cb..4dcb318 100644
--- a/samples/simple_embedding/README.md
+++ b/samples/simple_embedding/README.md
@@ -8,7 +8,7 @@
 # Background
 
 The main bytecode testing tool
-[iree-run-module](https://github.com/google/iree/tree/main/tools/iree-run-module-main.cc)
+[iree-run-module](https://github.com/iree-org/iree/tree/main/tools/iree-run-module-main.cc)
 requires a proper operating system support to set up the runtime environment to
 execute an IREE bytecode module. For embedded systems, the support such as file
 system or multi-thread asynchronous control may not be available. This sample
@@ -58,7 +58,7 @@
 
 The HAL device for different target backends. The device is a `module_loader` +
 `executor` combination. For example,
-[device_embedded_sync.c](https://github.com/google/iree/blob/main/iree/samples/simple_embedding/device_embedded_sync.c)
+[device_embedded_sync.c](https://github.com/iree-org/iree/blob/main/iree/samples/simple_embedding/device_embedded_sync.c)
 uses the embedded library loader and the synchronous executor:
 ```c
 iree_hal_sync_device_params_t params;
@@ -76,7 +76,7 @@
 ```
 
 Whereas for
-[device_dylib.c](https://github.com/google/iree/blob/main/iree/samples/simple_embedding/device_dylib.c),
+[device_dylib.c](https://github.com/iree-org/iree/blob/main/iree/samples/simple_embedding/device_dylib.c),
 the executor is replaced with the multi-thread ready asynchronous task executor:
 ```c
 ...
@@ -92,7 +92,7 @@
                                         iree_allocator_system(), device);
 ```
 An example that utilizes a higher-level driver registry is in
-[device_vulkan.c](https://github.com/google/iree/blob/main/iree/samples/simple_embedding/device_vulkan.c)
+[device_vulkan.c](https://github.com/iree-org/iree/blob/main/iree/samples/simple_embedding/device_vulkan.c)
 
 ### Load device-specific bytecode module
 To avoid the file IO, the bytecode module is converted into a data stream
@@ -105,4 +105,4 @@
 should support any architecture that IREE supports, and `device_embedded_sync`
 should support any architecture that supports `dylib-llvm-aot` codegen target
 backend (may need to add the bytecode module data if it is not already in
-[device_embedded_sync.c](https://github.com/google/iree/blob/main/iree/samples/simple_embedding/device_embedded_sync.c)).
+[device_embedded_sync.c](https://github.com/iree-org/iree/blob/main/iree/samples/simple_embedding/device_embedded_sync.c)).
diff --git a/samples/static_library/README.md b/samples/static_library/README.md
index 75e9ea2..39420a1 100644
--- a/samples/static_library/README.md
+++ b/samples/static_library/README.md
@@ -35,7 +35,7 @@
 1. Configure CMake for building the static library then demo. You'll need to set
 the flags building samples, the compiler, and the `dylib-llvm-aot`
 driver/backend. See
-[here](https://google.github.io/iree/building-from-source/getting-started/)
+[here](https://iree-org.github.io/iree/building-from-source/getting-started/)
 for general instructions on building using CMake):
 
   ```shell
@@ -75,7 +75,7 @@
 compile the library and demo with different options.
 
 For example, see
-[documentation](https://google.github.io/iree/building-from-source/android/)
+[documentation](https://iree-org.github.io/iree/building-from-source/android/)
 on cross compiling on Android.
 
 Note: separating the target from the host will require modifying dependencies in
diff --git a/samples/variables_and_state/README.md b/samples/variables_and_state/README.md
index d66434f..191bca1 100644
--- a/samples/variables_and_state/README.md
+++ b/samples/variables_and_state/README.md
@@ -13,7 +13,7 @@
 [`variables_and_state.ipynb`](./variables_and_state.ipynb)
 [Colab](https://research.google.com/colaboratory/) notebook:
 
-[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/iree/blob/main/samples/variables_and_state/variables_and_state.ipynb)
+[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/iree-org/iree/blob/main/samples/variables_and_state/variables_and_state.ipynb)
 
 Steps 4-5 are in [`main.c`](./main.c)
 
@@ -62,7 +62,7 @@
    `counter_vmvx.vmfb` files it generates
 
 2. Build the `iree_samples_variables_and_state` CMake target (see
-    [here](https://google.github.io/iree/building-from-source/getting-started/)
+    [here](https://iree-org.github.io/iree/building-from-source/getting-started/)
     for general instructions on building using CMake)
 
     ```
@@ -87,7 +87,7 @@
 
 For example, to use IREE's `cpu` target, which is optimized for CPU execution
 using LLVM, refer to the
-[documentation](https://google.github.io/iree/deployment-configurations/cpu/)
+[documentation](https://iree-org.github.io/iree/deployment-configurations/cpu/)
 and compile the imported `counter.mlir` file using `iree-compile`:
 
 ```
diff --git a/samples/variables_and_state/variables_and_state.ipynb b/samples/variables_and_state/variables_and_state.ipynb
index c713bdb..77c851d 100644
--- a/samples/variables_and_state/variables_and_state.ipynb
+++ b/samples/variables_and_state/variables_and_state.ipynb
@@ -146,7 +146,7 @@
       },
       "source": [
         "%%capture\n",
-        "!python -m pip install iree-compiler iree-tools-tf -f https://github.com/google/iree/releases"
+        "!python -m pip install iree-compiler iree-tools-tf -f https://github.com/iree-org/iree/releases"
       ],
       "execution_count": 4,
       "outputs": []
@@ -256,7 +256,7 @@
         "\n",
         "_Note: you can stop after each step and use intermediate outputs with other tools outside of Colab._\n",
         "\n",
-        "_See the [README](https://github.com/google/iree/tree/main/iree/samples/variables_and_state#changing-compilation-options) for more details and example command line instructions._\n",
+        "_See the [README](https://github.com/iree-org/iree/tree/main/iree/samples/variables_and_state#changing-compilation-options) for more details and example command line instructions._\n",
         "\n",
         "* _The \"imported MLIR\" can be used by IREE's generic compiler tools_\n",
         "* _The \"flatbuffer blob\" can be saved and used by runtime applications_\n",
@@ -271,7 +271,7 @@
       },
       "source": [
         "%%capture\n",
-        "!python -m pip install iree-compiler -f https://github.com/google/iree/releases"
+        "!python -m pip install iree-compiler -f https://github.com/iree-org/iree/releases"
       ],
       "execution_count": 6,
       "outputs": []
@@ -316,7 +316,7 @@
       },
       "source": [
         "%%capture\n",
-        "!python -m pip install iree-runtime -f https://github.com/google/iree/releases"
+        "!python -m pip install iree-runtime -f https://github.com/iree-org/iree/releases"
       ],
       "execution_count": 8,
       "outputs": []
@@ -491,4 +491,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}
diff --git a/samples/vision_inference/README.md b/samples/vision_inference/README.md
index d41cbf7..bf644bb 100644
--- a/samples/vision_inference/README.md
+++ b/samples/vision_inference/README.md
@@ -4,7 +4,7 @@
 model on an image using IREE's command line tools.
 
 A similar sample is implemented in C code over in the iree-samples repository
-at https://github.com/google/iree-samples/tree/main/cpp/vision_inference
+at https://github.com/iree-org/iree-samples/tree/main/cpp/vision_inference
 
 * This version of the sample uses a Python script to convert an image into the
   expected format then runs the compiled MNIST program through IREE's command
diff --git a/tests/e2e/models/edge_detection.mlir b/tests/e2e/models/edge_detection.mlir
index b438a87..3626c22 100644
--- a/tests/e2e/models/edge_detection.mlir
+++ b/tests/e2e/models/edge_detection.mlir
@@ -3,7 +3,7 @@
 // RUN: [[ $IREE_VULKAN_DISABLE == 1 ]] || (iree-run-mlir --iree-input-type=mhlo --iree-hal-target-backends=vulkan-spirv %s --function-input="1x128x128x1xf32" | FileCheck %s)
 
 // Image edge detection module generated by.
-// https://github.com/google/iree/blob/main/samples/colab/edge_detection.ipynb.
+// https://github.com/iree-org/iree/blob/main/samples/colab/edge_detection.ipynb.
 //
 // Input : a single 128x128 pixel image as a tensor<1x128x128x1xf32>, with pixels in [0.0, 1.0]
 // Output: a single image in the same format after running edge detection
diff --git a/tests/e2e/regression/libm_linking.mlir b/tests/e2e/regression/libm_linking.mlir
index 57e7009..01aa599 100644
--- a/tests/e2e/regression/libm_linking.mlir
+++ b/tests/e2e/regression/libm_linking.mlir
@@ -13,7 +13,7 @@
 // This test checks that the LLVM lowerings for certain operations are
 // correctly covered by our linker configurations.
 //
-// See https://github.com/google/iree/issues/4717 for more details.
+// See https://github.com/iree-org/iree/issues/4717 for more details.
 
 // CHECK: vm.func private @tanh
 func.func @tanh(%input : tensor<f32>) -> (tensor<f32>) {
diff --git a/tests/e2e/tosa_ops/BUILD b/tests/e2e/tosa_ops/BUILD
index 27052b3..e02c8ee 100644
--- a/tests/e2e/tosa_ops/BUILD
+++ b/tests/e2e/tosa_ops/BUILD
@@ -8,7 +8,7 @@
 # Each test file should have a name matching the corresponding TOSA op and test only the
 # functionality of that op (though may make use of other ops where necessary). Tests should be
 # written using the IREE Check framework.
-# See https://github.com/google/iree/blob/main/docs/developers/developing_iree/testing_guide.md#iree-core-end-to-end-tests.
+# See https://github.com/iree-org/iree/blob/main/docs/developers/developing_iree/testing_guide.md#iree-core-end-to-end-tests.
 
 load("//build_tools/bazel:enforce_glob.bzl", "enforce_glob")
 load("//build_tools/bazel:iree_check_test.bzl", "iree_check_single_backend_test_suite")
@@ -117,7 +117,7 @@
     ],
     include = ["*.mlir"],
     exclude = [
-        "reduce.mlir",  # Currently flakey https://github.com/google/iree/issues/5885
+        "reduce.mlir",  # Currently flakey https://github.com/iree-org/iree/issues/5885
     ],
 )
 
diff --git a/tests/e2e/xla_ops/BUILD b/tests/e2e/xla_ops/BUILD
index b7be258..ef0c7a0 100644
--- a/tests/e2e/xla_ops/BUILD
+++ b/tests/e2e/xla_ops/BUILD
@@ -8,7 +8,7 @@
 # Each test file should have a name matching the corresponding XLA HLO op and test only the
 # functionality of that op (though may make use of other ops where necessary). Tests should be
 # written using the IREE Check framework and should always pass on the reference VMVX backend.
-# See https://github.com/google/iree/blob/main/docs/developers/developing_iree/testing_guide.md#iree-core-end-to-end-tests.
+# See https://github.com/iree-org/iree/blob/main/docs/developers/developing_iree/testing_guide.md#iree-core-end-to-end-tests.
 
 load("//build_tools/bazel:enforce_glob.bzl", "enforce_glob")
 load("//build_tools/bazel:iree_check_test.bzl", "iree_check_single_backend_test_suite")
diff --git a/tests/microbenchmarks/mhlo_dot_general.mlir b/tests/microbenchmarks/mhlo_dot_general.mlir
index 343ab45..bb737a2 100644
--- a/tests/microbenchmarks/mhlo_dot_general.mlir
+++ b/tests/microbenchmarks/mhlo_dot_general.mlir
@@ -1,5 +1,5 @@
 // The following ops are sampled from mobile_bert
-// https://github.com/google/iree/blob/main/integrations/tensorflow/e2e/mobile_bert_squad_test.py
+// https://github.com/iree-org/iree/blob/main/integrations/tensorflow/e2e/mobile_bert_squad_test.py
 
 func.func @dot_general_4x384x32x384() -> tensor<4x384x384xf32> {
     %lhs = util.unfoldable_constant dense<1.0> : tensor<4x384x32xf32>
diff --git a/tools/android/run_module_app/build_apk.sh b/tools/android/run_module_app/build_apk.sh
index 3bad7c1..5167b59 100644
--- a/tools/android/run_module_app/build_apk.sh
+++ b/tools/android/run_module_app/build_apk.sh
@@ -245,7 +245,7 @@
 
 # Compile the R.java and create classes.dex out of it for Android.
 echo "Using javac: '${JAVAC_BIN?}'"
-${JAVAC?} "${IREE_RESOURCE_GEN_DIR?}"/com/google/iree/run_module/*.java
+${JAVAC?} "${IREE_RESOURCE_GEN_DIR?}"/com/iree-org/iree/run_module/*.java
 ${DX?} --output="${IREE_APK_PARTS_DIR?}/classes.dex" "${IREE_APK_PARTS_DIR?}"
 
 echo ">>> Packaging apk file <<<"