OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 37 Packages Projects Releases Wiki Activity
1d733d660d
pytorch/torch/nn
History
Yanli Zhao f5d231d593 move rebuild buckets from end of first iteration to beginning of second iteration (#44326) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/44326

Part of relanding PR #41954, this refactoring is to move rebuild_buckets call from end of first iteration to beginning of second iteration
ghstack-source-id: 112011490

Test Plan: unit tests

Reviewed By: mrshenli

Differential Revision: D23583017

fbshipit-source-id: ef67f79437a820d9b5699b651803622418499a83
2020-09-15 09:51:33 -07:00
..
backends remediation of S205607 2020-07-17 17:19:47 -07:00
intrinsic qat conv_fused.py: one more patch for forward compatibility (#44671) 2020-09-15 09:43:29 -07:00
modules add dilation to transposeconv's _output_padding method (#43793) 2020-09-14 21:28:27 -07:00
parallel move rebuild buckets from end of first iteration to beginning of second iteration (#44326) 2020-09-15 09:51:33 -07:00
qat Remove useless py2 compatibility import __future__, part 1 (#43808) 2020-09-02 19:15:11 -07:00
quantized [quant] ConvTranspose1d / ConvTranspose2d (#40371) 2020-09-14 14:25:06 -07:00
utils Add type annotations for torch.nn.utils.* (#43080) 2020-09-14 17:52:37 -07:00
__init__.py Ignore F401 in all __init__.py without putting noqa (#25823) 2019-10-23 15:28:13 -07:00
_reduction.py Fix type annotations and make MyPy run on torch/ (#36584) 2020-04-22 14:17:08 -07:00
common_types.py Move all torch.nn.modules type annotations inline (#38211) 2020-06-11 15:59:57 -07:00
cpp.py
functional.py Fix SmoothL1Loss when target.requires_grad is True. (#44486) 2020-09-11 12:13:36 -07:00
functional.pyi.in Enable typechecks for torch.nn.modules.[activation|upsampling] (#44093) 2020-09-03 13:20:04 -07:00
grad.py Grad input padding support for dilation argument (#33872) 2020-04-09 11:09:55 -07:00
init.py docs: Fixed docstring indentation for documentation (#37739) 2020-05-04 19:08:55 -07:00
parameter.py Add __torch_function__ for methods (#37091) 2020-08-05 20:44:13 -07:00
parameter.pyi Fix wrong typing (torch/nn/parameter.pyi) (#32617) 2020-01-25 16:19:33 -08:00