Function graph_conv2d

src/framework/audio/zipenhancer.cpp:179–209

}

core::TensorValue graph_conv2d(
    core::ModuleBuildContext & ctx,
    const core::TensorValue & input,
    const std::unordered_map<std::string, Param> & params,
    const std::string & weight_name,
    const std::string & bias_name,
    int64_t pad_t,
    int64_t pad_f,
    int64_t stride_t,
    int64_t stride_f,
    int64_t dil_t,
    int64_t dil_f) {
    const auto & weight = require_param(params, weight_name);
    const auto & bias = require_param(params, bias_name);
    if (weight.shape.size() != 4) {
        throw std::runtime_error("ZipEnhancer graph conv2d weight shape mismatch: " + weight_name);
    }
    return modules::Conv2dModule({
        input.shape.dims[1],
        weight.shape[0],
        weight.shape[2],
        weight.shape[3],
        static_cast<int>(stride_t),
        static_cast<int>(stride_f),
        static_cast<int>(pad_t),
        static_cast<int>(pad_f),
        static_cast<int>(dil_t),
        static_cast<int>(dil_f),
        true,
    }).build(ctx, input, graph_conv2d_weights(weight, bias));
}

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