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