| 9 | namespace resize { |
| 10 | |
| 11 | auto apply_on_var_node(const OpDef& def, const VarNodeArray& inputs) { |
| 12 | auto&& op = static_cast<const Resize&>(def); |
| 13 | mgb_assert(inputs.size() == 2); |
| 14 | OperatorNodeConfig config{op.make_name()}; |
| 15 | return opr::Resize::make(inputs[0], inputs[1], op.param(), config); |
| 16 | } |
| 17 | |
| 18 | OP_TRAIT_REG(Resize, Resize).apply_on_var_node(apply_on_var_node).fallback(); |
| 19 |