Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/mpi_collectives/
Dmpi_allgather_test.py28 average_allgather = False variable
55 if average_allgather:
97 all_gather = mpi.allreduce(to_gather[my_rank], average_allgather)