| 51 | } |
| 52 | |
| 53 | void NEPooling3dLayer::configure(const ITensor *input, ITensor *output, const Pooling3dLayerInfo &pool_info) |
| 54 | { |
| 55 | ARM_COMPUTE_TRACE_EVENT(ARM_COMPUTE_PROF_CAT_CPU, ARM_COMPUTE_PROF_LVL_CPU, "NEPooling3dLayer::configure"); |
| 56 | _impl->src = input; |
| 57 | _impl->dst = output; |
| 58 | _impl->op = std::make_unique<cpu::CpuPool3d>(); |
| 59 | _impl->op->configure(input->info(), output->info(), pool_info); |
| 60 | |
| 61 | _impl->run_pack = {{TensorType::ACL_SRC, _impl->src}, {TensorType::ACL_DST_0, _impl->dst}}; |
| 62 | _impl->workspace_tensors = manage_workspace<Tensor>(_impl->op->workspace(), _impl->memory_group, _impl->run_pack); |
| 63 | } |
| 64 | |
| 65 | Status |
| 66 | NEPooling3dLayer::validate(const ITensorInfo *input, const ITensorInfo *output, const Pooling3dLayerInfo &pool_info) |