Searched defs:communication (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_test.py | 128 def testReduce(self, collective_ops, device, communication): argument
183 def testGather(self, collective_ops, device, communication): argument
265 def testBroadcast(self, collective_ops, device, communication): argument
302 communication): argument
371 def testCollectiveGroupSizeOne(self, collective_ops, device, communication): argument
404 def testCollectiveInvalidKey(self, collective_ops, device, communication): argument
440 def testMultipleGroups(self, collective_ops, device, communication): argument
492 def testReduce(self, collective_ops, device, communication, argument
732 communication): argument
794 communication): argument
[all …]
|
| D | collective_ops_multi_worker_test.py | 168 def testAbortCommunication(self, device, communication): argument
229 def testAbortGroupParamsResolution(self, device, communication): argument
269 def testAbortInstanceParamsResolution(self, device, communication): argument
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 458 communication=CollectiveCommunication.AUTO, argument
531 communication, argument
670 communication, argument
767 def testReductionLocal(self, required_gpus, communication, argument
787 variable_length, communication): argument
804 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
|