Update URLs following iree-org move. (#9635)
We moved from the [`google` organization](https://github.com/google) to the [`iree-org` organization](https://github.com/iree-org).
Some of these have redirects in place, but using the canonical URLs is still preferred. Others (like the GitHub Pages website) do _not_ have redirects, so we have to update those URLs.
Mostly documentation and comment updates, _except_
* submodule URLs are also updated (following the redirects)
* GitHub releases are updated (including pip install commands for nightly releases using `--find-links https://github.com/iree-org/iree/releases`)
* Some workflows now check `if: github.repository == 'iree-org/iree'`
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>
> 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> |
