Searched defs:collective_keys (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ | ||
| D | cross_device_utils.py | 266 collective_keys, argument |
| D | cross_device_ops.py | 1028 def __init__(self, devices, group_size, collective_keys=None): argument |