| 248 | |
| 249 | #[derive(Module, Debug)] |
| 250 | pub struct EncoderBlock<B: Backend> { |
| 251 | res1: ResnetBlock<B>, |
| 252 | res2: ResnetBlock<B>, |
| 253 | downsampler: Option<PaddedConv2d<B>>, |
| 254 | } |
| 255 | |
| 256 | impl<B: Backend> EncoderBlock<B> { |
| 257 | fn forward(&self, x: Tensor<B, 4>) -> Tensor<B, 4> { |
nothing calls this directly
no outgoing calls
no test coverage detected