onnxruntime/js/web/docs
Guenther Schmuelling 0df2e14038
js/webgpu: argmax,argmin,softmax support (#16882)
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
2023-08-02 18:16:19 -07:00
..
webgl-operators.md [js/webgpu] generate operator table for webgpu (#15954) 2023-05-20 12:20:41 -07:00
webgpu-operators.md js/webgpu: argmax,argmin,softmax support (#16882) 2023-08-02 18:16:19 -07:00