(self, x, memory, src_mask, tgt_mask)
| 99 | self.sublayer = clones(SublayerConnection(size), 3) |
| 100 | |
| 101 | def forward(self, x, memory, src_mask, tgt_mask): |
| 102 | m = memory |
| 103 | x = self.sublayer[0](x, lambda x: self.self_attn(x, x, x, tgt_mask)) |
| 104 | x = self.sublayer[1](x, lambda x: self.src_attn(x, m, m, src_mask)) |
| 105 | return self.sublayer[2](x, self.feed_forward) |
| 106 | |
| 107 | class MultiHeadedAttention(pl.LightningModule): |
| 108 |
nothing calls this directly
no outgoing calls
no test coverage detected