Searched refs:BarrierOp (Results 1 – 11 of 11) sorted by relevance
150 builder.create<gpu::BarrierOp>(loc); in insertCopies()153 builder.create<gpu::BarrierOp>(loc); in insertCopies()
112 create<gpu::BarrierOp>(); in rewrite()134 create<gpu::BarrierOp>(); in rewrite()
442 class BarrierOp : public ResourceOpKernel<Barrier> { class444 explicit BarrierOp(OpKernelConstruction* context) in BarrierOp() function in tensorflow::barrier::BarrierOp497 TF_DISALLOW_COPY_AND_ASSIGN(BarrierOp);500 REGISTER_KERNEL_BUILDER(Name("Barrier").Device(DEVICE_CPU), BarrierOp);
75 omp::BarrierOp, omp::TaskwaitOp>(); in runOnOperation()
74 converter.getFirOpBuilder().create<mlir::omp::BarrierOp>( in genOMP()
256 def BarrierOp : OpenMP_Op<"barrier"> {
527 .Case([&](omp::BarrierOp) { in convertOmpOperation() argument
439 template <typename BarrierOp>440 static LogicalResult verifyMemorySemantics(BarrierOp op) { in verifyMemorySemantics()
210 struct BarrierOp { struct in __anon6528055b0211::AArch64Operand259 struct BarrierOp Barrier;
366 struct BarrierOp { struct in __anon1914c67d0111::AArch64Operand415 struct BarrierOp Barrier;
400 struct BarrierOp { struct in __anona500a0890111::AArch64Operand449 struct BarrierOp Barrier;