Searched defs:regroup (Results 1 – 2 of 2) sorted by relevance
52 def _make_per_replica(values, devices, regroup=False): argument
1300 def regroup(device_map, values, wrap_class=PerReplica): function