diff options
Diffstat (limited to 'arm_compute/runtime/CL/ICLTuner.h')
| -rw-r--r-- | arm_compute/runtime/CL/ICLTuner.h | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/arm_compute/runtime/CL/ICLTuner.h b/arm_compute/runtime/CL/ICLTuner.h index 0b238180eb..4bc8ddf632 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 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,13 @@ 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] inputs Inputs for the kernel to use + * @param[in, out] outputs Outputs for the kernel to use + */ + virtual void tune_kernel_dynamic(ICLKernel &kernel, const InputTensorMap &inputs, const OutputTensorMap &outputs) = 0; }; -} +} // namespace arm_compute #endif /*ARM_COMPUTE_ICLTUNER_H */ |