(self, x, size=None)
| 404 | self.prenorm = prenorm |
| 405 | |
| 406 | def forward(self, x, size=None): |
| 407 | x = x + self.drop_path(self.mixer(self.norm1(x))) |
| 408 | x = x + self.drop_path(self.mlp(self.norm2(x))) |
| 409 | return x |
| 410 | |
| 411 | |
| 412 | class OSRA_Attention(nn.Module): ### OSRA |
nothing calls this directly
no outgoing calls
no test coverage detected