onnxruntime/js/web/lib/wasm/jsep/webgpu
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
..
ops [js/webgpu] Optimize softmax by vector (#18153) 2023-10-30 16:05:35 -07:00
attribute-with-cache-key.ts
gpu-data-manager.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
op-resolve-rules.ts [js/webgpu] Support where (#17544) 2023-10-03 14:28:21 -07:00
program-manager.ts [js/webgpu] Change timestamp-query-in-passes to timestamp-query (#18108) 2023-10-26 16:33:03 -07:00
types.ts [js/webgpu] revise uniform support (#17871) 2023-10-11 16:41:46 -07:00