Searched refs:NcclAsyncOpBase (Results 1 – 1 of 1) sorted by relevance
43 class NcclAsyncOpBase : public AsyncOpKernel { class45 explicit NcclAsyncOpBase(OpKernelConstruction* c) : AsyncOpKernel(c) { in NcclAsyncOpBase() function in tensorflow::__anonc894cd950111::NcclAsyncOpBase62 TF_DISALLOW_COPY_AND_ASSIGN(NcclAsyncOpBase);65 class NcclReduceOpBase : public NcclAsyncOpBase {67 explicit NcclReduceOpBase(OpKernelConstruction* c) : NcclAsyncOpBase(c) { in NcclReduceOpBase()198 class NcclBroadcastSendKernel : public NcclAsyncOpBase {201 : NcclAsyncOpBase(c) {} in NcclBroadcastSendKernel()227 class NcclBroadcastRecvKernel : public NcclAsyncOpBase {230 : NcclAsyncOpBase(c) {} in NcclBroadcastRecvKernel()