/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
|