Searched defs:cross_device_ops_instance (Results 1 – 1 of 1) sorted by relevance
236 def _testIndexedSlicesAllReduce(self, devices, cross_device_ops_instance, argument399 def testIndexedSlicesAllReduce(self, cross_device_ops_instance, reduce_op, argument419 cross_device_ops_instance, batch_reduce): argument