diff options
author | Cedric Nugteren <web@cedricnugteren.nl> | 2018-05-05 14:06:33 +0200 |
---|---|---|
committer | Cedric Nugteren <web@cedricnugteren.nl> | 2018-05-05 14:06:33 +0200 |
commit | 2776d761768295b01a8be7c333dbb337805d7f77 (patch) | |
tree | 3fb955ea8e6d962e60dc6c8eba4b5e9800c3db91 /doc | |
parent | 8b381480f8bd3b40cc030b07a599da10092b8117 (diff) |
Added interface of batched convolution as GEMM
Diffstat (limited to 'doc')
-rw-r--r-- | doc/api.md | 8 |
1 files changed, 4 insertions, 4 deletions
@@ -3063,10 +3063,10 @@ Arguments to IM2COL: * `const size_t stride_w`: Integer size argument. This value must be positive. * `const size_t dilation_h`: Integer size argument. This value must be positive. * `const size_t dilation_w`: Integer size argument. This value must be positive. -* `const cl_mem im_buffer`: OpenCL buffer to store the input im vector. -* `const size_t im_offset`: The offset in elements from the start of the input im vector. -* `cl_mem col_buffer`: OpenCL buffer to store the output col vector. -* `const size_t col_offset`: The offset in elements from the start of the output col vector. +* `const cl_mem im_buffer`: OpenCL buffer to store the input im tensor. +* `const size_t im_offset`: The offset in elements from the start of the input im tensor. +* `cl_mem col_buffer`: OpenCL buffer to store the output col tensor. +* `const size_t col_offset`: The offset in elements from the start of the output col tensor. * `cl_command_queue* queue`: Pointer to an OpenCL command queue associated with a context and device to execute the routine on. * `cl_event* event`: Pointer to an OpenCL event to be able to wait for completion of the routine's OpenCL kernel(s). This is an optional argument. |