| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | hierarchical_tree_broadcaster.cc | 107 << device_name << " source_rank=" << col_params->source_rank in InitializeCollectiveParams()
131 int source_task = GetDeviceTask(col_params->source_rank, dev_per_task); in InitializeCollectiveParams()
136 perm.push_back(col_params->source_rank); in InitializeCollectiveParams()
138 col_params->instance.device_names[col_params->source_rank] == in InitializeCollectiveParams()
171 if (abs_di == col_params->source_rank) subdiv_source = di; in InitializeCollectiveParams()
225 int source_rank = impl.subdiv_source_rank[subdiv]; in TreeRecvFrom() local
226 if (my_rank == source_rank) return -1; in TreeRecvFrom()
227 if (source_rank == 0) { in TreeRecvFrom()
231 return (predecessor_rank < 0) ? source_rank : predecessor_rank; in TreeRecvFrom()
245 int source_rank = impl.subdiv_source_rank[subdiv]; in TreeSendTo() local
[all …]
|
| D | hierarchical_tree_broadcaster_test.cc | 732 cp.source_rank = 0; in TEST_F()
737 cp.source_rank = 2; in TEST_F()
742 cp.source_rank = 2; in TEST_F()
752 cp.source_rank = 0; in TEST_F()
763 cp.source_rank = 2; in TEST_F()
774 cp.source_rank = 9; in TEST_F()
805 cp.source_rank = 0; in TEST_F()
816 cp.source_rank = 2; in TEST_F()
827 cp.source_rank = 9; in TEST_F()
|
| D | collective_param_resolver_local.cc | 701 cp->source_rank = irec->source_rank; in CompleteInstanceFromInitializedIRec()
728 if (ir->source_rank >= 0) { in WaitForGroup()
730 " already has source ", ir->source_rank, in WaitForGroup()
734 ir->source_rank = cp->default_rank; in WaitForGroup()
756 if (init_source && ir->source_rank < 0) { in WaitForGroup()
|
| D | collective_param_resolver_local.h | 137 int source_rank GUARDED_BY(out_mu);
146 source_rank(-1), in InstanceRec()
|
| D | ring_alg.cc | 63 int section, int source_rank) { in RingAlgBufKey() argument
66 section, "):srcrank(", source_rank, ")"); in RingAlgBufKey()
70 return strings::StrCat(exec_key, ":", pass, ":", section, ":", source_rank); in RingAlgBufKey()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed.cc | 189 response->set_source_rank(ir->source_rank); in CompleteInstanceAsync()
295 int32 source_rank = resp.source_rank(); in UpdateInstanceCache() local
298 auto continue_with_ir = [cp, irp, source_rank, communicator_key, in UpdateInstanceCache()
310 if (ir->source_rank != source_rank) { in UpdateInstanceCache()
311 if (ir->source_rank >= 0) { in UpdateInstanceCache()
314 cp->instance.instance_key, " gives source_rank=", source_rank, in UpdateInstanceCache()
315 " but cache already holds value=", ir->source_rank); in UpdateInstanceCache()
319 ir->source_rank = source_rank; in UpdateInstanceCache()
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/ |
| D | mpi_ops.cc | 688 int source_rank = status.MPI_SOURCE; in BackgroundThreadLoop() local
695 MPI_Recv(NULL, 0, MPI_BYTE, source_rank, TAG_NOTIFY, MPI_COMM_WORLD, in BackgroundThreadLoop()
702 MPI_Recv(buffer, msg_length, MPI_BYTE, source_rank, TAG_NOTIFY, in BackgroundThreadLoop()
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| D | mpi_ops.cc | 689 int source_rank = status.MPI_SOURCE; in BackgroundThreadLoop() local
696 MPI_Recv(NULL, 0, MPI_BYTE, source_rank, TAG_NOTIFY, MPI_COMM_WORLD, in BackgroundThreadLoop()
703 MPI_Recv(buffer, msg_length, MPI_BYTE, source_rank, TAG_NOTIFY, in BackgroundThreadLoop()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 142 " is_source=", is_source, " source_rank=", source_rank, in ToString()
|
| D | collective.h | 119 int source_rank = -1; // broadcast only member
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 534 // Also gives the source_rank in case of broadcast.
537 int32 source_rank = 2; field
|