(self, hidden_states)
| 481 | self.LayerNorm = BertLayerNorm(config.hidden_size, eps=config.layer_norm_eps) |
| 482 | |
| 483 | def forward(self, hidden_states): |
| 484 | hidden_states = self.dense(hidden_states) |
| 485 | hidden_states = self.transform_act_fn(hidden_states) |
| 486 | hidden_states = self.LayerNorm(hidden_states) |
| 487 | return hidden_states |
| 488 | |
| 489 | |
| 490 | class BertLMPredictionHead(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected