| 193 | |
| 194 | #[derive(Module, Debug)] |
| 195 | pub struct Decoder<B: Backend> { |
| 196 | conv_in: Conv2d<B>, |
| 197 | mid: Mid<B>, |
| 198 | blocks: Vec<DecoderBlock<B>>, |
| 199 | norm_out: GroupNorm<B>, |
| 200 | silu: SILU, |
| 201 | conv_out: Conv2d<B>, |
| 202 | } |
| 203 | |
| 204 | impl<B: Backend> Decoder<B> { |
| 205 | fn forward(&self, x: Tensor<B, 4>) -> Tensor<B, 4> { |
nothing calls this directly
no outgoing calls
no test coverage detected