| 9 | namespace cambricon_runtime { |
| 10 | |
| 11 | auto apply_on_var_node(const OpDef& def, const VarNodeArray& inputs) { |
| 12 | auto&& op = static_cast<const CambriconRuntime&>(def); |
| 13 | SymbolVarArray symbol_var_inputs(inputs.begin(), inputs.end()); |
| 14 | OperatorNodeConfig config{op.make_name()}; |
| 15 | return opr::CambriconRuntimeOpr::make( |
| 16 | op.buf.c_str(), op.buf_size, op.symbol, symbol_var_inputs, |
| 17 | op.tensor_dim_mutable, config); |
| 18 | } |
| 19 | OP_TRAIT_REG(CambriconRuntime, CambriconRuntime) |
| 20 | .apply_on_var_node(apply_on_var_node) |
| 21 | .fallback(); |