onnxruntime
0fa3aeb6
- [CUDA] Add Strided Tensor Support for Expand->GatherElements for Training (#11976)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
3 years ago
[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.
References
#11976 - [CUDA] Add Strided Tensor Support for Expand->GatherElements for Training
Author
Lafi7e
Parents
75bda9f2
Loading