(self, x, kernel, **kwargs)
| 62 | return "xcorr_fast" |
| 63 | |
| 64 | def forward(self, x, kernel, **kwargs): |
| 65 | if self.training: |
| 66 | if self.use_slow: |
| 67 | return xcorr_slow(x, kernel, kwargs) |
| 68 | else: |
| 69 | return xcorr_fast(x, kernel, kwargs) |
| 70 | else: |
| 71 | return Corr.apply(x, kernel, 1, kwargs) |
| 72 | |
| 73 | class DRConv2d(nn.Module): |
| 74 | def __init__(self, in_channels, out_channels, kernel_size, region_num=2, **kwargs): |
nothing calls this directly
no test coverage detected