Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_all_reduce_thunk.cc247 Status NcclAllReduceThunk::RunNcclCollective(const ExecuteParams& params, in RunNcclCollective() function in xla::gpu::NcclAllReduceThunk
284 Status NcclAllReduceStartThunk::RunNcclCollective(const ExecuteParams& params, in RunNcclCollective() function in xla::gpu::NcclAllReduceStartThunk
359 Status NcclReduceScatterThunk::RunNcclCollective(const ExecuteParams& params, in RunNcclCollective() function in xla::gpu::NcclReduceScatterThunk
Dnccl_all_gather_thunk.cc105 Status NcclAllGatherThunk::RunNcclCollective(const ExecuteParams& params, in RunNcclCollective() function in xla::gpu::NcclAllGatherThunk
Dnccl_all_to_all_thunk.cc70 Status NcclAllToAllThunk::RunNcclCollective(const ExecuteParams& params, in RunNcclCollective() function in xla::gpu::NcclAllToAllThunk
Dnccl_collective_permute_thunk.cc113 Status NcclCollectivePermuteThunk::RunNcclCollective( in RunNcclCollective() function in xla::gpu::NcclCollectivePermuteThunk