Function graph_dense_block

src/framework/audio/zipenhancer.cpp:252–285

}

core::TensorValue graph_dense_block(
    core::ModuleBuildContext & ctx,
    const core::TensorValue & input,
    const std::unordered_map<std::string, Param> & params,
    const std::string & prefix) {
    auto skip = input;
    auto x = input;
    for (int i = 0; i < 4; ++i) {
        const int64_t dilation = int64_t{1} << i;
        const int64_t pad_top = dilation;
        const std::string base = prefix + ".dense_block." + std::to_string(i);
        x = graph_conv2d(
            ctx,
            graph_pad2d(ctx, skip, 1, 1, pad_top, 0),
            params,
            base + ".1.weight",
            base + ".1.bias",
            0,
            0,
            1,
            1,
            dilation,
            1);
        x = graph_instance_norm_prelu(
            ctx,
            x,
            params,
            base + ".2.weight",
            base + ".2.bias",
            base + ".3.weight");
        skip = modules::ConcatModule({1}).build(ctx, x, skip);
    }
    return x;
}

core::TensorValue graph_dense_encoder(
    core::ModuleBuildContext & ctx,