Changing the prefetch_to_device implementation to copy_to_device.prefetch
instead. This removes the dependency on FunctionBufferingResource which we want to deprecate. PiperOrigin-RevId: 223459089
Loading
Please sign in to comment
instead. This removes the dependency on FunctionBufferingResource which we want to deprecate. PiperOrigin-RevId: 223459089