Function graph_dense_encoder

src/framework/audio/zipenhancer.cpp:287–330

}

core::TensorValue graph_dense_encoder(
    core::ModuleBuildContext & ctx,
    const core::TensorValue & input,
    const std::unordered_map<std::string, Param> & params) {
    auto x = graph_conv2d(
        ctx,
        input,
        params,
        "dense_encoder.dense_conv_1.0.weight",
        "dense_encoder.dense_conv_1.0.bias",
        0,
        0,
        1,
        1,
        1,
        1);
    x = graph_instance_norm_prelu(
        ctx,
        x,
        params,
        "dense_encoder.dense_conv_1.1.weight",
        "dense_encoder.dense_conv_1.1.bias",
        "dense_encoder.dense_conv_1.2.weight");
    x = graph_dense_block(ctx, x, params, "dense_encoder.dense_block");
    x = graph_conv2d(
        ctx,
        x,
        params,
        "dense_encoder.dense_conv_2.0.weight",
        "dense_encoder.dense_conv_2.0.bias",
        0,
        1,
        1,
        2,
        1,
        1);
    return graph_instance_norm_prelu(
        ctx,
        x,
        params,
        "dense_encoder.dense_conv_2.1.weight",
        "dense_encoder.dense_conv_2.1.bias",
        "dense_encoder.dense_conv_2.2.weight");
}

std::vector<float> compact_rel_pos(int64_t seq_len) {
    constexpr float kPi = 3.14159265358979323846f;