diff options
author | Cedric Nugteren <web@cedricnugteren.nl> | 2017-07-02 12:10:22 +0200 |
---|---|---|
committer | Cedric Nugteren <web@cedricnugteren.nl> | 2017-07-02 12:10:22 +0200 |
commit | 84ec50e29dc123b956a85e1dc37dae123323e420 (patch) | |
tree | 52b79418233e713fcc828cd59cd9bb99ffdd90e1 /include/clblast.h | |
parent | 75c0e861b842dbd08def5e55696fd79d713afc96 (diff) |
Added interface and stubs for the im2col routine
Diffstat (limited to 'include/clblast.h')
-rw-r--r-- | include/clblast.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/include/clblast.h b/include/clblast.h index 354ca591..e073b211 100644 --- a/include/clblast.h +++ b/include/clblast.h @@ -617,6 +617,13 @@ StatusCode Omatcopy(const Layout layout, const Transpose a_transpose, cl_mem b_buffer, const size_t b_offset, const size_t b_ld, cl_command_queue* queue, cl_event* event = nullptr); +// Im2col function (non-BLAS function): SIM2COL/DIM2COL/CIM2COL/ZIM2COL/HIM2COL +template <typename T> +StatusCode Im2col(const size_t channels, const size_t height, const size_t width, const size_t kernel_h, const size_t kernel_w, const size_t pad_h, const size_t pad_w, const size_t stride_h, const size_t stride_w, const size_t dilation_h, const size_t dilation_w, + const cl_mem im_buffer, const size_t im_offset, + cl_mem col_buffer, const size_t col_offset, + cl_command_queue* queue, cl_event* event = nullptr); + // Batched version of AXPY: SAXPYBATCHED/DAXPYBATCHED/CAXPYBATCHED/ZAXPYBATCHED/HAXPYBATCHED template <typename T> StatusCode AxpyBatched(const size_t n, |