| 55 | } |
| 56 | |
| 57 | void ConvolutionForwardImpl::exec( |
| 58 | _megdnn_tensor_in src, _megdnn_tensor_in filter, _megdnn_tensor_out dst, |
| 59 | const PreprocessedFilter* preprocessed_filter, _megdnn_workspace workspace) { |
| 60 | check_exec( |
| 61 | src.layout, filter.layout, dst.layout, workspace.size, preprocessed_filter); |
| 62 | AlgoBase::ExecArgs args(this, src, filter, dst, workspace); |
| 63 | auto&& algo = get_algorithm(this, src.layout, filter.layout, dst.layout); |
| 64 | algo->exec(args); |
| 65 | } |
| 66 | |
| 67 | const char* ConvolutionForwardImpl::get_algorithm_set_name() const { |
| 68 | return "CUDA CONVOLUTION_FORWARD"; |
no test coverage detected