Add broadcast mode to for evaluation input. This new broadcast mode will read...
Add broadcast mode to for evaluation input. This new broadcast mode will read data from host 0 and evenly distribute them to all the replicas. This enables evaluation to run on TPU slice. PiperOrigin-RevId: 236968583
Loading
Please sign in to comment