fn find_width_height_dim_indices(shape: &[TensorDimension]) -> (usize, usize)