(self, config)
| 397 | |
| 398 | class BertSelfOutput(nn.Module): |
| 399 | def __init__(self, config): |
| 400 | super(BertSelfOutput, self).__init__() |
| 401 | self.dense = nn.Linear(config.hidden_size, config.hidden_size) |
| 402 | self.LayerNorm = BertLayerNorm(config.hidden_size, eps=1e-12) |
| 403 | self.dropout = nn.Dropout(config.hidden_dropout_prob) |
| 404 | |
| 405 | def forward(self, hidden_states, input_tensor): |
| 406 | hidden_states = self.dense(hidden_states) |
nothing calls this directly
no test coverage detected