(self, x)
| 625 | self.out_channel = out_channel |
| 626 | |
| 627 | def forward(self, x): |
| 628 | B, C, H, W = x.shape |
| 629 | x = self.proj(x).flatten(2).transpose(1, 2).contiguous() # B H*W C |
| 630 | if self.norm is not None: |
| 631 | x = self.norm(x) |
| 632 | return x |
| 633 | |
| 634 | def flops(self, H, W): |
| 635 | flops = 0 |
nothing calls this directly
no outgoing calls
no test coverage detected