(self, i, config, weights, biases)
| 432 | class BertLayer(nn.Module): |
| 433 | |
| 434 | def __init__(self, i, config, weights, biases): |
| 435 | super(BertLayer, self).__init__() |
| 436 | self.attention = BertAttention(i, config, weights, biases) |
| 437 | self.intermediate = BertIntermediate(config, weights, biases) |
| 438 | self.output = BertOutput(config, weights, biases) |
| 439 | self.weight = weights |
| 440 | self.biases = biases |
| 441 | |
| 442 | def forward(self, hidden_states, attention_mask, grads, collect_all_grads=False): |
| 443 | attention_output = self.attention(hidden_states, attention_mask) |
nothing calls this directly
no test coverage detected