Searched refs:local_rank (Results 1 – 8 of 8) sorted by relevance
97 for (int local_rank = 0; local_rank < num_ranks_per_node; ++local_rank) { in MakeReductionTestCase() local98 auto* device = GetDevice(local_rank); in MakeReductionTestCase()433 for (int local_rank = 0; local_rank < num_ranks_per_node; in TYPED_TEST() local434 ++local_rank) { in TYPED_TEST()435 auto* device = this->GetDevice(local_rank); in TYPED_TEST()438 const int global_rank = node * num_ranks_per_node + local_rank; in TYPED_TEST()
126 local_rank = session.run(mpi.local_rank())128 one_gpu = tf.GPUOptions(visible_device_list=str(local_rank))
226 int local_rank = 0; member614 int local_rank; in BackgroundThreadLoop() local615 MPI_Comm_rank(local_comm, &local_rank); in BackgroundThreadLoop()618 mpi_global.local_rank = local_rank; in BackgroundThreadLoop()985 flat(0) = mpi_global.local_rank; in Compute()
103 def local_rank(name=None): function
131 from tensorflow.contrib.mpi_collectives.python.ops.mpi_ops import local_rank
225 int local_rank = 0; member613 int local_rank; in BackgroundThreadLoop() local614 MPI_Comm_rank(local_comm, &local_rank); in BackgroundThreadLoop()617 mpi_global.local_rank = local_rank; in BackgroundThreadLoop()946 flat(0) = mpi_global.local_rank; in Compute()
151 int local_rank; member170 dr->local_rank = 0; // Will be populated later by OrderTaskDeviceMap. in BuildDevRecs()199 dr->local_rank = rank_it->second; in ParseRingOrder()237 dr->local_rank = next_rank; in OrderTaskDeviceMap()311 it.second.global_rank = it.second.local_rank + next_rank; in EstablishGlobalRank()
74 def local_rank(name=None): function