Searched refs:all_gather (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| D | mpi_allgather_test.py | 97 all_gather = mpi.allreduce(to_gather[my_rank], average_allgather)
107 all_gathered, local_gathered = sess.run([all_gather, correct_gather])
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops_test.py | 122 colred0 = collective_ops.all_gather(in0, 2, group_key, instance_key)
125 colred1 = collective_ops.all_gather(in1, 2, group_key, instance_key)
|
| D | collective_ops.py | 61 def all_gather(t, group_size, group_key, instance_key): function
|