Searched refs:my_rank (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| D | mpi_allreduce_test.py | 32 def dumpFailure(self, my_rank, out_loc_red, my_correct, out_all_red, argument
48 .format(my_rank, i, j, out_loc_red[i][j],
55 my_rank = int(os.environ['PMI_RANK'])
105 gpu_options = tf.GPUOptions(visible_device_list=str(my_rank))
115 input_feed = input_feed + my_rank * spread_var
128 print("rank {}: My output is {}".format(my_rank, my_output))
131 print("rank {}: Our output is {}".format(my_rank, our_output))
137 print("{}: iter {}".format(my_rank, i), flush=True)
146 self.dumpFailure(my_rank, out_loc_red, my_correct, out_all_red,
|
| D | mpi_allgather_test.py | 62 my_rank = int(os.environ['PMI_RANK'])
97 all_gather = mpi.allreduce(to_gather[my_rank], average_allgather)
100 gpu_options = tf.GPUOptions(visible_device_list=str(my_rank))
|