Reset OpenCL kernels if batch size changes

This commit is contained in:
Dmitry Kurtaev
2018-03-07 16:56:41 +03:00
parent 0a61ebdd66
commit 0f01b40dd5
6 changed files with 34 additions and 8 deletions
+5
View File
@@ -96,6 +96,11 @@ public:
}
#ifdef HAVE_OPENCL
void finalize(const std::vector<Mat*> &inputs, std::vector<Mat> &outputs)
{
lrnOp.release();
}
bool forward_ocl(InputArrayOfArrays inps, OutputArrayOfArrays outs, OutputArrayOfArrays internals)
{
std::vector<UMat> inputs;