Searched refs:update_ops_in_cross_replica_mode (Results 1 – 1 of 1) sorted by relevance
240 update_ops_in_cross_replica_mode=[True])) +245 update_ops_in_cross_replica_mode=[False])))247 renorm, update_ops_in_cross_replica_mode): argument256 update_ops_in_replica_mode=not update_ops_in_cross_replica_mode)263 if update_ops_in_cross_replica_mode: