onnxruntime
e5270e3b - shared allocator for on device training (#16432)

Commit
2 years ago
shared allocator for on device training (#16432) ### Description <!-- Describe your changes. --> New logic to share allocators among module, optimizer and eval sessions for Training scenario ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. --> Previously on device training using shared allocator by sharing EP, now with new mechanism to share allocator, we need to explicitly register allocator in the environment. --------- Co-authored-by: Lei Cao <leca@microsoft.com>
Author
Parents
Loading