diff options
Diffstat (limited to 'src/backends/cl/ClBackendContext.hpp')
| -rw-r--r-- | src/backends/cl/ClBackendContext.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/backends/cl/ClBackendContext.hpp b/src/backends/cl/ClBackendContext.hpp index bcac0d245e..af988a96dc 100644 --- a/src/backends/cl/ClBackendContext.hpp +++ b/src/backends/cl/ClBackendContext.hpp @@ -9,6 +9,7 @@ #include <mutex> #include <arm_compute/runtime/CL/CLTuner.h> +#include <arm_compute/runtime/CL/CLGEMMHeuristicsHandle.h> namespace armnn { @@ -35,6 +36,10 @@ private: std::unique_ptr<arm_compute::CLTuner> m_Tuner; std::string m_TuningFile; + +protected: + arm_compute::CLGEMMHeuristicsHandle m_MLGOTuner; + std::string m_MLGOTuningFile; }; } // namespace armnn
\ No newline at end of file |