xla

575e9237 - Keep the memory used for sending tensors to device under control, to avoid tripping over the ProtocolBuffer 2GB time bomb.