Home
last modified time | relevance | path

Searched defs:local_rank (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/contrib/mpi_collectives/python/ops/
Dmpi_ops.py74 def local_rank(name=None): function
/external/tensorflow/tensorflow/contrib/mpi_collectives/
Dmpi_ops.py103 def local_rank(name=None): function
Dmpi_ops.cc226 int local_rank = 0; member
614 int local_rank; in BackgroundThreadLoop() local
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager_test.cc97 for (int local_rank = 0; local_rank < num_ranks_per_node; ++local_rank) { in MakeReductionTestCase() local
433 for (int local_rank = 0; local_rank < num_ranks_per_node; in TYPED_TEST() local
/external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/
Dmpi_ops.cc225 int local_rank = 0; member
613 int local_rank; in BackgroundThreadLoop() local
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.cc151 int local_rank; member