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

dnn(ocl4dnn): fix args for 'max_pool_forward' kernel

Browse Source
This commit is contained in:
Alexander Alekhin 2018-07-09 18:02:20 +03:00
parent e41f19d2f3
commit b6255ab9e7
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_,
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);
}