mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
7fc292930c
- 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 |
||
|---|---|---|
| .. | ||
| ops | ||
| config.cpp | ||
| config.h | ||
| dynamic_ir.cpp | ||
| dynamic_ir.h | ||
| ir_builder.h | ||
| tensor_aten_ops.cpp | ||
| tensor_aten_ops.h | ||
| ts_autograd_functions.cpp | ||
| ts_autograd_functions.h | ||
| ts_backend_impl.cpp | ||
| ts_backend_impl.h | ||
| ts_eager_fallback.cpp | ||
| ts_eager_fallback.h | ||
| ts_lowering_context.cpp | ||
| ts_lowering_context.h | ||
| ts_native_functions.cpp | ||
| ts_node_lowering.cpp | ||
| ts_node_lowering.h | ||
| ts_node.cpp | ||
| ts_node.h | ||