(self, x, relative_pos_enc=None)
| 504 | return x |
| 505 | |
| 506 | def forward(self, x, relative_pos_enc=None): |
| 507 | if self.grad_checkpoint and x.requires_grad: |
| 508 | x = checkpoint.checkpoint(self._forward_impl, x, relative_pos_enc) |
| 509 | else: |
| 510 | x = self._forward_impl(x, relative_pos_enc) |
| 511 | return x |
| 512 | |
| 513 | |
| 514 | class RetBlock(nn.Module): |
nothing calls this directly
no test coverage detected