pytorch/torch/nn
2023-12-18 21:29:14 +00:00
..
attention Fix backward for SDPA NT jagged layout (#115576) 2023-12-12 18:35:40 +00:00
backends
intrinsic docs: fix docstring errors in quantized modules and others (#112695) 2023-11-07 23:52:16 +00:00
modules [Doc] Add padding size constraint in nn.ReflectionPad2d (#115995) 2023-12-18 21:29:14 +00:00
parallel DistributedDataParallel._post_forward, fix return (#114678) 2023-12-06 23:44:52 +00:00
qat docs: fix docstring errors in quantized modules and others (#112695) 2023-11-07 23:52:16 +00:00
quantizable docs: fix docstring errors in quantized modules and others (#112695) 2023-11-07 23:52:16 +00:00
quantized docs: fix docstring errors in quantized modules and others (#112695) 2023-11-07 23:52:16 +00:00
utils improve annotation device parameters where a device ordinal is allowed (#113647) 2023-11-17 14:41:22 +00:00
__init__.py Fix docstring errors in reductions.py, spawn.py, pool.py, parameter.py, cpp.py, grad.py, __init__.py, profiler.py, queue.py, graph.py (#113052) 2023-11-10 21:19:17 +00:00
_reduction.py [BE] Enable ruff's UP rules and autoformat nn/ mps/ and torch/ (#105436) 2023-07-21 07:38:46 +00:00
common_types.py
cpp.py Fix docstring errors in reductions.py, spawn.py, pool.py, parameter.py, cpp.py, grad.py, __init__.py, profiler.py, queue.py, graph.py (#113052) 2023-11-10 21:19:17 +00:00
functional.py Add decompositions for replication_pad (#115113) 2023-12-09 02:44:07 +00:00
functional.pyi.in Add python and C++ support for LPPool3d (#114199) 2023-12-08 18:18:44 +00:00
grad.py Fix docstring errors in reductions.py, spawn.py, pool.py, parameter.py, cpp.py, grad.py, __init__.py, profiler.py, queue.py, graph.py (#113052) 2023-11-10 21:19:17 +00:00
init.py Add support for torch.Generator type in TorchScript (#110413) 2023-11-21 23:07:21 +00:00
parameter.py Fix docstring errors in reductions.py, spawn.py, pool.py, parameter.py, cpp.py, grad.py, __init__.py, profiler.py, queue.py, graph.py (#113052) 2023-11-10 21:19:17 +00:00
parameter.pyi Back out "Reland "Make adding buffers more like adding parameters (#104069)" (#106224)" (#106743) 2023-08-08 15:27:34 +00:00