saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-11 17:48:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/js/web/lib/wasm
History
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
..
binding [js/web/training] Add CreateTrainingSession (#17891) 2023-10-26 09:22:10 -07:00
jsep [js/webgpu] Optimize softmax by vector (#18153) 2023-10-30 16:05:35 -07:00
proxy-worker [js/web/training] Add CreateTrainingSession (#17891) 2023-10-26 09:22:10 -07:00
proxy-messages.ts [js/web/training] Add CreateTrainingSession (#17891) 2023-10-26 09:22:10 -07:00
proxy-wrapper.ts [js/web/training] Add CreateTrainingSession (#17891) 2023-10-26 09:22:10 -07:00
run-options.ts
session-handler-for-training.ts [js/web/training] Add CreateTrainingSession (#17891) 2023-10-26 09:22:10 -07:00
session-handler.ts [js/web/training] Add CreateTrainingSession (#17891) 2023-10-26 09:22:10 -07:00
session-options.ts
wasm-common.ts
wasm-core-impl.ts [js/web/training] Add CreateTrainingSession (#17891) 2023-10-26 09:22:10 -07:00
wasm-factory.ts
wasm-training-core-impl.ts [js/web/training] Add CreateTrainingSession (#17891) 2023-10-26 09:22:10 -07:00
wasm-utils.ts