Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_utils.h43 bool IsNcclLaunchModeParallel();
Dnccl_collective_permute_thunk.cc184 if (!IsNcclLaunchModeParallel()) { in RunCollectivePermute()
Dnccl_utils.cc42 bool IsNcclLaunchModeParallel() { in IsNcclLaunchModeParallel() function