| 60 | 3, |
| 61 | kSingleOutput, |
| 62 | 1, |
| 63 | "Applies explicit asymmetric 2D padding followed by Conv2d.", |
| 64 | }; |
| 65 | |
| 66 | const core::ModuleSchema kSameWidthCausalConv2dSchema = { |
| 67 | "SameWidthCausalConv2d", |
| 68 | "nn.conv", |
| 69 | kConvInputs, |
| 70 | 3, |
| 71 | kSingleOutput, |
| 72 | 1, |
| 73 | "Applies causal-height and same-width 2D padding followed by Conv2d.", |
| 74 | }; |
| 75 | |
| 76 | const core::ModuleSchema kPixelNormCausalConv2dResBlockSchema = { |
| 77 | "PixelNormCausalConv2dResBlock", |
| 78 | "nn.conv", |
| 79 | kCausalResBlockInputs, |
| 80 | 7, |
| 81 | kSingleOutput, |
| 82 | 1, |
no test coverage detected