struct sdpa_dims_t

struct sdpa_dims_t
{
    // fields

    dim mb;
    dim seq_len;
    dim head_num;
    dim head_size;
    dim query_num;
};