diff options
Diffstat (limited to 'src/cpu/kernels/CpuSubKernel.h')
-rw-r--r-- | src/cpu/kernels/CpuSubKernel.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/cpu/kernels/CpuSubKernel.h b/src/cpu/kernels/CpuSubKernel.h index 323a3f1316..e835bac3d5 100644 --- a/src/cpu/kernels/CpuSubKernel.h +++ b/src/cpu/kernels/CpuSubKernel.h @@ -82,10 +82,16 @@ public: static const std::vector<SubKernel> &get_available_kernels(); + size_t get_split_dimension() const + { + return _split_dimension; + } + private: ConvertPolicy _policy{}; SubKernelPtr _run_method{ nullptr }; std::string _name{}; + size_t _split_dimension{ Window::DimY }; }; } // namespace kernels } // namespace cpu |