Home
last modified time | relevance | path

Searched defs:num_participants (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcollective_decomposer_utils.cc122 size_t num_participants = in CreateStartIndicesForCollectiveDecomposition() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_all_to_all_thunk.cc89 int num_participants; in RunAllToAll() local
Dnccl_collective_permute_thunk.cc54 const int64_t num_participants = in GetNcclCollectivePermuteConfig() local
Dnccl_all_reduce_thunk.cc381 int num_participants = 0; in RunReduceScatter() local
/external/tensorflow/tensorflow/python/distribute/
Ddistribute_coordinator.py63 def __init__(self, num_participants): argument
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc550 int num_participants = participants_.size(); in DoAllReduce() local