Commit 63b5b166 authored by Jun Ru Anderson's avatar Jun Ru Anderson Committed by Mandeep Singh Baines
Browse files

[fix] add TransformerEncoderLayer to stubs (#5)


Co-authored-by: default avatarJun Ru Anderson <andersonic@fb.com>
parent e4b2ffd4
...@@ -44,6 +44,7 @@ from .pooling import AdaptiveAvgPool1d as AdaptiveAvgPool1d, AdaptiveAvgPool2d a ...@@ -44,6 +44,7 @@ from .pooling import AdaptiveAvgPool1d as AdaptiveAvgPool1d, AdaptiveAvgPool2d a
from .rnn import GRU as GRU, GRUCell as GRUCell, LSTM as LSTM, LSTMCell as LSTMCell, RNN as RNN, RNNBase as RNNBase, \ from .rnn import GRU as GRU, GRUCell as GRUCell, LSTM as LSTM, LSTMCell as LSTMCell, RNN as RNN, RNNBase as RNNBase, \
RNNCell as RNNCell, RNNCellBase as RNNCellBase RNNCell as RNNCell, RNNCellBase as RNNCellBase
from .sparse import Embedding as Embedding, EmbeddingBag as EmbeddingBag from .sparse import Embedding as Embedding, EmbeddingBag as EmbeddingBag
from .transformer import TransformerEncoderLayer as TransformerEncoderLayer
from .upsampling import Upsample as Upsample, UpsamplingBilinear2d as UpsamplingBilinear2d, \ from .upsampling import Upsample as Upsample, UpsamplingBilinear2d as UpsamplingBilinear2d, \
UpsamplingNearest2d as UpsamplingNearest2d UpsamplingNearest2d as UpsamplingNearest2d
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment