Searched refs:NcclGpuKernel (Results 1 – 6 of 6) sorted by relevance
46 class NcclGpuKernel : public GpuKernel {48 NcclGpuKernel() : group_name_(""), nccl_data_type_(ncclHalf) {} in NcclGpuKernel() function49 ~NcclGpuKernel() override = default;
28 class NcclSendGpuKernel : public NcclGpuKernel {
28 class NcclRecvGpuKernel : public NcclGpuKernel {
33 class SyncBatchNormGradGpuKernel : public NcclGpuKernel {
44 class NcclCollectiveGpuKernel : public NcclGpuKernel {
33 class SyncBatchNormGpuKernel : public NcclGpuKernel {