Searched refs:LoopCondOp (Results 1 – 7 of 7) sorted by relevance
119 class LoopCondOp : public OpKernel {121 explicit LoopCondOp(OpKernelConstruction* context);122 ~LoopCondOp() override;128 TF_DISALLOW_COPY_AND_ASSIGN(LoopCondOp);
457 LoopCondOp::LoopCondOp(OpKernelConstruction* context) : OpKernel(context) {} in LoopCondOp() function in tensorflow::LoopCondOp458 LoopCondOp::~LoopCondOp() = default;460 void LoopCondOp::Compute(OpKernelContext* context) { in Compute()471 bool LoopCondOp::IsExpensive() { return false; } in IsExpensive()473 REGISTER_KERNEL_BUILDER(Name("LoopCond").Device(DEVICE_CPU), LoopCondOp);478 LoopCondOp);483 LoopCondOp);
124 llvm::isa_and_nonnull<tf_executor::LoopCondOp>(operand.getDefiningOp())) in FindDeviceFromOperands()
326 rewriter.replaceOpWithNewOp<tf_executor::LoopCondOp>( in matchAndRewrite()
212 tf_executor::LoopCondOp, tf_executor::MergeOp, in IsSupportedNonTFOp()
849 void LoopCondOp::print(OpAsmPrinter &p) { in print()864 ParseResult LoopCondOp::parse(OpAsmParser &parser, OperationState &result) { in parse()
1899 return builder_.create<mlir::tf_executor::LoopCondOp>(loc, types, operands, in CreateOperation()