diff options
| author | Cedric Nugteren <web@cedricnugteren.nl> | 2018-07-29 10:37:14 +0200 |
|---|---|---|
| committer | Cedric Nugteren <web@cedricnugteren.nl> | 2018-07-29 10:37:14 +0200 |
| commit | 2dd539f911dc9e53f188ed404ba95a795ee56fb6 (patch) | |
| tree | dca6b8bc72ea6152893cda3cfb9aa4ee62a13880 /include/clblast_cuda.h | |
| parent | 5903820ba2736ec4a5f338c2325f59f511b4a0e8 (diff) | |
Removed complex numbers support for CONVGEMM
Diffstat (limited to 'include/clblast_cuda.h')
| -rw-r--r-- | include/clblast_cuda.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/include/clblast_cuda.h b/include/clblast_cuda.h index 01044037..1bbd898e 100644 --- a/include/clblast_cuda.h +++ b/include/clblast_cuda.h @@ -608,7 +608,7 @@ StatusCode Im2col(const size_t channels, const size_t height, const size_t width CUdeviceptr col_buffer, const size_t col_offset, const CUcontext context, const CUdevice device); -// Batched convolution as GEMM (non-BLAS function): SCONVGEMM/DCONVGEMM/CCONVGEMM/ZCONVGEMM/HCONVGEMM +// Batched convolution as GEMM (non-BLAS function): SCONVGEMM/DCONVGEMM/HCONVGEMM template <typename T> StatusCode Convgemm(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 size_t num_kernels, const size_t batch_count, const CUdeviceptr im_buffer, const size_t im_offset, |