The table below shows the supported TOSA ops.
| op | supported | comment |
|---|---|---|
| Data node ops | ||
| const | ✔️ | |
| Unary elementwise ops | ||
| abs | ✔️ | |
| cast | ✔️ | |
| ceil | ✔️ | |
| clamp | ✔️ | |
| clz | ✔️ | |
| exp | ✔️ | |
| floor | ✔️ | |
| log | ✔️ | |
| negate | ✔️ | |
| reciprocal | ✔️ | |
| rescale | ✔️ | |
| rsqrt | ✔️ | |
| tanh | ✔️ | |
| Binary elementwise ops | ||
| add | ✔️ | |
| arithmetic_right_shift | ✔️ | |
| equal | ✔️ | |
| greater_equal | ✔️ | |
| logical_left_shift | ✔️ | |
| maximum | ✔️ | |
| minimum | ✔️ | |
| mul | ✔️ | |
| pow | ✔️ | |
| sub | ✔️ | |
| table | ✔️ | |
| Ternary elementwise ops | ||
| select | ✔️ | |
| Other ops | ||
| argmax | ✔️ | |
| avg_pool2d | ✅ | Quantization and and acc_type not supported |
| concat | ✔️ | |
| conv2d | ✅ | Quantization and dilation not supported |
| depthwise_conv2d | ✅ | Quantization and dilation not supported |
| fully_connected | ✅ | Quantization not supported |
| gather | ✔️ | |
| matmul | ✅ | Quantization not supported |
| max_pool2d | ✅ | Quantization not supported |
| reduce_all | ✔️ | |
| reduce_any | ✔️ | |
| reduce_max | ✔️ | |
| reduce_min | ✔️ | |
| reduce_prod | ✔️ | |
| reduce_sum | ✔️ | |
| reshape | ✔️ | |
| slice | ✅ | Only for 1D to 4D inputs |
| pad | ✅ | Quantization not supported |
| tile | ✔️ | |
| transpose | ✔️ |