aboutsummaryrefslogtreecommitdiff
path: root/src/core/NEON/kernels/NEPadLayerKernel.h
diff options
context:
space:
mode:
authorDana Zlotnik <dana.zlotnik@arm.com>2021-10-07 15:31:54 +0300
committerDana Zlotnik <dana.zlotnik@arm.com>2021-10-18 09:37:41 +0000
commit4cdd6b80754b3abbf54650d9359cf940a4aaf772 (patch)
tree6f0096f6d2abbc8410d4aec59f96263a05af5c2e /src/core/NEON/kernels/NEPadLayerKernel.h
parent6d9c982a5aec543d1f7f198f0fee10a7a3a78ddb (diff)
downloadComputeLibrary-4cdd6b80754b3abbf54650d9359cf940a4aaf772.tar.gz
Implement Minimum Workload Size (MWS) in all CPPKernels used by small networks
* create get_mws method in ICPPKernel class that retuns default value for all kernels * overwrite the default value for all the kernels used by small networks (according to banchmark case) Resolves COMPMID-4648 Change-Id: I46d7cae61217213279d2ee740edc73f600b6d576 Signed-off-by: Dana Zlotnik <dana.zlotnik@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6412 Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: SiCong Li <sicong.li@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/NEON/kernels/NEPadLayerKernel.h')
-rw-r--r--src/core/NEON/kernels/NEPadLayerKernel.h9
1 files changed, 9 insertions, 0 deletions
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
*