Function flow_mlp

src/models/heartmula/codec.cpp:834–845

    normed = modules::AddModule{}.build(
        ctx,
        modules::MulModule{}.build(ctx, normed, add_one(ctx, parts.scale_mlp)),
        parts.shift_mlp);
    auto mlp = flow_mlp(ctx, normed, weights, hidden_size);
    return modules::AddModule{}.build(ctx, x, modules::MulModule{}.build(ctx, parts.gate_mlp, mlp));
}

core::TensorValue final_layer_norm_modulation(
    core::ModuleBuildContext & ctx,
    const core::TensorValue & input,
    const core::TensorValue & embedded_timestep,
    const core::TensorValue & scale_shift_table,
    int64_t hidden_size) {
    const int64_t batch = input.shape.dims[0];
    const int64_t frames = input.shape.dims[1];
    auto table = core::reshape_tensor(ctx, scale_shift_table, core::TensorShape::from_dims({1, 2, hidden_size}));