Home
last modified time | relevance | path

Searched defs:regroup (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/distribute/python/
Dcross_device_ops_test.py52 def _make_per_replica(values, devices, regroup=False): argument
/external/tensorflow/tensorflow/python/distribute/
Dvalues.py1300 def regroup(device_map, values, wrap_class=PerReplica): function