Args: x: shape (b, c, t)
(self, x: torch.Tensor)
| 136 | ) |
| 137 | |
| 138 | def forward(self, x: torch.Tensor): |
| 139 | """ |
| 140 | Args: |
| 141 | x: shape (b, c, t) |
| 142 | """ |
| 143 | h = x |
| 144 | h = self.block1(h) |
| 145 | h = self.block2(h) |
| 146 | if self.in_channels != self.out_channels: |
| 147 | x = self.nin_shortcut(x) |
| 148 | return x + h |
| 149 | |
| 150 | def forward_chunk(self, x: torch.Tensor, cache: torch.Tensor = None): |
| 151 | """ |
nothing calls this directly
no outgoing calls
no test coverage detected