diff options
Diffstat (limited to 'arm_compute/core/CL/kernels')
4 files changed, 6 insertions, 5 deletions
diff --git a/arm_compute/core/CL/kernels/CLCol2ImKernel.h b/arm_compute/core/CL/kernels/CLCol2ImKernel.h index 24d0fdd914..3779325efe 100644 --- a/arm_compute/core/CL/kernels/CLCol2ImKernel.h +++ b/arm_compute/core/CL/kernels/CLCol2ImKernel.h @@ -86,7 +86,7 @@ public: // Inherited methods overridden: void run(const Window &window, cl::CommandQueue &queue) override; -private: +public: const ICLTensor *_input; ICLTensor *_output; std::pair<unsigned int, unsigned int> _convolved_dims; diff --git a/arm_compute/core/CL/kernels/CLGEMMMatrixMultiplyKernel.h b/arm_compute/core/CL/kernels/CLGEMMMatrixMultiplyKernel.h index ee7e7c0e97..13802b97ad 100644 --- a/arm_compute/core/CL/kernels/CLGEMMMatrixMultiplyKernel.h +++ b/arm_compute/core/CL/kernels/CLGEMMMatrixMultiplyKernel.h @@ -80,7 +80,7 @@ public: // Inherited methods overridden: void run(const Window &window, cl::CommandQueue &queue) override; -private: +public: const ICLTensor *_input0; const ICLTensor *_input1; ICLTensor *_output; 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; diff --git a/arm_compute/core/CL/kernels/CLPoolingLayerKernel.h b/arm_compute/core/CL/kernels/CLPoolingLayerKernel.h index e9ce28b3f9..c13507785b 100644 --- a/arm_compute/core/CL/kernels/CLPoolingLayerKernel.h +++ b/arm_compute/core/CL/kernels/CLPoolingLayerKernel.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017 ARM Limited. + * Copyright (c) 2017-2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -72,7 +72,7 @@ public: void run(const Window &window, cl::CommandQueue &queue) override; BorderSize border_size() const override; -private: +public: const ICLTensor *_input; ICLTensor *_output; PoolingLayerInfo _pool_info; |