Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc523 for (int participant_idx = 0; participant_idx < participants_.size(); in DoAllReduce() local
524 participant_idx++) { in DoAllReduce()
527 input_buffers[participant_idx][buffer_idx][idx]); in DoAllReduce()
529 for (int participant_idx = 0; participant_idx < participants_.size(); in DoAllReduce() local
530 participant_idx++) { in DoAllReduce()
531 output_buffers[participant_idx][buffer_idx][idx] = out; in DoAllReduce()