Function load_buffer

pub fn load_buffer(
    data: &[u8],
    device: &Device,
) -> Result<HashMap<String, Tensor>>