onnxruntime/js/web/lib/wasm/jsep
Jiajia Qin 785e2b1eae
[js/webgpu] Optimize softmax by vector (#18153)
### Description
This PR enables `softmax` outputs max supported components instead of
scalar for each thread.

Softmax with input[0]: [12,4096,4096] becomes 47.86 ms from 55.11 ms
2023-10-30 16:05:35 -07:00
..
webgpu [js/webgpu] Optimize softmax by vector (#18153) 2023-10-30 16:05:35 -07:00
backend-webgpu.ts [js/webgpu] Change timestamp-query-in-passes to timestamp-query (#18108) 2023-10-26 16:33:03 -07:00
init.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
log.ts [js/webgpu] support proxy for webgpu (#15851) 2023-05-15 16:23:13 -07:00
tensor-view.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
util.ts [js/webgpu] allow 0 sized tensor for tensor view (#16540) 2023-06-30 12:05:04 -07:00