BypassModule: combines original (pre-downsample) with processed (upsampled) output. `output = bypass_scale * src_orig + (1 - bypass_scale) * processed`
| 31 | /// BypassModule: combines original (pre-downsample) with processed (upsampled) output. |
| 32 | /// `output = bypass_scale * src_orig + (1 - bypass_scale) * processed` |
| 33 | struct BypassModule { |
| 34 | bypass_scale: Tensor, // [dim] |
| 35 | } |
| 36 | |
| 37 | impl BypassModule { |
| 38 | fn load(dim: usize, vb: VarBuilder) -> Result<Self> { |
nothing calls this directly
no outgoing calls
no test coverage detected