Searched defs:cross_device_ops (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops_test.py | 140 def _testReductionAndBroadcast(self, cross_device_ops, devices): argument
298 def testReductionAndBroadcast(self, cross_device_ops, devices): argument
427 def testReductionAndBroadcast(self, cross_device_ops, devices): argument
|
| D | mirrored_strategy.py | 426 def __init__(self, devices=None, cross_device_ops=None): argument
439 def __init__(self, devices=None, cross_device_ops=None): argument
451 def __init__(self, container_strategy, devices=None, cross_device_ops=None): argument
|