(self, hidden_states: torch.Tensor)
| 309 | return output |
| 310 | |
| 311 | def forward(self, hidden_states: torch.Tensor) -> torch.Tensor: |
| 312 | if self.use_conv: |
| 313 | height = self._upsample_2d(hidden_states, self.Conv2d_0.weight, kernel=self.fir_kernel) |
| 314 | height = height + self.Conv2d_0.bias.reshape(1, -1, 1, 1) |
| 315 | else: |
| 316 | height = self._upsample_2d(hidden_states, kernel=self.fir_kernel, factor=2) |
| 317 | |
| 318 | return height |
| 319 | |
| 320 | |
| 321 | class KUpsample2D(nn.Module): |
nothing calls this directly
no test coverage detected