Provide a hint about the number of iterations to while_loop in the case of for...
Provide a hint about the number of iterations to while_loop in the case of for loops over tensors of known size. This allows using this type of for loops on TPU. PiperOrigin-RevId: 192166460
Loading
Please sign in to comment