| 74 | } |
| 75 | |
| 76 | LocalForwardImpl::FloatNoncontigBatchKernParam LocalForwardImpl::make_float_kern_param( |
| 77 | _megdnn_tensor_in src, _megdnn_tensor_in filter, _megdnn_tensor_out dst, |
| 78 | _megdnn_workspace workspace) const { |
| 79 | return {src.get_ref_ptr(), filter.get_ref_ptr(), dst.get_ref_ptr(), |
| 80 | // n |
| 81 | src.layout.shape[0], |
| 82 | // ic, ih, iw, oc, oh, ow, fh, fw |
| 83 | src.layout.shape[1], src.layout.shape[2], src.layout.shape[3], |
| 84 | dst.layout.shape[1], dst.layout.shape[2], dst.layout.shape[3], |
| 85 | filter.layout.shape[3], filter.layout.shape[4], |
| 86 | // ph, pw, sh, sw |
| 87 | param().pad_h, param().pad_w, param().stride_h, param().stride_w, |
| 88 | // inp_bs, out_bs |
| 89 | src.layout.stride[0], dst.layout.stride[0], workspace.raw_ptr}; |
| 90 | } |
| 91 | |
| 92 | void LocalForwardImpl::exec_use_float_noncontig_batch( |
| 93 | _megdnn_tensor_in src, _megdnn_tensor_in filter, _megdnn_tensor_out dst, |
no test coverage detected