Searched refs:BarrierOpKernel (Results 1 – 1 of 1) sorted by relevance
502 class BarrierOpKernel : public AsyncOpKernel { class504 explicit BarrierOpKernel(OpKernelConstruction* context) in BarrierOpKernel() function in tensorflow::barrier::BarrierOpKernel523 class InsertManyOp : public BarrierOpKernel {526 : BarrierOpKernel(context) { in InsertManyOp()567 class TakeManyOp : public BarrierOpKernel {570 : BarrierOpKernel(context) { in TakeManyOp()631 class BarrierCloseOp : public BarrierOpKernel {634 : BarrierOpKernel(context) { in BarrierCloseOp()653 class BarrierIncompleteSizeOp : public BarrierOpKernel {656 : BarrierOpKernel(context) {} in BarrierIncompleteSizeOp()[all …]