diff options
Diffstat (limited to 'arm_compute/core/CL/kernels')
-rw-r--r-- | arm_compute/core/CL/kernels/CLScaleKernel.h | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/arm_compute/core/CL/kernels/CLScaleKernel.h b/arm_compute/core/CL/kernels/CLScaleKernel.h index b089e8f61a..db87519c0f 100644 --- a/arm_compute/core/CL/kernels/CLScaleKernel.h +++ b/arm_compute/core/CL/kernels/CLScaleKernel.h @@ -58,10 +58,23 @@ public: * @return a status */ static Status validate(const ITensorInfo *input, const ITensorInfo *output, InterpolationPolicy policy, BorderMode border_mode, SamplingPolicy sampling_policy = SamplingPolicy::CENTER); + /** Input tensor accessor. + * + * @return Pointer to input tensor. + */ + const ICLTensor *input() const; + /** Output tensor accessor. + * + * @return Pointer to output tensor. + */ + const ICLTensor *output() const; // Inherited methods overridden: BorderSize border_size() const override; void run(const Window &window, cl::CommandQueue &queue) override; + +public: + InterpolationPolicy _interpolationPolicy = InterpolationPolicy::BILINEAR; }; } // namespace arm_compute #endif /*__ARM_COMPUTE_CLSCALEKERNEL_H__ */ |