onnxruntime/js/web/lib
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
..
onnxjs js/webgpu: argmax,argmin,softmax support (#16882) 2023-08-02 18:16:19 -07:00
wasm js/webgpu: argmax,argmin,softmax support (#16882) 2023-08-02 18:16:19 -07:00
backend-onnxjs.ts
backend-wasm.ts [js/webgpu] support proxy for webgpu (#15851) 2023-05-15 16:23:13 -07:00
build-def.d.ts [js/web] WebGPU backend via JSEP (#14579) 2023-04-24 15:21:18 -07:00
index.ts [js] add API that allows to get package version (#16207) 2023-06-09 16:18:53 -07:00
version.ts [js] add API that allows to get package version (#16207) 2023-06-09 16:18:53 -07:00