aboutsummaryrefslogtreecommitdiff
path: root/src/core/cpu/kernels/CpuPool2dKernel.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/core/cpu/kernels/CpuPool2dKernel.h')
-rw-r--r--src/core/cpu/kernels/CpuPool2dKernel.h5
1 files changed, 5 insertions, 0 deletions
diff --git a/src/core/cpu/kernels/CpuPool2dKernel.h b/src/core/cpu/kernels/CpuPool2dKernel.h
index 95298004e9..ff7d7bb21d 100644
--- a/src/core/cpu/kernels/CpuPool2dKernel.h
+++ b/src/core/cpu/kernels/CpuPool2dKernel.h
@@ -65,12 +65,17 @@ public:
const char *name() const override;
private:
+ using PoolingKernelPtr = std::add_pointer<void(const ITensor *, ITensor *, ITensor *, PoolingLayerInfo &, const Window &, const Window &)>::type;
+
+private:
PoolingLayerInfo _pool_info{};
DataLayout _data_layout{ DataLayout::UNKNOWN };
unsigned int _num_elems_processed_per_iteration{ 0 };
BorderSize _border_size{ 0 };
Size2D _pool_size{};
int _pool_stride_x{};
+ PoolingKernelPtr _run_method{ nullptr };
+ std::string _name{};
};
} // namespace kernels
} // namespace cpu