onnxruntime/orttraining
ashbhandare cc275e7529
Gradient Accumulation optimization verified for correctness (#8273)
* Fetching frontier tensors to frontend

* Move before session initialize call

* Fetch tensor and add to cache

* Rest of the changes for using cache

* Review comments

* Review changes

* Review comments

* switch to shared_ptr

* Fix bug after rebase

* FE docstring change
2021-08-17 16:24:44 -07:00
..
orttraining Gradient Accumulation optimization verified for correctness (#8273) 2021-08-17 16:24:44 -07:00
pytorch_frontend_examples Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
tools [ROCm] dockerfile updates (#7955) 2021-06-10 23:50:19 -07:00