diff options
Diffstat (limited to 'arm_compute/core/CL/kernels/CLIm2ColKernel.h')
-rw-r--r-- | arm_compute/core/CL/kernels/CLIm2ColKernel.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/arm_compute/core/CL/kernels/CLIm2ColKernel.h b/arm_compute/core/CL/kernels/CLIm2ColKernel.h index 43812e42a3..45111fcedd 100644 --- a/arm_compute/core/CL/kernels/CLIm2ColKernel.h +++ b/arm_compute/core/CL/kernels/CLIm2ColKernel.h @@ -113,9 +113,10 @@ private: /** Common signature for the kernel to run */ using Im2ColFunction = void (CLIm2ColKernel::*)(const Window &, cl::CommandQueue &); -private: +public: const ICLTensor *_input; ICLTensor *_output; + PadStrideInfo _conv_info; std::pair<unsigned int, unsigned int> _convolved_dims; unsigned int _num_elems_processed_per_iteration; Im2ColFunction _run_func; |