Antonio Kim 27e31ab6e8 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-06 21:27:02 +00:00

..

__init__.py

Add lazy tensor python bindings (#74508)

2022-03-29 13:40:11 +00:00

test_bindings.py

Add lazy tensor python bindings (#74508)

2022-03-29 13:40:11 +00:00

test_debug_util.py

[LTC] Restore GetPythonFrames (#89122)

2022-11-18 03:37:14 +00:00

test_extract_compiled_graph.py

[BE] [3/3] Rewrite super() calls in test (#94592)

2023-02-12 22:20:53 +00:00

test_generator.py

Add support for torch.Generator type in TorchScript (#110413)

2023-11-06 21:27:02 +00:00

test_meta_kernel.py

Make addmm meta kernel consistent with mm (#84960)

2022-09-19 15:41:19 +00:00

test_reuse_ir.py

Rectify native_batch_norm schema by splitting it into two legit schemas (#88697)

2022-11-23 23:23:17 +00:00

test_step_closures.py

Add step closures (#84300)

2022-09-06 20:55:34 +00:00

test_ts_opinfo.py

Add support for torch.Generator type in TorchScript (#110413)

2023-11-06 21:27:02 +00:00