pytorch/test/cpp
Joel Schlosser ee482edf0a Callable activation function support for Transformer modules (C++) (#62342)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/60747

Enhances the C++ versions of `Transformer`, `TransformerEncoderLayer`, and `TransformerDecoderLayer` to support callables as their activation functions. The old way of specifying activation function still works as well.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/62342

Reviewed By: malfet

Differential Revision: D30022592

Pulled By: jbschlosser

fbshipit-source-id: d3c62410b84b1bd8c5ed3a1b3a3cce55608390c4
2021-08-02 08:06:39 -07:00
..
api Callable activation function support for Transformer modules (C++) (#62342) 2021-08-02 08:06:39 -07:00
c10d Exclude time of waiting in queue from gloo communication prof… (#61342) 2021-07-28 22:24:26 -07:00
common
dist_autograd Fix distributed autograd gradients synchronization (#57792) 2021-05-09 17:32:59 -07:00
jit [PyTorch Edge] Add test for lite interpreter operator caching (#62306) 2021-07-29 20:14:45 -07:00
lite_interpreter_runtime [Pytorch Delegated Backend] Save function name in debug info (#57481) 2021-05-25 13:19:02 -07:00
rpc Move torch/lib/c10d to torch/csrc/distributed/c10d (#60543) 2021-06-24 12:38:51 -07:00
tensorexpr Modernize override (#61744) 2021-07-23 23:04:46 -07:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00