Fix functional.While(), functional.For(rewrite_with_while)
When executing on GPU, synchronously copy cond result from device to host. PiperOrigin-RevId: 196580820
Loading
Please sign in to comment
When executing on GPU, synchronously copy cond result from device to host. PiperOrigin-RevId: 196580820