Add get_shape function for the convert_dtype