Since the operation is executed on the GPU, the inputs as well as other tensors must be loaded onto the device before the kernel can be launched.


The indices will also be moved to the device, so there shouldn’t be a difference.