Function reshape_heads

src/models/ace_step/diffusion.cpp:99–109

    const core::TensorValue & q_heads,
    const core::TensorValue & k_heads,
    const core::TensorValue & v_heads,
    int64_t dim,
    const std::optional<core::TensorValue> & attention_mask,
    core::BackendType backend_type) {
    (void)backend_type;
    return modules::ScaledDotProductAttentionModule({
        dim,
        modules::ScaledDotProductAttentionLowering::Explicit,
        GGML_PREC_F32,
    }).build(ctx, q_heads, k_heads, v_heads, attention_mask);
}

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