Home
last modified time | relevance | path

Searched defs:RunNcclCollective (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_all_gather_thunk.cc70 Status NcclAllGatherThunk::RunNcclCollective(const ExecuteParams& params, in RunNcclCollective() function in xla::gpu::NcclAllGatherThunk
Dnccl_all_to_all_thunk.cc62 Status NcclAllToAllThunk::RunNcclCollective(const ExecuteParams& params, in RunNcclCollective() function in xla::gpu::NcclAllToAllThunk
Dnccl_all_reduce_thunk.cc119 Status NcclAllReduceThunk::RunNcclCollective(const ExecuteParams& params, in RunNcclCollective() function in xla::gpu::NcclAllReduceThunk