pytorch
2efc56d9 - OpInfo: Sample input cleanup (1/n) (#86231)

Commit
2 years ago
OpInfo: Sample input cleanup (1/n) (#86231) This rewrites various sample and error input functions to: - use the convention of `make_arg = functools.partial(make_tensor, ...)` - use the new natural syntax for `SampleInput` construction - yield instead of returning a lists, to reduce memory consumption Pull Request resolved: https://github.com/pytorch/pytorch/pull/86231 Approved by: https://github.com/mruberry
Author
Committer
Parents
Loading