| 51 | } |
| 52 | |
| 53 | void CustomOpDef::compute( |
| 54 | std::shared_ptr<SmallVector<DeviceTensorND>> inputs, |
| 55 | std::shared_ptr<SmallVector<DeviceTensorND>> outputs) const { |
| 56 | custom::dispatch_custom_op(m_op, m_param, inputs, outputs); |
| 57 | } |
| 58 | |
| 59 | std::tuple<SmallVector<LogicalTensorDesc>, bool> CustomOpDef::infer_output_attrs( |
| 60 | const SmallVector<TensorPtr>& inputs) const { |
no test coverage detected