Function gemm_rhs_shape 

pub fn gemm_rhs_shape<T: HasShape>(input_a: &T) -> Result<Shape, SimError>

Construct an input B shape for a Gemm consuming input_a as input A.