(self, x)
| 147 | self.shortcut = nn.Identity() |
| 148 | |
| 149 | def forward(self, x): |
| 150 | if self.up: |
| 151 | in_rest, in_conv = self.in_layers[:-1], self.in_layers[-1] |
| 152 | h = in_rest(x) |
| 153 | h = F.interpolate(h, scale_factor=2, mode="bilinear", align_corners=False) |
| 154 | x = F.interpolate(x, scale_factor=2, mode="bilinear", align_corners=False) |
| 155 | h = in_conv(h) |
| 156 | else: |
| 157 | h = self.in_layers(x) |
| 158 | |
| 159 | h = self.out_layers(h) |
| 160 | h = h + self.shortcut(x) |
| 161 | return h |
| 162 | |
| 163 | |
| 164 | def compose_triplane_channelwise(feat_maps): |
nothing calls this directly
no outgoing calls
no test coverage detected