Akifumi Imanishi aa1fd6b45a Add LazyBatchNormXd (#51548) ... Summary: This PR implements UninitializedBuffer and LazyBatchnormXd based on https://github.com/pytorch/pytorch/issues/44538. (cc. emcastillo and albanD) Pull Request resolved: https://github.com/pytorch/pytorch/pull/51548 Reviewed By: zhangguanheng66 Differential Revision: D26276903 Pulled By: albanD fbshipit-source-id: 0ac706974178363f8af075e59b41d5989418922f		2021-02-05 10:27:04 -08:00
..
__init__.py	Add LazyBatchNormXd (#51548)	2021-02-05 10:27:04 -08:00
_functions.py	Fix SyncBatchNorm usage without stats tracking (#50126)	2021-01-07 18:31:13 -08:00
activation.py	Drop unused imports (#49972)	2021-01-13 12:26:17 -08:00
adaptive.py	Move all torch.nn.modules type annotations inline (#38211)	2020-06-11 15:59:57 -07:00
batchnorm.py	Add LazyBatchNormXd (#51548)	2021-02-05 10:27:04 -08:00
channelshuffle.py	Add --check-untyped-defs to mypy.ini and test suite (#37594)	2020-05-07 06:36:01 -07:00
container.py	add type annotations to torch.nn.modules.container (#48969)	2021-01-19 15:12:17 -08:00
conv.py	add type annotations to conv_fused/blas_compare/blas_compare_setup (#51235)	2021-02-02 08:50:49 -08:00
distance.py	Move all torch.nn.modules type annotations inline (#38211)	2020-06-11 15:59:57 -07:00
dropout.py	Fix HTTP links in documentation to HTTPS (#40878)	2020-07-06 20:05:21 -07:00
flatten.py	added List as an option to the unflattened_size (#49838)	2020-12-29 16:50:37 -08:00
fold.py	Move all torch.nn.modules type annotations inline (#38211)	2020-06-11 15:59:57 -07:00
instancenorm.py	annotate a few torch.nn.modules.* modules (#45772)	2020-11-02 13:04:59 -08:00
lazy.py	Add LazyBatchNormXd (#51548)	2021-02-05 10:27:04 -08:00
linear.py	Add type annotations to torch.overrides (#50824)	2021-01-25 13:20:09 -08:00
loss.py	Add Gaussian NLL Loss (#50886)	2021-01-22 06:56:49 -08:00
module.py	Add LazyBatchNormXd (#51548)	2021-02-05 10:27:04 -08:00
normalization.py	add type annotations to torch.nn.modules.normalization (#49035)	2021-01-12 07:40:15 -08:00
padding.py	Add type annotations to torch.nn.modules.padding (#49494)	2021-01-12 15:34:28 -08:00
pixelshuffle.py	Add PixelUnshuffle (#49334)	2020-12-22 20:14:55 -08:00
pooling.py	Fix mypy type hint for AdaptiveAvgPool2,3d, AdaptiveMaxPool2,3d (#49963)	2021-01-06 09:47:15 -08:00
rnn.py	unbreak mypy torch/quantization (#49549)	2020-12-21 21:02:48 -08:00
sparse.py	Support int32 indices and offsets in nn.EmbeddingBag (#46758)	2020-11-03 23:33:50 -08:00
transformer.py	Move all torch.nn.modules type annotations inline (#38211)	2020-06-11 15:59:57 -07:00
upsampling.py	Enable typechecks for torch.nn.modules.[activation|upsampling] (#44093)	2020-09-03 13:20:04 -07:00
utils.py	Drop unused imports (#49972)	2021-01-13 12:26:17 -08:00