Searched refs:CommunicationImplementation (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops_test.py | 56 CommunicationImplementation = collective_util.CommunicationImplementation variable
316 CommunicationImplementation.AUTO,
317 CommunicationImplementation.RING,
318 CommunicationImplementation.NCCL,
325 implementation == CommunicationImplementation.NCCL):
328 implementation == CommunicationImplementation.NCCL):
357 CommunicationImplementation.AUTO,
358 CommunicationImplementation.RING,
359 CommunicationImplementation.NCCL,
367 implementation == CommunicationImplementation.NCCL):
[all …]
|
| D | collective_util.py | 32 class CommunicationImplementation(enum.Enum): class
51 CollectiveCommunication = CommunicationImplementation
93 implementation=CommunicationImplementation.AUTO):
127 implementation=CommunicationImplementation.AUTO):
131 implementation = CommunicationImplementation(implementation.upper())
132 if not isinstance(implementation, CommunicationImplementation):
160 if options.implementation != CommunicationImplementation.AUTO:
|
| D | cross_device_ops.py | 1016 CollectiveCommunication = collective_util.CommunicationImplementation
1017 CommunicationImplementation = collective_util.CommunicationImplementation variable
1096 options.implementation == CommunicationImplementation.NCCL and
1098 implementation = CommunicationImplementation.AUTO.value
1209 options.implementation == CommunicationImplementation.NCCL and
1211 implementation = CommunicationImplementation.AUTO.value
1279 options.implementation == CommunicationImplementation.NCCL and
1281 implementation = CommunicationImplementation.AUTO.value
1341 if (options.implementation == CommunicationImplementation.NCCL and
1343 implementation = CommunicationImplementation.AUTO.value
|
| D | collective_all_reduce_strategy.py | 239 communication=collective_util.CommunicationImplementation.AUTO,
261 communication=collective_util.CommunicationImplementation.AUTO):
280 communication=collective_util.CommunicationImplementation.AUTO,
704 collective_util.CommunicationImplementation.NCCL):
862 implementation=collective_util.CommunicationImplementation.RING))
889 collective_util.CommunicationImplementation.NCCL) and
|
| D | mirrored_strategy.py | 332 implementation=collective_util.CommunicationImplementation.NCCL)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.experimental.-communication-implementation.pbtxt | 1 path: "tensorflow.distribute.experimental.CommunicationImplementation"
3 is_instance: "<enum \'CommunicationImplementation\'>"
6 mtype: "<enum \'CommunicationImplementation\'>"
10 mtype: "<enum \'CommunicationImplementation\'>"
14 mtype: "<enum \'CommunicationImplementation\'>"
|
| D | tensorflow.distribute.experimental.-collective-communication.pbtxt | 3 is_instance: "<enum \'CommunicationImplementation\'>"
6 mtype: "<enum \'CommunicationImplementation\'>"
10 mtype: "<enum \'CommunicationImplementation\'>"
14 mtype: "<enum \'CommunicationImplementation\'>"
|
| D | tensorflow.distribute.experimental.-communication-options.pbtxt | 7 …\'], varargs=None, keywords=None, defaults=[\'0\', \'None\', \'CommunicationImplementation.AUTO\']…
|
| D | tensorflow.distribute.experimental.pbtxt | 16 name: "CommunicationImplementation"
|
| D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 22 …'cluster_resolver\'], varargs=None, keywords=None, defaults=[\'CommunicationImplementation.AUTO\',…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.experimental.-communication-implementation.pbtxt | 1 path: "tensorflow.distribute.experimental.CommunicationImplementation"
3 is_instance: "<enum \'CommunicationImplementation\'>"
6 mtype: "<enum \'CommunicationImplementation\'>"
10 mtype: "<enum \'CommunicationImplementation\'>"
14 mtype: "<enum \'CommunicationImplementation\'>"
|
| D | tensorflow.distribute.experimental.-collective-communication.pbtxt | 3 is_instance: "<enum \'CommunicationImplementation\'>"
6 mtype: "<enum \'CommunicationImplementation\'>"
10 mtype: "<enum \'CommunicationImplementation\'>"
14 mtype: "<enum \'CommunicationImplementation\'>"
|
| D | tensorflow.distribute.experimental.-communication-options.pbtxt | 7 …\'], varargs=None, keywords=None, defaults=[\'0\', \'None\', \'CommunicationImplementation.AUTO\']…
|
| D | tensorflow.distribute.experimental.pbtxt | 16 name: "CommunicationImplementation"
|
| D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 21 …'cluster_resolver\'], varargs=None, keywords=None, defaults=[\'CommunicationImplementation.AUTO\',…
|
| /external/tensorflow/ |
| D | RELEASE.md | 451 ….experimental.CollectiveCommunication` to `tf.distribute.experimental.CommunicationImplementation`.
|