Home
last modified time | relevance | path

Searched refs:cross_device_ops_lib (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/v1/
Dcross_device_ops_test.py33 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib unknown
284 cross_device_ops_lib.ReductionToOneDevice()),
287 cross_device_ops_lib.ReductionToOneDevice(
291 cross_device_ops_lib.ReductionToOneDevice(
304 cross_device_ops_lib.AllReduceCrossDeviceOps("nccl", 1)),
307 cross_device_ops_lib.AllReduceCrossDeviceOps("nccl", 0)),
309 cross_device_ops_lib.NcclAllReduce()),
312 cross_device_ops_lib.HierarchicalCopyAllReduce(8)),
323 cross_device_ops_lib.AllReduceCrossDeviceOps
331 cross_device_ops_lib.select_cross_device_ops(["/cpu:0"]),
[all …]
/external/tensorflow/tensorflow/python/distribute/
Dmirrored_strategy.py24 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib unknown
356 isinstance(self._cross_device_ops, cross_device_ops_lib.NcclAllReduce)
358 cross_device_ops_lib.NcclAllReduce)):
376 self._cross_device_ops = cross_device_ops_lib.CollectiveAllReduce(
389 cross_device_ops_lib.select_cross_device_ops(devices))
428 cross_device_ops_lib.ReductionToOneDevice) or
437 self._inferred_cross_device_ops = cross_device_ops_lib.NcclAllReduce()
705 return cross_device_ops_lib.ReductionToOneDevice()
729 return cross_device_ops_lib.reduce_non_distributed_value(
732 …(not cross_device_ops_lib._devices_match(value, destinations) or # pylint: disable=protected-acce…
[all …]
Dcollective_all_reduce_strategy.py29 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib unknown
332 cross_device_ops_lib.CollectiveAllReduce)
374 self._cross_device_ops = cross_device_ops_lib.CollectiveAllReduce(
379 self._host_cross_device_ops = cross_device_ops_lib.CollectiveAllReduce(
489 self._cross_device_ops = cross_device_ops_lib.CollectiveAllReduce(
494 self._host_cross_device_ops = cross_device_ops_lib.CollectiveAllReduce(
683 cross_device_ops_lib.CollectiveAllReduce)
767 return cross_device_ops_lib.reduce_non_distributed_value(
Dparameter_server_strategy.py24 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib unknown
178 cross_device_ops_lib.ReductionToOneDevice(reduce_to_device=_LOCAL_CPU))
417 if not cross_device_ops_lib.check_destinations(destinations):
497 for d in cross_device_ops_lib.get_devices_from(destinations):
519 return cross_device_ops_lib.reduce_non_distributed_value(
Dtpu_strategy.py34 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib unknown
1200 devices = cross_device_ops_lib.get_devices_from(destinations)
1211 output = cross_device_ops_lib.simple_broadcast(output, destinations)
1232 return cross_device_ops_lib.reduce_non_distributed_value(
Dmirrored_strategy_test.py31 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib unknown
1152 cross_device_ops=cross_device_ops_lib.ReductionToOneDevice(
1292 return cross_device_ops_lib.ReductionToOneDevice()
1319 cross_device_ops_lib.ReductionToOneDevice)
Dcross_device_ops_test.py33 from tensorflow.python.distribute import cross_device_ops as cross_device_ops_lib unknown
181 collective = cross_device_ops_lib.CollectiveAllReduce(