xref: /aosp_15_r20/external/android-nn-driver/ArmnnDevice.hpp (revision 3e777be0405cee09af5d5785ff37f7cfb5bee59a)
1*3e777be0SXin Li //
2*3e777be0SXin Li // Copyright © 2017 Arm Ltd. All rights reserved.
3*3e777be0SXin Li // SPDX-License-Identifier: MIT
4*3e777be0SXin Li //
5*3e777be0SXin Li 
6*3e777be0SXin Li #pragma once
7*3e777be0SXin Li 
8*3e777be0SXin Li #include "DriverOptions.hpp"
9*3e777be0SXin Li 
10*3e777be0SXin Li #include <armnn/ArmNN.hpp>
11*3e777be0SXin Li 
12*3e777be0SXin Li namespace armnn_driver
13*3e777be0SXin Li {
14*3e777be0SXin Li 
15*3e777be0SXin Li class ArmnnDevice
16*3e777be0SXin Li {
17*3e777be0SXin Li protected:
18*3e777be0SXin Li     ArmnnDevice(DriverOptions options);
~ArmnnDevice()19*3e777be0SXin Li     virtual ~ArmnnDevice() {}
20*3e777be0SXin Li 
21*3e777be0SXin Li protected:
22*3e777be0SXin Li     armnn::IRuntimePtr m_Runtime;
23*3e777be0SXin Li     armnn::IGpuAccTunedParametersPtr m_ClTunedParameters;
24*3e777be0SXin Li     DriverOptions m_Options;
25*3e777be0SXin Li };
26*3e777be0SXin Li 
27*3e777be0SXin Li } // namespace armnn_driver
28