pytorch/torchgen/dest
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 Apply ufmt linter to all py files under torchgen (#81570) 2022-07-16 03:52:25 +00:00
lazy_ir.py Add support for torch.Generator type in TorchScript (#110413) 2023-11-21 23:07:21 +00:00
lazy_ts_lowering.py std/var: support floating point correction value (#94073) 2023-02-23 05:50:45 +00:00
native_functions.py Apply ufmt linter to all py files under torchgen (#81570) 2022-07-16 03:52:25 +00:00
register_dispatch_key.py [BE] Enable Ruff's Flake8 PYI030 (#111103) 2023-10-12 13:31:59 +00:00
ufunc.py Use OrderedSet in ufunc codegen (#82567) 2022-08-01 18:11:06 +00:00