pytorch
6b0ca72b - Add prim, ref, and OpInfo for arange (#81734)

Commit
2 years ago
Add prim, ref, and OpInfo for arange (#81734) Per title. - See https://github.com/pytorch/pytorch/issues/81959 for discussion on overloading TODO: - ~Handle remaining TensorOptions: layout, pin_memory (won't do in this PR)~ - ~Add sample inputs for floating point and complex numbers (done for floating point, won't do for complex in this PR)~ Pull Request resolved: https://github.com/pytorch/pytorch/pull/81734 Approved by: https://github.com/ngimel
Author
Committer
Parents
Loading