| 90 | } |
| 91 | |
| 92 | void LocalForwardImpl::exec_use_float_noncontig_batch( |
| 93 | _megdnn_tensor_in src, _megdnn_tensor_in filter, _megdnn_tensor_out dst, |
| 94 | _megdnn_workspace workspace) { |
| 95 | check_exec(src.layout, filter.layout, dst.layout, workspace.size); |
| 96 | auto fp = make_float_kern_param(src, filter, dst, workspace); |
| 97 | auto kptr = dispatch_float_noncontig_batch(src.layout, filter.layout, dst.layout); |
| 98 | auto kern = [fp, kptr]() { kptr(fp); }; |
| 99 | static_cast<naive::HandleImpl*>(handle())->dispatch_kern(kern); |
| 100 | } |
| 101 | |
| 102 | void LocalBackwardDataImpl::exec( |
| 103 | _megdnn_tensor_in filter, _megdnn_tensor_in diff, _megdnn_tensor_out grad, |
nothing calls this directly
no test coverage detected