HomeSort by relevance Sort by last modified time
    Searched full:_apply_all_reduce (Results 1 - 1 of 1) sorted by null

  /external/tensorflow/tensorflow/contrib/nccl/python/ops/
nccl_ops.py 47 return _apply_all_reduce('sum', tensors)
93 return _apply_all_reduce('prod', tensors)
110 return _apply_all_reduce('min', tensors)
127 return _apply_all_reduce('max', tensors)
211 def _apply_all_reduce(reduction, tensors): function

Completed in 68 milliseconds