Add an optional sleep op to the beginning of the TPU infeed thread.
This allows the compilation thread to finish compiling a large model and prevents a JF request timeout, currently set to a 60 second default. PiperOrigin-RevId: 179070643
Loading
Please sign in to comment