onnxruntime/orttraining
Vincent Wang 0fa3aeb65c
[CUDA] Add Strided Tensor Support for Expand->GatherElements for Training (#11976)
* strided tensor for expand and gather_elements

* bugfix

* simplify CoalesceDimensions

* resolve comments

* resolve more comments.
2022-07-25 16:05:26 +08:00
..
orttraining [CUDA] Add Strided Tensor Support for Expand->GatherElements for Training (#11976) 2022-07-25 16:05:26 +08:00
pytorch_frontend_examples Set black's target version (#11370) 2022-04-27 14:52:19 -07:00
tools [UPDATE] update AMD CI pipeline to Rocm5.2 with torch1.11 (#12162) 2022-07-14 16:38:16 +08:00