pytorch/torch/nn
2022-12-10 14:37:31 +00:00
..
backends
intrinsic [quant][ao_migration] nn.intrinsic.quantized migration to ao (#86172) 2022-10-08 00:01:38 +00:00
modules [Resubmit] state_dict_pre_hook (#90435) 2022-12-08 07:54:14 +00:00
parallel Fix non-existing parameters in docstrings in torch/nn (#90596) 2022-12-10 14:37:31 +00:00
qat [quant] Subpackage import in nn.quantized (#84141) 2022-09-01 11:35:03 +00:00
quantizable [quant][ao_migration] torch.nn.quantizabletorch.ao.nn.quantizable. (#78717) 2022-08-25 16:50:37 +00:00
quantized Rename 'torch/ao/nn/quantized._reference' to 'torch/ao/nn/quantized/reference'. (#84974) 2022-09-16 17:49:07 +00:00
utils Fix non-existing parameters in docstrings in torch/nn (#90596) 2022-12-10 14:37:31 +00:00
__init__.py
_reduction.py
common_types.py
cpp.py
functional.py Call _sdp_attention in nn.functional.mha (#89470) 2022-12-02 19:46:22 +00:00
functional.pyi.in
grad.py Integrate xdoctest - Rebased (#82797) 2022-08-12 02:08:01 +00:00
init.py More doctest refinements. (#83317) 2022-08-22 20:07:26 +00:00
parameter.py Revert "[follow-up] Python Attr Serialization (#88913)" 2022-12-02 20:14:11 +00:00
parameter.pyi