35 struct NEPoolingLayer::Impl
37 ITensor *
src{
nullptr};
38 ITensor *
dst{
nullptr};
39 ITensor *indices{
nullptr};
40 std::unique_ptr<cpu::CpuPool2d> op{
nullptr};
41 MemoryGroup memory_group{};
42 ITensorPack run_pack{};
43 WorkspaceData<Tensor> workspace_tensors{};
50 _impl->memory_group =
MemoryGroup(std::move(memory_manager));
57 _impl->indices = indices;
58 _impl->op = std::make_unique<cpu::CpuPool2d>();
59 _impl->op->configure(
input->info(), output->
info(), pool_info, (indices) ? indices->
info() :
nullptr);
64 _impl->workspace_tensors = manage_workspace<Tensor>(_impl->op->workspace(), _impl->memory_group, _impl->run_pack);
79 _impl->op->run(_impl->run_pack);