Function layout_to_strides

pub fn layout_to_strides(
    layout: &ABLayout,
    m: usize,
    n: usize,
    k: usize,
) -> (usize, usize, usize, usize, usize, usize)