Fix a replicate_model_fn_test that is dependent on the number of hardware GPUs.
It has been causing failures when run on a machine with 4 GPUs. PiperOrigin-RevId: 177670759
Loading
Please sign in to comment
It has been causing failures when run on a machine with 4 GPUs. PiperOrigin-RevId: 177670759