saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/torch/nn/parallel
History
Xuehai Pan 93a33bf3ac [BE] update type annotations for basic utilities in torch/__init__.py (#129001) 
Changes:

1. Make some arguments positional-only as we only support Python 3.8+
2. Clean up `torch.typename(obj)` implementation.
3. Update type annotations., especially `is_tensor()` and `is_masked_tensor()` using `TypeGuard`.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/129001
Approved by: https://github.com/malfet
2024-06-24 18:04:38 +00:00
..
__init__.py [BE][Easy] enable UFMT for torch/nn/parallel (#128596) 2024-06-17 16:29:22 +00:00
_functions.py [BE][Easy] enable UFMT for torch/nn/parallel (#128596) 2024-06-17 16:29:22 +00:00
comm.py [BE][Easy] enable UFMT for torch/nn/parallel (#128596) 2024-06-17 16:29:22 +00:00
data_parallel.py [BE][Easy] enable UFMT for torch/nn/parallel (#128596) 2024-06-17 16:29:22 +00:00
distributed.py [BE] update type annotations for basic utilities in torch/__init__.py (#129001) 2024-06-24 18:04:38 +00:00
parallel_apply.py [BE][Easy] enable UFMT for torch/nn/parallel (#128596) 2024-06-17 16:29:22 +00:00
replicate.py [BE][Easy] enable UFMT for torch/nn/parallel (#128596) 2024-06-17 16:29:22 +00:00
scatter_gather.py Document the torch.nn.parallel.scatter_gather.gather function (#128566) 2024-06-17 16:44:17 +00:00