Function zero_masked_query_rows

src/models/ace_step/planner.cpp:229–239

    return core::ensure_backend_addressable_layout(ctx, value);
}

core::TensorValue zero_masked_query_rows(
    core::ModuleBuildContext & ctx,
    const core::TensorValue & input,
    const core::TensorValue & query_mask) {
    core::validate_shape(
        query_mask,
        core::TensorShape::from_dims({input.shape.dims[0], input.shape.dims[1], 1, 1}),
        "query_mask");
    const auto mask = core::wrap_tensor(ggml_repeat(ctx.ggml, query_mask.tensor, input.tensor), input.shape, GGML_TYPE_F32);
    return core::wrap_tensor(ggml_mul(ctx.ggml, input.tensor, mask.tensor), input.shape, GGML_TYPE_F32);
}

ggml_type planner_phase2_activation_type(core::BackendType backend_type) {