diff options
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 |