Searched defs:communication (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_test.py | 124 def testReduce(self, collective_ops, device, communication): argument
171 def testGather(self, collective_ops, device, communication): argument
218 def testBroadcast(self, collective_ops, device, communication): argument
255 communication): argument
299 def testCollectiveGroupSizeOne(self, collective_ops, device, communication): argument
325 def testCollectiveInvalidKey(self, collective_ops, device, communication): argument
354 def testMultipleGroups(self, collective_ops, device, communication): argument
395 communication): argument
449 communication): argument
508 def testAbortCommunication(self, collective_op, device, communication): argument
[all …]
|
| D | collective_ops_multi_worker_test.py | 172 def testAbortCommunication(self, device, communication): argument
233 def testAbortGroupParamsResolution(self, device, communication): argument
273 def testAbortInstanceParamsResolution(self, device, communication): argument
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 461 communication=CollectiveCommunication.AUTO, argument
532 communication, argument
671 communication, argument
763 def testReductionLocal(self, required_gpus, communication, argument
783 variable_length, communication): argument
800 def testEagerMultiThread(self, communication): argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | collective_all_reduce_strategy.py | 239 communication=collective_util.CommunicationImplementation.AUTO, argument
261 communication=collective_util.CommunicationImplementation.AUTO): argument
280 communication=collective_util.CommunicationImplementation.AUTO, argument
|