mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-23 19:32:23 +00:00
0df2e14038
argmax and argmin are similar to reduce. Eventually we need to add optimized flavors of the shader. softmax is optimized but only works on the last axis for now which should be the common use case. todo: enable more ut for argmax/argmin |
||
|---|---|---|
| .. | ||
| 3rd-party | ||
| argminmax.ts | ||
| binary-op.ts | ||
| common.ts | ||
| concat.ts | ||
| conv-grouped.ts | ||
| conv-transpose.ts | ||
| conv.ts | ||
| conv2d-mm.ts | ||
| expand.ts | ||
| fuse-utils.ts | ||
| gelu.ts | ||
| gemm.ts | ||
| matmul.ts | ||
| pool.ts | ||
| reduce.ts | ||
| resize.ts | ||
| slice.ts | ||
| softmax.ts | ||
| split.ts | ||
| transpose.ts | ||
| unary-op.ts | ||