Primitive | Forward | Backward | Test | Bench |
---|---|---|---|---|
au_gru |
✅ | ❌ | ❌ | ❌ |
batch_norm |
✅ | ❌ | ❌ | ❌ |
binary |
✅ | ⬜ | ✅ | ✅ |
concat |
❌ | ⬜ | ❌ | ❌ |
convolution |
❌ | ❌ | ❌ | ❌ |
deconvolution |
❌ | ❌ | ❌ | ❌ |
eltwise |
✅ | ✅ | ✅ | ❌ |
gemm |
❌ | ⬜ | ❌ | ❌ |
group_norm |
❌ | ❌ | ❌ | ❌ |
gru |
❌ | ❌ | ❌ | ❌ |
inner_product |
✅ | ✅ | ✅ | ❌ |
layer_norm |
❌ | ❌ | ❌ | ❌ |
lbr_augru |
❌ | ❌ | ❌ | ❌ |
lbr_gru |
❌ | ❌ | ❌ | ❌ |
lrn |
❌ | ❌ | ❌ | ❌ |
lstm |
❌ | ❌ | ❌ | ❌ |
matmul |
✅ | ⬜ | ✅ | ❌ |
pooling |
❌ | ❌ | ❌ | ❌ |
prelu |
✅ | ❌ | ❌ | ❌ |
reduction |
✅ | ⬜ | ✅ | ❌ |
reorder |
❌ | ⬜ | ❌ | ❌ |
resampling |
❌ | ❌ | ❌ | ❌ |
shuffle |
❌ | ❌ | ❌ | ❌ |
softmax |
❌ | ❌ | ❌ | ❌ |
sum |
❌ | ⬜ | ❌ | ❌ |
vanilla_rnn |
❌ | ❌ | ❌ | ❌ |
- Missing support for GPU (SYCL and OpenCL)
- Missing postops