Searched defs:cross_device_ops (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | mirrored_strategy.py | 285 def __init__(self, devices=None, cross_device_ops=None): argument
301 def __init__(self, devices=None, cross_device_ops=None): argument
317 def __init__(self, container_strategy, devices=None, cross_device_ops=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/v1/ |
| H A D | cross_device_ops_test.py | 159 def _testReductionAndBroadcast(self, cross_device_ops, devices): argument
317 def testReductionAndBroadcast(self, cross_device_ops, devices): argument
|