Searched refs:_dgx1_links (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 1059 _dgx1_links = [[1, 2, 3, 4], [0, 2, 3, 5], [0, 1, 3, 6], [0, 1, 2, 7], variable
1070 for i, (gpu_link, dgx1_link) in enumerate(zip(gpu_links, _dgx1_links)):
|