| 34 | namespace opencl |
| 35 | { |
| 36 | void ClPool2d::configure(const ClCompileContext &compile_context, |
| 37 | ITensorInfo *src, |
| 38 | ITensorInfo *dst, |
| 39 | const PoolingLayerInfo &info, |
| 40 | ITensorInfo *indices) |
| 41 | { |
| 42 | ARM_COMPUTE_ERROR_ON_NULLPTR(src); |
| 43 | ARM_COMPUTE_LOG_PARAMS(src, dst, info, indices); |
| 44 | |
| 45 | // Configure pooling kernel |
| 46 | auto k = std::make_unique<kernels::ClPool2dKernel>(); |
| 47 | k->set_target(CLScheduler::get().target()); |
| 48 | k->configure(compile_context, src, dst, info, indices); |
| 49 | _kernel = std::move(k); |
| 50 | |
| 51 | // Tune kernels |
| 52 | CLScheduler::get().tune_kernel_static(*_kernel); |
| 53 | } |
| 54 | |
| 55 | Status ClPool2d::validate(const ITensorInfo *src, |
| 56 | const ITensorInfo *dst, |
nothing calls this directly
no test coverage detected