diff options
Diffstat (limited to 'src/core/cpu/kernels/CpuPool2dKernel.h')
-rw-r--r-- | src/core/cpu/kernels/CpuPool2dKernel.h | 5 |
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 |