(self, x)
| 207 | self.op = avg_pool_nd(dims, kernel_size=stride, stride=stride) |
| 208 | |
| 209 | def forward(self, x): |
| 210 | assert x.shape[1] == self.channels |
| 211 | return self.op(x) |
| 212 | |
| 213 | |
| 214 | class ResBlock(TimestepBlock): |
nothing calls this directly
no outgoing calls
no test coverage detected