forked from projectceladon/nn_gpu
-
Notifications
You must be signed in to change notification settings - Fork 0
/
device.h
executable file
·28 lines (21 loc) · 840 Bytes
/
device.h
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
#ifndef ANDROID_HARDWARE_NEURALNETWORKS_V1_0_DEVICE_H
#define ANDROID_HARDWARE_NEURALNETWORKS_V1_0_DEVICE_H
#include "hal_types.h"
NAME_SPACE_BEGIN
class Device : public IDevice {
public:
Device(const char* name) : mName(name) {}
~Device() override {}
Return<void> getCapabilities(getCapabilities_cb _hidl_cb) override;
Return<void> getSupportedOperations(const Model& model, getSupportedOperations_cb cb) override;
Return<ErrorStatus> prepareModel(const Model& model,
const sp<IPreparedModelCallback>& callback) override;
Return<DeviceStatus> getStatus() override;
// Starts and runs the driver service. Typically called from main().
// This will return only once the service shuts down.
int run();
protected:
std::string mName;
};
NAME_SPACE_STOP
#endif