diff options
Diffstat (limited to 'src/core/NEON')
-rw-r--r-- | src/core/NEON/kernels/NEPadLayerKernel.cpp | 8 | ||||
-rw-r--r-- | src/core/NEON/kernels/NEPadLayerKernel.h | 9 |
2 files changed, 17 insertions, 0 deletions
diff --git a/src/core/NEON/kernels/NEPadLayerKernel.cpp b/src/core/NEON/kernels/NEPadLayerKernel.cpp index 3e2c57a18c..60986812be 100644 --- a/src/core/NEON/kernels/NEPadLayerKernel.cpp +++ b/src/core/NEON/kernels/NEPadLayerKernel.cpp @@ -258,4 +258,12 @@ void NEPadLayerKernel::run(const Window &window, const ThreadInfo &info) (this->*_func)(window); } } + +size_t NEPadLayerKernel::get_mws(const CPUInfo &platform, size_t thread_count) const +{ + ARM_COMPUTE_UNUSED(platform, thread_count); + + return ICPPKernel::small_network_mws; +} + } // namespace arm_compute diff --git a/src/core/NEON/kernels/NEPadLayerKernel.h b/src/core/NEON/kernels/NEPadLayerKernel.h index 00cda7dc22..b3b0725af8 100644 --- a/src/core/NEON/kernels/NEPadLayerKernel.h +++ b/src/core/NEON/kernels/NEPadLayerKernel.h @@ -79,6 +79,15 @@ public: // Inherited methods overridden: void run(const Window &window, const ThreadInfo &info) override; + /** Return minimum workload size of the relevant kernel + * + * @param[in] platform The CPU platform used to create the context. + * @param[in] thread_count Number of threads in the execution. + * + * @return[out] small_network_mws Minimum workload size for requsted configuration. + */ + size_t get_mws(const CPUInfo &platform, size_t thread_count) const override; + private: /** Template function to run the padding function with constant padding * |