(self, tgt,
tgt_mask: Optional[Tensor] = None,
tgt_key_padding_mask: Optional[Tensor] = None,
query_pos: Optional[Tensor] = None)
| 57 | return tgt |
| 58 | |
| 59 | def forward(self, tgt, |
| 60 | tgt_mask: Optional[Tensor] = None, |
| 61 | tgt_key_padding_mask: Optional[Tensor] = None, |
| 62 | query_pos: Optional[Tensor] = None): |
| 63 | if self.normalize_before: |
| 64 | return self.forward_pre(tgt, tgt_mask, |
| 65 | tgt_key_padding_mask, query_pos) |
| 66 | return self.forward_post(tgt, tgt_mask, |
| 67 | tgt_key_padding_mask, query_pos) |
| 68 | |
| 69 | |
| 70 | class CrossAttentionLayer(nn.Module): |
nothing calls this directly
no test coverage detected