| 79 | kCausalResBlockInputs, |
| 80 | 7, |
| 81 | kSingleOutput, |
| 82 | 1, |
| 83 | "Applies a PixelNorm/Silu causal Conv2d residual block.", |
| 84 | }; |
| 85 | |
| 86 | const core::ModuleSchema kCausalConv2dUpsampleSchema = { |
| 87 | "CausalConv2dUpsample", |
| 88 | "nn.conv", |
| 89 | kConvInputs, |
| 90 | 3, |
| 91 | kSingleOutput, |
| 92 | 1, |
| 93 | "Applies nearest 2D upsampling followed by causal Conv2d and temporal crop.", |
| 94 | }; |
| 95 | |
| 96 | const core::ModuleSchema kDepthwiseConv2dSchema = { |
| 97 | "DepthwiseConv2d", |
| 98 | "nn.conv", |
| 99 | kConvInputs, |
| 100 | 3, |
| 101 | kSingleOutput, |
| 102 | 1, |
| 103 | "Applies a depthwise 2D convolution to channel-first inputs [batch, channels, height, width].", |
| 104 | }; |
| 105 | |
| 106 | const core::ModuleSchema kConvTranspose1dSchema = { |
no test coverage detected