nn_gpu/executor_manager.h at master · feijiang1/nn_gpu · GitHub

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
#ifndef ANDROID_HARDWARE_NEURALNETWORKS_V1_0_EXECUTOR_MANAGER_H
#define ANDROID_HARDWARE_NEURALNETWORKS_V1_0_EXECUTOR_MANAGER_H

#include <vector>

#include "base_executor.h"

namespace android {
namespace hardware {
namespace neuralnetworks {
namespace V1_0 {
namespace implementation {

class ExecutorManager
{
public:
    static bool initPerProcess();
    static void deinitPerProcess();
    static void getCapabilities(Capabilities& cap);
    static std::vector<bool> getSupportedOperations(const Model& model);
    static BaseExecutor* createExecutor(const Model& model);
private:
    enum ExecutorType
    {
        ET_GLES_CS,
    };
    static ExecutorType type;
};

}  // namespace implementation
}  // namespace V1_0
}  // namespace neuralnetworks
}  // namespace hardware
}  // namespace android

#endif