Function graph_linear

src/framework/audio/zipenhancer.cpp:379–393

}

core::TensorValue graph_linear(
    core::ModuleBuildContext & ctx,
    const core::TensorValue & input,
    const std::unordered_map<std::string, Param> & params,
    const std::string & prefix,
    int64_t out_features) {
    const std::string weight_name = prefix + ".weight";
    const std::string bias_name = prefix + ".bias";
    const auto & weight = require_param(params, weight_name);
    const auto & bias = require_param(params, bias_name);
    return modules::LinearModule({input.shape.last_dim(), out_features, true}).build(
        ctx,
        input,
        modules::LinearWeights{graph_param(weight), graph_param(bias)});
}

core::TensorValue graph_add(core::ModuleBuildContext & ctx, const core::TensorValue & lhs, const core::TensorValue & rhs) {
    return modules::AddModule().build(ctx, lhs, rhs);