Searched refs:TakeManyOp (Results 1 – 1 of 1) sorted by relevance
567 class TakeManyOp : public BarrierOpKernel { class569 explicit TakeManyOp(OpKernelConstruction* context) in TakeManyOp() function in tensorflow::barrier::TakeManyOp626 TF_DISALLOW_COPY_AND_ASSIGN(TakeManyOp);629 REGISTER_KERNEL_BUILDER(Name("BarrierTakeMany").Device(DEVICE_CPU), TakeManyOp);