| 200 | } |
| 201 | |
| 202 | VarNodeArray apply_on_var_node(const OpDef& def, const cg::VarNodeArray& inputs) { |
| 203 | auto&& op = static_cast<const CustomOpDef&>(def); |
| 204 | OperatorNodeConfig config; |
| 205 | VarNodeArray outputs = |
| 206 | opr::CustomOpNode::make(op.impl(), inputs, op.param(), config); |
| 207 | return outputs; |
| 208 | } |
| 209 | |
| 210 | std::tuple<SmallVector<LogicalTensorDesc>, bool> infer_output_attrs_fallible( |
| 211 | const OpDef& def, const SmallVector<LogicalTensorDesc>& inputs) { |