Function load_weight_norm_conv1d

src/models/ace_step/vae_decoder.cpp:242–270

}

WeightNormConv1dWeights load_weight_norm_conv1d(
    core::BackendWeightStore & store,
    const assets::TensorSource & source,
    const std::string & prefix,
    int64_t out_channels,
    int64_t in_channels,
    int64_t kernel,
    int stride,
    int padding,
    int dilation,
    assets::TensorStorageType storage_type,
    bool use_bias) {
    const auto g = source.require_f32(prefix + ".weight_g", {out_channels, 1, 1});
    const auto v = source.require_f32(prefix + ".weight_v", {out_channels, in_channels, kernel});
    const auto graph_storage_type =
        compatible_regular_conv_storage_type(source, prefix + ".weight_v", storage_type);
    WeightNormConv1dWeights out = {};
    out.stride = stride;
    out.padding = padding;
    out.dilation = dilation;
    out.conv.weight = store.make_from_f32(
        core::TensorShape::from_dims({out_channels, in_channels, kernel}),
        graph_storage_type,
        apply_weight_norm(g, v, out_channels, in_channels * kernel));
    if (use_bias) {
        out.conv.bias = store.load_f32_tensor(source, prefix + ".bias", {out_channels});
    }
    return out;
}

WeightNormConvTranspose1dWeights load_weight_norm_conv_transpose1d(
    core::BackendWeightStore & store,