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
|