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