Making the device name comparison more robust for RemoteCall and
FunctionBufferingResource. Earlier target_device could be '/cpu:0' and source_device be '/job:localhost/replica:0/task:0/device:CPU:0' and we'd deem them to be different devices which is incorrect. PiperOrigin-RevId: 202032979
Loading
Please sign in to comment