Searched refs:default_rank (Results 1 – 15 of 15) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | collective_nccl_reducer.cc | 54 if (col_params->default_rank == 0 && col_params->group.num_tasks > 1) { in InitializeInstanceBeforeGroupDiscovery() 137 col_params_->instance.task_names[col_params_->default_rank]); in Run() 147 col_params_->default_rank, std::move(done_callback)); in Run() 150 << col_params_->instance.task_names[col_params_->default_rank] in Run()
|
D | collective_nccl_reducer_test.cc | 223 col_params_.default_rank = rank; in DeviceInstance()
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | hierarchical_tree_broadcaster_test.cc | 333 int default_rank = wi * num_devices_per_worker + di; in Init() local 335 default_rank, col_params_.instance.device_names[default_rank], in Init() 580 col_params_.default_rank = rank; in DeviceInstance() 733 cp.default_rank = 0; in TEST_F() 738 cp.default_rank = 2; in TEST_F() 743 cp.default_rank = 0; in TEST_F() 753 cp.default_rank = 0; in TEST_F() 764 cp.default_rank = 0; in TEST_F() 775 cp.default_rank = 9; in TEST_F() 806 cp.default_rank = 0; in TEST_F() [all …]
|
D | collective_param_resolver_local.cc | 394 cp->default_rank = i; in SetDefaultRank() 414 ir->shared.default_rank = -1; in InitInstanceSharedParams() 722 CHECK_GE(cp->default_rank, 0); in WaitForGroup() 723 if (!ir->known[cp->default_rank]) { in WaitForGroup() 724 ir->known[cp->default_rank] = true; in WaitForGroup() 732 cp->default_rank); in WaitForGroup() 734 ir->source_rank = cp->default_rank; in WaitForGroup() 737 if (init_nccl && cp->default_rank == 0) { in WaitForGroup() 745 cp->instance.device_names[cp->default_rank]); in WaitForGroup() 769 cp->instance.device_names[cp->default_rank], in WaitForGroup()
|
D | ring_alg.cc | 166 col_params->instance.device_names[col_params->default_rank]; in InitializeCollectiveParams() 230 DCHECK_EQ(permuted_di, col_params->default_rank); in InitializeCollectiveParams() 384 VLOG(3) << "DispatchSend rank=" << col_params_->default_rank << " send key " in DispatchSend() 403 VLOG(3) << "DispatchRecv rank=" << col_params_->default_rank << " recv key " in DispatchRecv()
|
D | ring_reducer_test.cc | 426 col_params_.default_rank = my_device_id; in DeviceInstance() 587 cp.default_rank = 0; in TEST_F() 604 cp.default_rank = 3; in TEST_F() 621 cp.default_rank = 0; in TEST_F() 651 cp.default_rank = 0; in TEST_F()
|
D | collective_param_resolver_local_test.cc | 198 EXPECT_EQ(cps[i].default_rank, i); in TEST_F() 250 EXPECT_EQ(cps[i].default_rank, i); in TEST_F()
|
D | ring_gatherer_test.cc | 402 col_params_.default_rank = my_device_id; in DeviceInstance() 560 cp.default_rank = 0; in TEST_F() 571 cp.default_rank = 3; in TEST_F()
|
D | ring_gatherer.cc | 86 << " default_rank " << col_params_->default_rank << "\n" in Run()
|
D | ring_reducer.cc | 78 << " default_rank " << col_params_->default_rank << "\n" in Run()
|
D | base_collective_executor.cc | 337 col_params.instance.task_names[col_params.default_rank]; in Launched()
|
D | hierarchical_tree_broadcaster.cc | 82 col_params->instance.device_names[col_params->default_rank]; in InitializeCollectiveParams()
|
/external/tensorflow/tensorflow/core/framework/ |
D | collective.cc | 141 strings::StrAppend(&v, " default_rank=", default_rank, in ToString() 173 device_name(col_params.instance.device_names[col_params.default_rank]) {} in CollectiveContext()
|
D | collective.h | 117 int default_rank = -1; // index of this op within device_names member
|
/external/tensorflow/tensorflow/core/distributed_runtime/ |
D | collective_param_resolver_distributed_test.cc | 265 EXPECT_EQ(cp_[idx].default_rank, idx); in ValidateCollectiveParams()
|