pytorch/torch/nn
linmin 2a286cbdf4 Allow register_buffer with Tensor-like object (#159455)
As torch allows extending the tensor with `__torch_function__`, it would be desirable to allow registering it as a buffer.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/159455
Approved by: https://github.com/mikaylagawarecki
2025-08-01 15:31:38 +00:00
..
attention Add kernel options to flex docs (#158875) 2025-07-23 19:05:19 +00:00
backends
intrinsic
modules Allow register_buffer with Tensor-like object (#159455) 2025-08-01 15:31:38 +00:00
parallel XCCL changes for DDP (#155497) 2025-07-03 05:18:08 +00:00
qat [BE][PYFMT] migrate PYFMT for {torch,test}/{nn,optim}/** to ruff format (#144548) 2025-06-14 11:27:04 +00:00
quantizable [BE][PYFMT] migrate PYFMT for {torch,test}/{nn,optim}/** to ruff format (#144548) 2025-06-14 11:27:04 +00:00
quantized [BE][PYFMT] migrate PYFMT for {torch,test}/{nn,optim}/** to ruff format (#144548) 2025-06-14 11:27:04 +00:00
utils [BE] More torch.nn docs coverage test (except for torch.nn.parallel) (#158654) 2025-07-25 22:03:55 +00:00
__init__.py
_reduction.py
common_types.py [BE][Ez]: Denote common types as TypeAlias (#154527) 2025-05-29 02:00:13 +00:00
cpp.py
functional.py Fix the Doc of padding in avg_poolnd (#159142) 2025-07-31 02:02:48 +00:00
functional.pyi.in fix type hints for interpolation functions (#157202) 2025-07-09 03:11:37 +00:00
grad.py
init.py [BE][12/16] fix typos in torch/ (#156602) 2025-07-02 22:55:29 +00:00
parameter.py [BE] More torch.nn docs coverage test (except for torch.nn.parallel) (#158654) 2025-07-25 22:03:55 +00:00
parameter.pyi remove allow-untyped-defs from torch/nn/parameter.pyi (#144654) 2025-01-13 19:02:31 +00:00