EagerTensor.device reflects the op's device and not the tensor's memory space.
This matches graph mode's behavior. PiperOrigin-RevId: 187537818
Loading
Please sign in to comment
This matches graph mode's behavior. PiperOrigin-RevId: 187537818