Home
last modified time | relevance | path

Searched refs:AllReduceSpecTuple (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dcross_device_ops.py759 AllReduceSpecTuple = collections.namedtuple("AllReduceSpecTuple", variable
862 return AllReduceSpecTuple(spec[0], 1, -1)
864 return AllReduceSpecTuple(spec[0], spec[1], -1)
866 return AllReduceSpecTuple(*spec)
870 self._all_reduce_spec.append(AllReduceSpecTuple(all_reduce_spec, 1, -1))