Home
last modified time | relevance | path

Searched refs:subdiv_idx (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dring_alg.cc309 void RingAlg::InitRingField(RingField* rf, int chunk_idx, int subdiv_idx, in InitRingField() argument
318 DCHECK_EQ(field_idx, (chunk_idx * num_subdivs_) + subdiv_idx); in InitRingField()
320 rf->subdiv_idx = subdiv_idx; in InitRingField()
322 rf->rank = col_params_->subdiv_rank[subdiv_idx]; in InitRingField()
329 .subdiv_permutations[subdiv_idx][recv_from_rank]; in InitRingField()
331 .subdiv_permutations[subdiv_idx][send_to_rank]; in InitRingField()
371 " subdiv=", subdiv_idx, " sc_idx=", sc_idx, in DebugString()
389 .subdiv_permutations[rf->subdiv_idx][send_to_rank]; in DispatchSend()
415 col_ctx_->device_locality, rf->subdiv_idx, done); in DispatchRecv()
Dring_reducer.cc145 void RingReducer::InitRingField(RingField* rf, int chunk_idx, int subdiv_idx, in InitRingField() argument
147 RingAlg::InitRingField(rf, chunk_idx, subdiv_idx, field_idx); in InitRingField()
165 for (int subdiv_idx = 0; subdiv_idx < num_subdivs_; ++subdiv_idx) { in RunAsyncParts() local
166 int rf_index = (chunk_idx * num_subdivs_) + subdiv_idx; in RunAsyncParts()
167 InitRingField(&rfv_[rf_index], chunk_idx, subdiv_idx, rf_index); in RunAsyncParts()
Dring_gatherer.cc129 for (int subdiv_idx = 0; subdiv_idx < num_subdivs_; ++subdiv_idx) { in RunAsyncParts() local
130 int rf_index = (chunk_idx * num_subdivs_) + subdiv_idx; in RunAsyncParts()
131 InitRingField(&rfv_[rf_index], chunk_idx, subdiv_idx, rf_index); in RunAsyncParts()
Dring_alg.h69 int16 subdiv_idx; // minor division index member
85 virtual void InitRingField(RingField* rf, int chunk_idx, int subdiv_idx,
Dring_reducer.h45 void InitRingField(RingField* rf, int chunk_idx, int subdiv_idx,