pytorch/torch/_prims
Antonio Kim 7fc292930c Add support for torch.Generator type in TorchScript (#110413)
- Add support for `torch.Generator` type in TorchScript
- Add `generator` args to all `torch.nn.init` functions that call `uniform_` or `normal_`
- Add support for `torch.Generator` in LTC's TorchScript backend (CC: @wconstab)

CC: @eellison @davidberard98 @GlebKazantaev @behzad-a
Pull Request resolved: https://github.com/pytorch/pytorch/pull/110413
Approved by: https://github.com/wconstab, https://github.com/albanD, https://github.com/glebk-cerebras, https://github.com/davidberard98
2023-11-21 23:07:21 +00:00
..
__init__.py Add support for torch.Generator type in TorchScript (#110413) 2023-11-21 23:07:21 +00:00
context.py Remove dynamo+nvfuser (#105789) 2023-08-08 22:29:32 +00:00
debug_prims.py [Reland] Update mypy to 1.4.1 (#105227) 2023-07-15 20:30:20 +00:00
executor.py Remove dynamo+nvfuser (#105789) 2023-08-08 22:29:32 +00:00
rng_prims.py [HigherOrderOp] fallthrough some keys by default. (#110478) 2023-10-05 16:25:42 +00:00