Searched refs:all_prod (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ | ||
| D | nccl_ops.py | 79 def all_prod(tensors): function |
| D | nccl_ops_test.py | 134 self._Test(partial(_NcclAllReduce, nccl_ops.all_prod), lambda x, y: x * y) |