pytorch/torch/nn/utils
whywhy-rtx3090 7647c398ff Allow optional positional arguments for torch.func.functional_call (#134643)
This PR resolves #134408. Add an additional test and have passed the local test.

Do you think we should add a post-check to ensure `args` and `kwargs` are not both `None`? It seems to be possible to have modules without inputs.

This PR does not include any such post-check.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/134643
Approved by: https://github.com/zou3519
2024-09-12 15:22:06 +00:00
..
_expanded_weights [BE] enable UFMT for torch/nn/utils (#128595) 2024-06-13 18:34:57 +00:00
__init__.py [BE] enable UFMT for torch/nn/utils (#128595) 2024-06-13 18:34:57 +00:00
_deprecation_utils.py [BE] enable UFMT for torch/nn/utils (#128595) 2024-06-13 18:34:57 +00:00
_named_member_accessor.py Let dynamo inline functional_call (#128646) 2024-07-30 14:22:23 +00:00
_per_sample_grad.py [BE] enable UFMT for torch/nn/utils (#128595) 2024-06-13 18:34:57 +00:00
clip_grad.py Rewording doc string for clip_grad_norm_ (#133406) 2024-08-15 16:21:27 +00:00
convert_parameters.py [BE] enable UFMT for torch/nn/utils (#128595) 2024-06-13 18:34:57 +00:00
fusion.py Add proper casting to fuse_linear_bn_weights (#134105) 2024-08-22 14:26:12 +00:00
init.py [BE] enable UFMT for torch/nn/utils (#128595) 2024-06-13 18:34:57 +00:00
memory_format.py [docs] fix incorrect example in convert_conv3d_weight_memory_format (#129318) 2024-08-07 20:06:59 +00:00
parametrizations.py [BE] enable UFMT for torch/nn/utils (#128595) 2024-06-13 18:34:57 +00:00
parametrize.py Revert "[BE] typing for decorators - _jit_internal (#131573)" 2024-07-28 03:29:32 +00:00
prune.py [BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200) 2024-08-15 15:50:19 +00:00
rnn.py docs: torch.nn.utils.rnn.pack_padded_sequence: docs improve (#135417) 2024-09-09 03:16:11 +00:00
spectral_norm.py [BE] enable UFMT for torch/nn/utils (#128595) 2024-06-13 18:34:57 +00:00
stateless.py Allow optional positional arguments for torch.func.functional_call (#134643) 2024-09-12 15:22:06 +00:00
weight_norm.py [BE] enable UFMT for torch/nn/utils (#128595) 2024-06-13 18:34:57 +00:00