mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-07 17:44:04 +08:00
Code Issues Actions Packages Projects 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
commit 7fe0727930
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/dnn/src/ocl4dnn/src/ocl4dnn_pool.cpp
View File

@ -132,9 +132,10 @@ bool OCL4DNNPool<Dtype>::Forward(const UMat& bottom,
width_, width_,
pooled_height_, pooled_height_,
pooled_width_, pooled_width_,
ocl::KernelArg::PtrWriteOnly(top), ocl::KernelArg::PtrWriteOnly(top)
ocl::KernelArg::PtrWriteOnly(top_mask)
); );
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); ret = oclk_max_pool_forward.run(1, global, local, false);
} }