Searched refs:addn_op (Results 1 – 2 of 2) sorted by relevance
32 AddNOperator* addn_op = static_cast<AddNOperator*>(addn_it->get()); in Run() local33 CHECK_GE(addn_op->inputs.size(), 2); in Run()34 CHECK_EQ(addn_op->outputs.size(), 1); in Run()37 if (addn_op->inputs.size() != 2) { in Run()43 add_op->inputs.push_back(addn_op->inputs[0]); in Run()44 add_op->inputs.push_back(addn_op->inputs[1]); in Run()45 add_op->outputs = addn_op->outputs; in Run()49 DeleteOpAndArrays(model, addn_op); in Run()
187 auto addn_op = cast<AddNOp>(op); in matchAndRewrite() local191 if (getElementTypeOrSelf(addn_op.getType()).isa<VariantType>()) in matchAndRewrite()193 llvm::SmallVector<Value, 4> operands(addn_op.inputs().begin(), in matchAndRewrite()194 addn_op.inputs().end()); in matchAndRewrite()202 (i + 1 < n) ? rewriter.create<AddV2Op>(addn_op.getLoc(), in matchAndRewrite()209 rewriter.replaceOp(addn_op, operands[0]); in matchAndRewrite()