Tests of tf.keras.layers compiled with static shapes, dynamic shapes and training enabled.
IREE has three main backend targets: vmvx , llvm and vulkan-spirv. We also test TFLite in our infrastructure for benchmarking purposes.
Last Updated: 2020/12/8
Note: Layers like Dropout are listed as passing in this table, but they function similar to identity layers in these tests. See the third table for the coverage of these layers during training.
These tests also only modify required tf.keras.layers arguments. See the full API tests below for coverage on of non-default layer configurations.
| target | tflite | vmvx | vulkan-spirv |
|---|---|---|---|
| AdditiveAttention | ✓ | ✗ | ✗ |
| AlphaDropout | ✗ | ✗ | ✗ |
| Attention | ✓ | ✗ | ✗ |
| BatchNormalization | ✗ | ✗ | ✗ |
| Dropout | ✓ | ✗ | ✗ |
| GRU | ✗ | ✗ | ✗ |
| GaussianDropout | ✗ | ✗ | ✗ |
| GaussianNoise | ✗ | ✗ | ✗ |
| LSTM | ✗ | ✗ | ✗ |
| MultiHeadAttention | ✓ | ✗ | ✗ |
| SpatialDropout1D | ✓ | ✗ | ✗ |
| SpatialDropout2D | ✓ | ✗ | ✗ |
| SpatialDropout3D | ✓ | ✗ | ✗ |