Searched refs:NcclStubKernel (Results 1 – 1 of 1) sorted by relevance
262 class NcclStubKernel : public AsyncOpKernel { class264 explicit NcclStubKernel(OpKernelConstruction* c) : AsyncOpKernel(c) {} in NcclStubKernel() function in tensorflow::__anonc894cd950111::NcclStubKernel272 NcclStubKernel);273 REGISTER_KERNEL_BUILDER(Name("NcclReduce").Device(DEVICE_GPU), NcclStubKernel);