Searched defs:nnapi_wrapper (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/nnapi/ |
| H A D | nnapi_wrapper.h | 23 struct nnapi_wrapper { struct
24 int(*_getDeviceCount)(uint32_t* numDevices);
25 int(*_getDevice)(uint32_t devIndex, ANeuralNetworksDevice** device);
26 int(*Device_getName)(const ANeuralNetworksDevice* device, const char** name);
27 int(*Device_getVersion)(const ANeuralNetworksDevice* device, const char** version);
28 int(*Device_getFeatureLevel)(const ANeuralNetworksDevice* device, int64_t* featureLevel);
29 …odel* model, const ANeuralNetworksDevice* const* devices, uint32_t numDevices, bool* supportedOps);
30 …uralNetworksDevice* const* devices, uint32_t numDevices, ANeuralNetworksCompilation** compilation);
31 int(*Execution_compute)(ANeuralNetworksExecution* execution);
32 …ory_createFromFd)(size_t size, int protect, int fd, size_t offset, ANeuralNetworksMemory** memory);
[all …]
|