| 56 | pool_kernel_t kernel_run = nullptr; |
| 57 | |
| 58 | void pool_kernel(int i, int id, void* data, const float* input, float* output, int in_h, int in_w, |
| 59 | int out_h, int out_w, int kernel_h, int kernel_w, int stride_h, int stride_w, |
| 60 | int pad_h0, int pad_w0, int pad_h1, int pad_w1, int is_caffe) |
| 61 | { |
| 62 | int step = ((int*)data)[0]; |
| 63 | int in_channel_size = in_h * in_w; |
| 64 | int out_channel_size = out_h * out_w; |
| 65 | const float* cur_input = input + id * step * in_channel_size; |
| 66 | float* cur_output = output + id * step * out_channel_size; |
| 67 | kernel_run(cur_input, cur_output, step, in_h, in_w, out_h, out_w, kernel_h, kernel_w, |
| 68 | stride_h, stride_w, pad_h0, pad_w0, pad_h1, pad_w1, is_caffe); |
| 69 | } |
| 70 | |
| 71 | bool Prerun(Node* node) |
| 72 | { |
nothing calls this directly
no test coverage detected