|
onnx2versal
|
Argmax over CHUNK_CNT chunks of CHUNK_SIZE vector. More...

Modules | |
| ArgmaxKernels | |
Classes | |
| class | ArgmaxGraph< ARGMAX, CHUNK_CNT, CHUNK_SIZE > |
| Single instance graph. More... | |
| ARGMAX | Argmax Kernel |
| CHUNK_CNT | number of chunks |
| CHUNK_SIZE | size of chunk per iteration to calculate |