| 297 | assets::require_tensor_shape( |
| 298 | weights, |
| 299 | "flow_matching.estimator.proj_out.ffn_2.weight", |
| 300 | {config.out_channels, config.out_channels}); |
| 301 | assets::require_tensor_shape( |
| 302 | weights, |
| 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, |
no outgoing calls
no test coverage detected