Searched refs:wrap_class (Results 1 – 1 of 1) sorted by relevance
1300 def regroup(device_map, values, wrap_class=PerReplica): argument1311 return [regroup(device_map, tuple(v[i] for v in values), wrap_class)1319 regroup(device_map, tuple(v[i] for v in values), wrap_class)1335 return {key: regroup(device_map, tuple(v[key] for v in values), wrap_class)1382 return wrap_class(device_map, values)