Searched refs:NcclReduceOpBase (Results 1 – 1 of 1) sorted by relevance
65 class NcclReduceOpBase : public NcclAsyncOpBase { class67 explicit NcclReduceOpBase(OpKernelConstruction* c) : NcclAsyncOpBase(c) { in NcclReduceOpBase() function in tensorflow::__anonc894cd950111::NcclReduceOpBase92 class NcclAllReduceOpKernel : public NcclReduceOpBase {95 : NcclReduceOpBase(c) {} in NcclAllReduceOpKernel()128 class NcclReduceSendKernel : public NcclReduceOpBase {131 : NcclReduceOpBase(c) {} in NcclReduceSendKernel()159 class NcclReduceRecvKernel : public NcclReduceOpBase {162 : NcclReduceOpBase(c) {} in NcclReduceRecvKernel()