Searched defs:cross_device_ops (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | mirrored_strategy.py | 277 def __init__(self, devices=None, cross_device_ops=None): argument
293 def __init__(self, devices=None, cross_device_ops=None): argument
309 def __init__(self, container_strategy, devices=None, cross_device_ops=None): argument
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 162 def _testReductionAndBroadcast(self, cross_device_ops, devices): argument
320 def testReductionAndBroadcast(self, cross_device_ops, devices): argument
|