pytorch/torch/nn/modules
Junjie Bai 4864000e55 Add aten mkldnn ops: relu, max_pool2d and avg_pool2d
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/19205

Reviewed By: dzhulgakov

Differential Revision: D14850598

fbshipit-source-id: 5bbd5909c06df9c980de680ffb81bf772766c0ba
2019-04-26 13:41:44 -07:00
..
__init__.py Add an identity module (#19249) 2019-04-19 10:12:18 -07:00
_functions.py (#14267) 2019-03-06 13:39:11 -08:00
activation.py Add aten mkldnn ops: relu, max_pool2d and avg_pool2d 2019-04-26 13:41:44 -07:00
adaptive.py Correct docstring of vision/init functions 2019-03-01 11:40:23 -08:00
batchnorm.py Fix momentum setting in BatchNorm forward pass. (#18764) 2019-04-08 16:30:00 -07:00
container.py fix nn.Sequential doc 2019-04-23 14:58:16 -07:00
conv.py Removes variable which is assigned but not used (#19194) 2019-04-23 08:48:03 -07:00
distance.py Fix math formatting of PairwiseDistance and CosineSimilarity docs and fix math formatting of CTC loss docs. 2019-04-23 07:24:07 -07:00
dropout.py Correct recurrent/linear/dropout/sparse layers docstrings 2019-02-19 05:23:04 -08:00
fold.py Fix typo in docstring 2019-03-20 11:16:36 -07:00
instancenorm.py make InstanceNorm1d raise an error if the input is 2D (#11992) 2019-03-29 06:50:04 -07:00
linear.py Add an identity module (#19249) 2019-04-19 10:12:18 -07:00
loss.py Fix the documentation for BCEWithLogitsLoss (#17218, #16804) (#19212) 2019-04-23 09:54:24 -07:00
module.py Update module.py documentation. (#19347) 2019-04-18 06:46:24 -07:00
normalization.py Turn on F401: Unused import warning. (#18598) 2019-03-30 09:01:17 -07:00
padding.py Correct padding and activations docstrings in nn module 2019-02-19 08:16:52 -08:00
pixelshuffle.py Correct docstring of vision/init functions 2019-03-01 11:40:23 -08:00
pooling.py Turn on F401: Unused import warning. (#18598) 2019-03-30 09:01:17 -07:00
rnn.py Respect order of Parameters in rnn.py (#18198) 2019-04-18 11:18:20 -07:00
sparse.py EmbeddingBag CPU forward with per_sample_weights. (#18735) 2019-04-09 18:12:55 -07:00
upsampling.py Turn on F401: Unused import warning. (#18598) 2019-03-30 09:01:17 -07:00
utils.py migrating deprecated calls without abc module for containers (#11515) 2018-09-13 15:09:22 -07:00