Guilherme Leobas 7bf3b7cdc5 Rewrite _reparametrize_module to use contextmanager (#138203) ... Pull Request resolved: https://github.com/pytorch/pytorch/pull/138203 Approved by: https://github.com/zou3519 ghstack dependencies: #136033, #140604		2024-12-20 12:02:27 +00:00
..
attention	[FlexAttention] Optimzing learned bias perf to dq calc (#142281)	2024-12-15 21:44:32 +00:00
backends
intrinsic
modules	[BE] replace incorrect .. note:: invocations (#142868)	2024-12-11 19:58:18 +00:00
parallel	remove allow-untyped-defs for torch/nn/parallel/__init__.py (#143437)	2024-12-18 08:50:37 +00:00
qat
quantizable
quantized
utils	Rewrite _reparametrize_module to use contextmanager (#138203)	2024-12-20 12:02:27 +00:00
__init__.py
_reduction.py
common_types.py
cpp.py
functional.py	Fix mha torch._check in jit tracing (#142059)	2024-12-05 18:38:17 +00:00
functional.pyi.in	Gaussian nll loss scalar variance support (#138931)	2024-11-21 18:20:09 +00:00
grad.py
init.py
parameter.py
parameter.pyi