| .. |
|
_expanded_weights
|
expanded weights: instance norm faster rule
|
2022-04-19 19:40:09 +00:00 |
|
__init__.py
|
move the stateless util to public API!
|
2022-04-21 13:42:24 +00:00 |
|
_per_sample_grad.py
|
expanded weights without fast rules (#70140)
|
2022-02-22 20:35:16 +00:00 |
|
_stateless.py
|
Added setattr to functional_call. (#77137)
|
2022-05-17 05:40:46 +00:00 |
|
clip_grad.py
|
Update clip_grad_norm_ documentation
|
2022-04-22 14:07:34 +00:00 |
|
convert_parameters.py
|
[*.py] Rename "Arguments:" to "Args:" (#49736)
|
2020-12-28 09:34:47 -08:00 |
|
fusion.py
|
[fusion] Add ConvTranspose+BN fusion support (#70022)
|
2021-12-20 18:42:48 -08:00 |
|
init.py
|
Helper function for skipping module parameter / buffer initialization (#57555)
|
2021-05-24 11:28:32 -07:00 |
|
memory_format.py
|
|
|
|
parametrizations.py
|
Prefer mT and mH over transpose(-2, -1) and transpose(-2, -1).conj() (#64181)
|
2021-10-18 13:02:25 -07:00 |
|
parametrize.py
|
Support for tensor subclasses as parameters
|
2022-04-27 19:28:55 +00:00 |
|
prune.py
|
[lint] upgrade mypy to latest version
|
2022-05-03 20:51:34 +00:00 |
|
rnn.py
|
Convert type comments to annotations in torch/nn (#72662)
|
2022-02-11 06:35:42 +00:00 |
|
rnn.pyi
|
Add pad_sequence as a native function (#57868)
|
2021-05-11 11:18:13 -07:00 |
|
spectral_norm.py
|
Type annotate torch.nn.Module ctor (#61334)
|
2021-07-16 13:59:06 -07:00 |
|
stateless.py
|
Added setattr to functional_call. (#77137)
|
2022-05-17 05:40:46 +00:00 |
|
weight_norm.py
|
torch.nn.modules.LazyModuleMixin and torch.nn.LazyLinear (Shape Inference II) (#44538)
|
2020-10-19 13:13:54 -07:00 |