Searched refs:SwitchNOp (Results 1 – 5 of 5) sorted by relevance
54 class SwitchNOp : public OpKernel {56 explicit SwitchNOp(OpKernelConstruction* context) : OpKernel(context) {} in SwitchNOp() function59 ~SwitchNOp() override {} in ~SwitchNOp()61 TF_DISALLOW_COPY_AND_ASSIGN(SwitchNOp);
42 void SwitchNOp::Compute(OpKernelContext* context) { in Compute()62 SwitchNOp);74 SwitchNOp)93 SwitchNOp)138 SwitchNOp)
463 LogicalResult Verify(SwitchNOp switchn) { in Verify()515 void Print(SwitchNOp switchn, OpAsmPrinter &p) { in Print()
118 tf_executor::NextIterationSinkOp, tf_executor::SwitchNOp, in IsSupportedNonTFOp()
1741 return builder_.create<mlir::tf_executor::SwitchNOp>(loc, types, operands, in CreateOperation()