Searched refs:cross_tower_ops (Results 1 – 2 of 2) sorted by relevance
93 cross_tower_ops=None): argument94 assert not (cross_device_ops and cross_tower_ops)101 cross_device_ops or cross_tower_ops,
173 * You can specify various parameters for all reduce with the `cross_tower_ops`