Home
last modified time | relevance | path

Searched refs:subdiv_stride (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dring_alg.cc140 int subdiv_stride = kAvgDevPerTask / num_subdivs; in GenerateSubdivsInCollectiveParams() local
141 if (subdiv_stride == 0) subdiv_stride = 1; in GenerateSubdivsInCollectiveParams()
144 int subdiv_offset = subdiv_stride * sdi; in GenerateSubdivsInCollectiveParams()
Dring_gatherer_test.cc188 int subdiv_stride = num_devices / num_subdivs; in Init() local
191 subdiv_stride); in Init()
192 col_params_.subdiv_rank[sdi] = sdi * subdiv_stride; in Init()
Dring_reducer_test.cc210 int subdiv_stride = num_devices / num_subdivs; in Init() local
213 subdiv_stride); in Init()
214 col_params_.subdiv_rank[sdi] = sdi * subdiv_stride; in Init()