Searched refs:default_rank (Results 1 – 18 of 18) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | hierarchical_tree_broadcaster_test.cc | 290 cp->default_rank = 0; in TEST_F()
295 cp->default_rank = 2; in TEST_F()
300 cp->default_rank = 0; in TEST_F()
316 cp->default_rank = 0; in TEST_F()
327 cp->default_rank = 0; in TEST_F()
338 cp->default_rank = 9; in TEST_F()
372 cp->default_rank = 0; in TEST_F()
383 cp->default_rank = 0; in TEST_F()
394 cp->default_rank = 5; in TEST_F()
|
| D | all_to_all.cc | 114 auto default_rank = col_params_->default_rank; in Run() local
116 DispatchSend(default_rank, i, &input_chunks_[i], CheckCounterAndCallDone()); in Run()
118 DispatchRecv(i, default_rank, &output_chunks_[i], in Run()
|
| D | ring_reducer_test.cc | 234 cp->default_rank = 0; in TEST_F()
251 cp->default_rank = 3; in TEST_F()
272 cp->default_rank = 0; in TEST_F()
308 cp->default_rank = 0; in TEST_F()
324 cp->default_rank = 0; in TEST_F()
332 cp->default_rank = 0; in TEST_F()
351 cp->default_rank = 0; in TEST_F()
369 cp->default_rank = 0; in TEST_F()
|
| 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 | collective_param_resolver_local.cc | 533 cp->default_rank = i; in SetDefaultRank()
545 ir->shared->default_rank = -1; in InitInstanceSharedParams()
812 CHECK_GE(cp->default_rank, 0); in WaitForGroup()
813 if (!ir->known[cp->default_rank]) { in WaitForGroup()
814 ir->known[cp->default_rank] = true; in WaitForGroup()
822 cp->default_rank); in WaitForGroup()
824 ir->source_rank = cp->default_rank; in WaitForGroup()
|
| D | ring_alg.cc | 182 col_params->group.members[col_params->default_rank].device.name(); in InitializeCollectiveParams()
245 DCHECK_EQ(permuted_di, col_params->default_rank); in InitializeCollectiveParams()
401 VLOG(3) << "DispatchSend rank=" << col_params_->default_rank << " send key " in DispatchSend()
421 VLOG(3) << "DispatchRecv rank=" << col_params_->default_rank << " recv key " in DispatchRecv()
|
| D | ring_gatherer_test.cc | 194 cp->default_rank = 0; in TEST_F()
207 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 | collective_test_util.cc | 192 col_params->default_rank = rank; in CreateCollectiveParams()
|
| D | hierarchical_tree_broadcaster.cc | 84 col_params->group.members[col_params->default_rank].device.name(); in InitializeCollectiveParams()
|
| D | base_collective_executor.cc | 483 col_params.group.members[col_params.default_rank].task; in UnblockDependencies()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | collective_communicator.cc | 86 col_params->group.members[col_params->default_rank].task); in Enqueue()
94 col_ctx->output, col_ctx->col_params->default_rank, in Enqueue()
124 << col_params->group.members[col_params->default_rank].task in Enqueue()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 140 strings::StrAppend(&v, " default_rank=", default_rank, in ToString()
173 col_params->group.members[col_params->default_rank].device.name()) {} in CollectiveContext()
|
| D | collective.h | 149 int default_rank = -1; // index of this op within device_names member
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 262 col_params_->default_rank = rank; in DeviceInstance()
353 << col_params_->default_rank; in RunBroadcast()
386 << col_params_->default_rank; in RunGather()
508 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 | 254 EXPECT_EQ(cp_[device_name]->default_rank, idx); in ValidateCollectiveParams()
|