Home
last modified time | relevance | path

Searched refs:RunNcclCollective (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_all_reduce_thunk.h48 Status RunNcclCollective(const ExecuteParams& params,
Dnccl_all_to_all_thunk.h48 Status RunNcclCollective(const ExecuteParams& params,
Dnccl_all_gather_thunk.h50 Status RunNcclCollective(const ExecuteParams& params,
Dnccl_collective_thunk.cc105 TF_RETURN_IF_ERROR(RunNcclCollective(params, comm)); in ExecuteOnStream()
Dnccl_collective_thunk.h126 virtual Status RunNcclCollective(const ExecuteParams& params,
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