Tests of the tf , tf.math , tf.nn , tf.signal and tf.strings APIs.
IREE has three main backend targets: vmla , llvm and vulkan-spirv. We also test TFLite in our infrastructure for benchmarking purposes.
Last Updated: 2020/12/8
Note: To be thorough, these tests use high rank tensors and test int dtypes where TensorFlow allows them to be used. Both of these choices disproportionately affect TFLite coverage, and don't represent coverage for simple use cases.
| target | tflite | vmla | vulkan-spirv |
|---|---|---|---|
| abs | ✗ | ✓ | ✓ |
| add | ✗ | ✓ | ✓ |
| angle | ✗ | ✗ | ✗ |
| asinh | ✗ | ✗ | ✗ |
| atanh | ✗ | ✗ | ✗ |
| conj | ✗ | ✓ | ✓ |
| cos | ✗ | ✗ | ✗ |
| cosh | ✗ | ✗ | ✗ |
| count_nonzero | ✗ | ✗ | ✗ |
| cumprod | ✗ | ✗ | ✗ |
| cumsum | ✗ | ✗ | ✗ |
| divide | ✗ | ✗ | ✗ |
| divide_no_nan | ✗ | ✗ | ✗ |
| exp | ✗ | ✓ | ✓ |
| expm1 | ✗ | ✗ | ✗ |
| imag | ✗ | ✓ | ✓ |
| l2_normalize | ✗ | ✓ | ✗ |
| log | ✗ | ✗ | ✗ |
| log1p | ✗ | ✗ | ✗ |
| multiply | ✗ | ✓ | ✓ |
| multiply_no_nan | ✗ | ✗ | ✗ |
| negative | ✗ | ✗ | ✗ |
| pow | ✗ | ✗ | ✗ |
| real | ✗ | ✓ | ✓ |
| reciprocal | ✗ | ✗ | ✗ |
| reciprocal_no_nan | ✗ | ✗ | ✗ |
| reduce_euclidean_norm | ✗ | ✗ | ✗ |
| reduce_std | ✗ | ✗ | ✗ |
| reduce_variance | ✗ | ✗ | ✗ |
| rsqrt | ✗ | ✗ | ✗ |
| sigmoid | ✗ | ✗ | ✗ |
| sign | ✗ | ✗ | ✗ |
| sin | ✗ | ✗ | ✗ |
| sinh | ✗ | ✗ | ✗ |
| sqrt | ✗ | ✗ | ✗ |
| square | ✗ | ✓ | ✓ |
| squared_difference | ✗ | ✓ | ✓ |
| subtract | ✗ | ✓ | ✓ |
| tan | ✗ | ✗ | ✗ |
| tanh | ✗ | ✗ | ✗ |
| truediv | ✗ | ✓ | ✓ |
| xdivy | ✗ | ✗ | ✗ |
| xlog1py | ✗ | ✗ | ✗ |
| xlogy | ✗ | ✗ | ✗ |
| zero_fraction | ✗ | ✗ | ✗ |