potok/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-07-29 23:33:05 +04:00
Code Packages Releases Wiki Activity

Merge pull request #11924 from alalek:dnn_ocl_fix_max_pool_forward

Browse Source
This commit is contained in:
Alexander Alekhin
2018-07-09 16:25:34 +00:00
parent 529d38613b b6255ab9e7
commit 7fe0727930
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/src/ocl4dnn/src/ocl4dnn_pool.cpp
+3 -2
View File
@@ -132,9 +132,10 @@ bool OCL4DNNPool<Dtype>::Forward(const UMat& bottom,
width_,
pooled_height_,
pooled_width_,
ocl::KernelArg::PtrWriteOnly(top),
ocl::KernelArg::PtrWriteOnly(top_mask)
ocl::KernelArg::PtrWriteOnly(top)
);
if (computeMaxIdx)
oclk_max_pool_forward.set(8, ocl::KernelArg::PtrWriteOnly(top_mask)); // TODO remove magic number. Extend cv::ocl::Kernel API
ret = oclk_max_pool_forward.run(1, global, local, false);
}