| 303 | "scalar_model.encoder.0.parametrizations.weight.original1", |
| 304 | {config.init_channel, config.num_bands, config.default_kernel_size}); |
| 305 | assets::require_tensor_shape( |
| 306 | weights, |
| 307 | "scalar_model.encoder.1.conv.weight", |
| 308 | {config.init_channel, config.init_channel, config.default_kernel_size}); |
| 309 | assets::require_tensor_shape( |
| 310 | weights, |
| 311 | "scalar_model.decoder.0.parametrizations.weight.original1", |
| 312 | {decoder_in_channels, config.latent_hidden_dim, config.delay_kernel_size}); |
| 313 | assets::require_tensor_shape( |
| 314 | weights, |
| 315 | "scalar_model.decoder.1.up_conv.layer.parametrizations.weight.original1", |
nothing calls this directly
no test coverage detected