| 554 | } |
| 555 | |
| 556 | void compute_impl( |
| 557 | std::shared_ptr<const CustomOp> op, const Param& param, |
| 558 | std::shared_ptr<::megdnn::SmallVector<::mgb::DeviceTensorND>> inputs, |
| 559 | std::shared_ptr<::megdnn::SmallVector<::mgb::DeviceTensorND>> outputs) { |
| 560 | std::vector<custom::Tensor> custom_inputs; |
| 561 | for (size_t i = 0; i < inputs->size(); ++i) { |
| 562 | custom_inputs.emplace_back(to_custom_tensor(inputs->operator[](i))); |
| 563 | } |
| 564 | std::vector<custom::Tensor> custom_outputs; |
| 565 | for (size_t i = 0; i < outputs->size(); ++i) { |
| 566 | custom_outputs.emplace_back(to_custom_tensor(outputs->operator[](i))); |
| 567 | } |
| 568 | op->compute(custom_inputs, param, custom_outputs); |
| 569 | } |
| 570 | |
| 571 | void dispatch_custom_op( |
| 572 | std::shared_ptr<const CustomOp> op, const Param& param, |
no test coverage detected