pytorch/test/cpp_api_parity
Will Feng d041d0784e [C++ API] RNNCell / LSTMCell / GRUCell layers (#34400)
Summary:
This PR adds `RNNCell` / `LSTMCell` / `GRUCell` layers to the C++ frontend, with implementations exactly matching the Python API equivalent.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/34400

Differential Revision: D20316859

Pulled By: yf225

fbshipit-source-id: bb7cee092622334043c0d0fd0fcb4e75e707699c
2020-03-13 21:52:24 -07:00
..
__init__.py Improve C++ maxpool and avgpool (#26521) 2019-09-25 13:52:58 -07:00
parity-tracker.md [C++ API] RNNCell / LSTMCell / GRUCell layers (#34400) 2020-03-13 21:52:24 -07:00
sample_module.py Improve C++ maxpool and avgpool (#26521) 2019-09-25 13:52:58 -07:00
torch_nn_modules.py Add torch.nn.GELU for GELU activation (#28944) 2019-11-03 21:55:05 -08:00