pub fn tensor_map_address<T: CubePrimitive, K: TensorMapKind>( tensor_map: &TensorMap<T, K>, ) -> u64