(self, x)
| 1476 | named_apply(partial(_init_conv, scheme=scheme), self.conv) |
| 1477 | |
| 1478 | def forward(self, x): |
| 1479 | x = self.conv(x) |
| 1480 | x = x.permute(0, 2, 3, 1) |
| 1481 | x = self.attn(x) |
| 1482 | x = x.permute(0, 3, 1, 2) |
| 1483 | return x |
| 1484 | |
| 1485 | |
| 1486 | class MaxxVitStage(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected