onnxruntime/js/web/lib/wasm
Caroline Zhu eb03032925
[js/web/training] lazyResetGrad implementation (#18711)
### Description
* implemented lazyResetGrad function

### Motivation and Context
* we are in the process of adding language bindings to enable training
on web
* lazyresetgrad ensures that the gradients are calculated correctly
after the first runTrainStep call

---------

Co-authored-by: Ashwini Khade <askhade@microsoft.com>
2023-12-11 17:36:54 -08:00
..
binding [js/web/training] Add CreateTrainingSession (#17891) 2023-10-26 09:22:10 -07:00
jsep [js/webgpu] allow to specify callback for profiling data (#18732) 2023-12-07 14:10:28 -08:00
proxy-worker [js/web] set noUnusedParameters to true and fix a few bugs (#18404) 2023-11-15 09:16:29 -08: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-inference.ts [js/web/training] runTrainStep implementation (#18006) 2023-11-02 08:32:50 -07:00
session-handler-training.ts [js/web/training] lazyResetGrad implementation (#18711) 2023-12-11 17:36:54 -08:00
session-options.ts [WebNN EP] Support numThreads option for WebNN CPU device (#18054) 2023-11-12 16:45:10 -08:00
wasm-common.ts [js/webgpu] support IO binding (#17480) 2023-09-29 11:24:42 -07:00
wasm-core-impl.ts [js/web/training] runTrainStep implementation (#18006) 2023-11-02 08:32:50 -07:00
wasm-factory.ts Add "glue" between training WASM artifacts and training web (#17474) 2023-10-12 11:16:56 -07:00
wasm-training-core-impl.ts [js/web/training] lazyResetGrad implementation (#18711) 2023-12-11 17:36:54 -08:00
wasm-utils.ts