| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | permuter.cc | 69 DispatchSend(col_params_->default_rank, in Run()
70 col_params_->instance.permutation[col_params_->default_rank], in Run()
73 if (col_params_->default_rank == col_params_->instance.permutation[i]) { in Run()
|
| D | hierarchical_tree_broadcaster_test.cc | 347 int default_rank = wi * num_devices_per_worker + di; in Init() local
349 default_rank, col_params_->group.device_names[default_rank], in Init()
582 col_params_->default_rank = rank; in DeviceInstance()
737 cp->default_rank = 0; in TEST_F()
742 cp->default_rank = 2; in TEST_F()
747 cp->default_rank = 0; in TEST_F()
758 cp->default_rank = 0; in TEST_F()
769 cp->default_rank = 0; in TEST_F()
780 cp->default_rank = 9; in TEST_F()
812 cp->default_rank = 0; in TEST_F()
[all …]
|
| D | permuter_test.cc | 195 int default_rank = wi * num_devices_per_worker + di; in Init() local
196 permutation_.push_back(default_rank); in Init()
220 int default_rank = wi * num_devices_per_worker + di; in Init() local
222 default_rank, col_params_->group.device_names[default_rank], in Init()
345 col_params_->default_rank = rank; in DeviceInstance()
|
| D | ring_alg.cc | 167 col_params->group.device_names[col_params->default_rank]; in InitializeCollectiveParams()
231 DCHECK_EQ(permuted_di, col_params->default_rank); in InitializeCollectiveParams()
386 VLOG(3) << "DispatchSend rank=" << col_params_->default_rank << " send key " in DispatchSend()
406 VLOG(3) << "DispatchRecv rank=" << col_params_->default_rank << " recv key " in DispatchRecv()
|
| D | collective_param_resolver_local.cc | 548 cp->default_rank = i; in SetDefaultRank()
557 ir->shared->default_rank = -1; in InitInstanceSharedParams()
795 CHECK_GE(cp->default_rank, 0); in WaitForGroup()
796 if (!ir->known[cp->default_rank]) { in WaitForGroup()
797 ir->known[cp->default_rank] = true; in WaitForGroup()
805 cp->default_rank); in WaitForGroup()
807 ir->source_rank = cp->default_rank; in WaitForGroup()
|
| D | ring_reducer_test.cc | 437 col_params_->default_rank = my_device_id; in DeviceInstance()
603 cp->default_rank = 0; in TEST_F()
620 cp->default_rank = 3; in TEST_F()
638 cp->default_rank = 0; in TEST_F()
669 cp->default_rank = 0; in TEST_F()
|
| D | ring_gatherer_test.cc | 411 col_params_->default_rank = my_device_id; in DeviceInstance()
571 cp->default_rank = 0; in TEST_F()
582 cp->default_rank = 3; in TEST_F()
|
| D | ring_gatherer.cc | 88 << " default_rank " << col_params_->default_rank << "\n" in Run()
|
| D | collective_param_resolver_local_test.cc | 205 EXPECT_EQ(cps[i]->default_rank, i); in TEST_F()
259 EXPECT_EQ(cps[i]->default_rank, i); in TEST_F()
|
| D | ring_reducer.cc | 84 << " default_rank " << col_params_->default_rank << "\n" in Run()
|
| D | hierarchical_tree_broadcaster.cc | 83 col_params->group.device_names[col_params->default_rank]; in InitializeCollectiveParams()
|
| D | base_collective_executor.cc | 460 col_params.group.task_names[col_params.default_rank]; in UnblockDependencies()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | collective_communicator.cc | 75 col_params->group.task_names[col_params->default_rank]); in Enqueue()
82 col_ctx->output, col_ctx->col_params->default_rank, in Enqueue()
112 << col_params->group.task_names[col_params->default_rank] in Enqueue()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 154 strings::StrAppend(&v, " default_rank=", default_rank, in ToString()
185 device_name(col_params->group.device_names[col_params->default_rank]) {} in CollectiveContext()
|
| D | collective.h | 141 int default_rank = -1; // index of this op within device_names member
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 260 col_params_->default_rank = rank; in DeviceInstance()
351 << col_params_->default_rank; in RunBroadcast()
384 << col_params_->default_rank; in RunGather()
505 di->col_params_->is_source = di->col_params_->default_rank == source_rank_; in InitDevice()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed_test.cc | 262 EXPECT_EQ(cp_[device_name]->default_rank, idx); in ValidateCollectiveParams()
|