blob: 8b66e774812ad8168344dc088b3dc78d9648f2b1 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// See LICENSE file in the project root for full license information.
//
#pragma once
#include "HalInterfaces.h"
#include "NeuralNetworks.h"
#include <armnn/ArmNN.hpp>
#include <memory>
#include <set>
#include <string>
namespace armnn_driver
{
class DriverOptions
{
public:
DriverOptions(armnn::Compute computeDevice);
DriverOptions(int argc, char** argv);
DriverOptions(DriverOptions&& other) = default;
armnn::Compute GetComputeDevice() const { return m_ComputeDevice; }
bool IsVerboseLoggingEnabled() const { return m_VerboseLogging; }
const std::string& GetRequestInputsAndOutputsDumpDir() const { return m_RequestInputsAndOutputsDumpDir; }
bool UseAndroidNnCpuExecutor() const { return m_UseAndroidNnCpuExecutor; }
const std::set<unsigned int>& GetForcedUnsupportedOperations() const { return m_ForcedUnsupportedOperations; }
const std::string& GetClTunedParametersFile() const { return m_ClTunedParametersFile; }
armnn::IClTunedParameters::Mode GetClTunedParametersMode() const { return m_ClTunedParametersMode; }
private:
armnn::Compute m_ComputeDevice;
bool m_VerboseLogging;
bool m_UseAndroidNnCpuExecutor;
std::string m_RequestInputsAndOutputsDumpDir;
std::set<unsigned int> m_ForcedUnsupportedOperations;
std::string m_ClTunedParametersFile;
armnn::IClTunedParameters::Mode m_ClTunedParametersMode;
};
class ArmnnDriver : public IDevice {
public:
ArmnnDriver(DriverOptions options);
virtual ~ArmnnDriver() {}
virtual Return<void> getCapabilities(getCapabilities_cb _hidl_cb) override;
virtual Return<void> getSupportedOperations(const Model &model,
getSupportedOperations_cb _hidl_cb) override;
virtual Return<ErrorStatus> prepareModel(const Model &model,
const android::sp<IPreparedModelCallback>& callback);
virtual Return<DeviceStatus> getStatus() override;
private:
armnn::IRuntimePtr m_Runtime;
armnn::IClTunedParametersPtr m_ClTunedParameters;
DriverOptions m_Options;
};
}
|
