| 45 | } |
| 46 | |
| 47 | void PoolingForwardImpl::exec( |
| 48 | _megdnn_tensor_in src, _megdnn_tensor_out dst, _megdnn_workspace workspace) { |
| 49 | check_exec(src.layout, dst.layout, workspace.size); |
| 50 | { |
| 51 | AlgoBase::ExecArgs args(this, src, dst, workspace); |
| 52 | auto algo = get_algorithm(this, src.layout, dst.layout); |
| 53 | algo->exec(args); |
| 54 | } |
| 55 | } |
| 56 | |
| 57 | size_t PoolingBackwardImpl::get_workspace_in_bytes( |
| 58 | const TensorLayout& src, const TensorLayout& dst, const TensorLayout& diff, |
nothing calls this directly
no test coverage detected