diff options
author | Adnan AlSinan <adnan.alsinan@arm.com> | 2022-01-24 10:20:40 +0000 |
---|---|---|
committer | Adnan AlSinan <adnan.alsinan@arm.com> | 2022-01-27 10:06:22 +0000 |
commit | 0ef2c2176fd99319342a4174e15c0263ede236cd (patch) | |
tree | 927d8d056fbe37005edb89a98f84c7d720cf50b6 /src/cpu/kernels/CpuDirectConv2dKernel.h | |
parent | 21391c3f8b234c1a141e1c2d192072c91c254b33 (diff) | |
download | ComputeLibrary-0ef2c2176fd99319342a4174e15c0263ede236cd.tar.gz |
Remove padding from CpuDirectConv2dKernel
- Replaces NCHW kernels with a generic one
- Removes padding
Signed-off-by: Adnan AlSinan <adnan.alsinan@arm.com>
Change-Id: Ia3d8f788b49a878085b296c11eb83d0152b49eef
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6996
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/cpu/kernels/CpuDirectConv2dKernel.h')
-rw-r--r-- | src/cpu/kernels/CpuDirectConv2dKernel.h | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/src/cpu/kernels/CpuDirectConv2dKernel.h b/src/cpu/kernels/CpuDirectConv2dKernel.h index 1f5568743e..09fa5898cc 100644 --- a/src/cpu/kernels/CpuDirectConv2dKernel.h +++ b/src/cpu/kernels/CpuDirectConv2dKernel.h @@ -66,7 +66,6 @@ public: // Inherited methods overridden: void run_op(ITensorPack &tensors, const Window &window, const ThreadInfo &info) override; const char *name() const override; - BorderSize border_size() const override; private: /* Template function for optimized convolution NHWC */ @@ -77,12 +76,12 @@ private: template <typename T> void convolve_nhwc(const Window &window, const ITensor *src, const ITensor *weights, ITensor *dst); + /* Template function for convolution NCHW */ + template <typename T> + void convolve_nchw(const Window &window, const ITensor *src, const ITensor *weights, ITensor *dst); + PadStrideInfo _conv_info{}; - BorderSize _border_size{}; unsigned int _kernel_size{ 0 }; - unsigned int _num_weight_elems_read_per_row{ 0 }; - unsigned int _num_elems_read_per_iteration{ 0 }; - unsigned int _num_elems_written_per_iteration{ 0 }; DataLayout _data_layout{ DataLayout::UNKNOWN }; }; } // namespace kernels |