Lines Matching refs:TransformerEncoderLayer
218 layer = nn.TransformerEncoderLayer(
305 model = torch.nn.TransformerEncoderLayer(
332 torch.nn.TransformerEncoderLayer(
419 …torch.nn.TransformerEncoderLayer(d_model=4, nhead=2, dim_feedforward=16, dropout=0.0, batch_first=…
465 layer = nn.TransformerEncoderLayer(
651 encoder_layer = nn.TransformerEncoderLayer(
722 layer = torch.nn.TransformerEncoderLayer(d_model, 1, 6, batch_first=True)
763 encoder_layer = nn.TransformerEncoderLayer(
784 encoder_layer = nn.TransformerEncoderLayer(
811 class MyCustomLayer(nn.TransformerEncoderLayer):
822 class MyCustomLayer(nn.TransformerEncoderLayer):
857 class MyCustomLayer(nn.TransformerEncoderLayer):
1104 …encoder_layer = torch.nn.TransformerEncoderLayer(d_model=2, nhead=2, dim_feedforward=8, batch_firs…
1151 …encoder_layer = torch.nn.TransformerEncoderLayer(d_model, nhead, bias=False, batch_first=True, dev…
1170 layer = nn.TransformerEncoderLayer(
1292 …encoder_layer = torch.nn.TransformerEncoderLayer(d_model=2, nhead=2, dim_feedforward=8, batch_firs…