Home
last modified time | relevance | path

Searched refs:CollectiveAllReduce (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dcollective_all_reduce_strategy.py332 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)
Dcross_device_ops.py1021 class CollectiveAllReduce(CrossDeviceOps): class
1074 super(CollectiveAllReduce, self).__init__()
1377 return CollectiveAllReduce(self._devices, self._group_size, collective_keys)
Dmirrored_strategy.py376 self._cross_device_ops = cross_device_ops_lib.CollectiveAllReduce(
Dcross_device_ops_test.py181 collective = cross_device_ops_lib.CollectiveAllReduce(
/external/tensorflow/tensorflow/python/distribute/v1/
Dcross_device_ops_test.py478 collective_all_reduce_ops = cross_device_ops_lib.CollectiveAllReduce(
510 collective_all_reduce_ops = cross_device_ops_lib.CollectiveAllReduce(