| 1780 | } |
| 1781 | |
| 1782 | VARP _Select(VARP select, VARP input0, VARP input1) { |
| 1783 | std::unique_ptr<MNN::OpT> selectOp(new MNN::OpT); |
| 1784 | selectOp->type = MNN::OpType_Select; |
| 1785 | return (Variable::create(Expr::create(std::move(selectOp), {select, input0, input1}))); |
| 1786 | } |
| 1787 | |
| 1788 | std::vector<VARP> _TopKV2(VARP input0, VARP input1) { |
| 1789 | std::unique_ptr<OpT> op(new OpT); |