[Multiple GPU] Which chunk of data will be sent to which device

Hi,
Is it possible to know which chunk of data will be sent to which device?
I have tried get_device() to know the gpu ID, but I am not sure it is which chunk of data: first split, or second split.