diff options
Diffstat (limited to 'arm_compute/runtime/CL/ICLTuner.h')
| -rw-r--r-- | arm_compute/runtime/CL/ICLTuner.h | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/arm_compute/runtime/CL/ICLTuner.h b/arm_compute/runtime/CL/ICLTuner.h index ba4eeac8e0..fa7a1424b8 100644 --- a/arm_compute/runtime/CL/ICLTuner.h +++ b/arm_compute/runtime/CL/ICLTuner.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2019 ARM Limited. + * Copyright (c) 2017-2020, 2022 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -24,6 +24,8 @@ #ifndef ARM_COMPUTE_ICLTUNER_H #define ARM_COMPUTE_ICLTUNER_H +#include "arm_compute/core/experimental/Types.h" + namespace arm_compute { class ICLKernel; @@ -49,6 +51,12 @@ public: * @param[in] kernel Kernel to tune */ virtual void tune_kernel_dynamic(ICLKernel &kernel) = 0; + /** Tune OpenCL kernel dynamically + * + * @param[in] kernel Kernel to tune + * @param[in, out] tensors Tensors for the kernel to use + */ + virtual void tune_kernel_dynamic(ICLKernel &kernel, ITensorPack &tensors) = 0; }; -} +} // namespace arm_compute #endif /*ARM_COMPUTE_ICLTUNER_H */ |