Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.cc716 bool init_nccl, const IRConsumer& f) { in WaitForGroup() argument
737 if (init_nccl && cp->default_rank == 0) { in WaitForGroup()
766 if (init_nccl && ir->communicator_key.empty()) { in WaitForGroup()
Dcollective_param_resolver_local.h207 bool init_source, bool init_nccl, const IRConsumer& f)