Home
last modified time | relevance | path

Searched refs:ANeuralNetworksDevice (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/nnapi/
Dnnapi_handler.cc108 [](uint32_t devIndex, ANeuralNetworksDevice** device) -> int { in SetNnapiSupportedDevice()
115 reinterpret_cast<ANeuralNetworksDevice*>(NnApiHandler::kNnapiDevice); in SetNnapiSupportedDevice()
117 *device = reinterpret_cast<ANeuralNetworksDevice*>( in SetNnapiSupportedDevice()
123 [](const ANeuralNetworksDevice* device, const char** name) -> int { in SetNnapiSupportedDevice()
125 reinterpret_cast<ANeuralNetworksDevice*>(NnApiHandler::kNnapiDevice)) { in SetNnapiSupportedDevice()
129 if (device == reinterpret_cast<ANeuralNetworksDevice*>( in SetNnapiSupportedDevice()
138 [](const ANeuralNetworksDevice* device, int64_t* featureLevel) -> int { in SetNnapiSupportedDevice()
140 reinterpret_cast<ANeuralNetworksDevice*>(NnApiHandler::kNnapiDevice)) { in SetNnapiSupportedDevice()
144 if (device == reinterpret_cast<ANeuralNetworksDevice*>( in SetNnapiSupportedDevice()
Dnnapi_handler.h70 [](uint32_t devIndex, ANeuralNetworksDevice** device) -> int { in GetDeviceReturns()
72 reinterpret_cast<ANeuralNetworksDevice*>(NnApiHandler::kNnapiDevice); in GetDeviceReturns()
77 void StubGetDeviceWith(int(stub)(uint32_t, ANeuralNetworksDevice**)) { in StubGetDeviceWith() argument
84 [](const ANeuralNetworksDevice* device, const char** name) -> int { in GetDeviceNameReturns()
92 void StubGetDeviceNameWith(int(stub)(const ANeuralNetworksDevice*, in StubGetDeviceNameWith() argument
193 const ANeuralNetworksDevice* const* devices, uint32_t numDevices, in CompilationCreateForDevicesReturns()
201 ANeuralNetworksModel* model, const ANeuralNetworksDevice* const* devices, in StubCompilationCreateForDevicesWith()
249 const ANeuralNetworksDevice* const* devices, uint32_t numDevices, in GetSupportedOperationsForDevicesReturns()
255 const ANeuralNetworksDevice* const* devices, in StubGetSupportedOperationsForDevicesWith()
Dnnapi_implementation.h624 ANeuralNetworksDevice** device);
642 int (*ANeuralNetworksDevice_getName)(const ANeuralNetworksDevice* device,
676 int (*ANeuralNetworksDevice_getVersion)(const ANeuralNetworksDevice* device,
698 const ANeuralNetworksDevice* device, int64_t* featureLevel);
716 int (*ANeuralNetworksDevice_getType)(const ANeuralNetworksDevice* device,
740 const ANeuralNetworksDevice* const* devices, uint32_t numDevices,
762 ANeuralNetworksModel* model, const ANeuralNetworksDevice* const* devices,
1172 const ANeuralNetworksDevice* device, const char* extensionName,
DNeuralNetworksTypes.h469 typedef struct ANeuralNetworksDevice ANeuralNetworksDevice; typedef
561 ANeuralNetworksDevice** device);
564 const ANeuralNetworksDevice* device, const char** name);
567 const ANeuralNetworksDevice* device, int32_t* type);
570 const ANeuralNetworksDevice* device, const char** version);
573 const ANeuralNetworksDevice* device, int64_t* featureLevel);
577 const ANeuralNetworksDevice* const* devices, uint32_t numDevices,
581 ANeuralNetworksModel* model, const ANeuralNetworksDevice* const* devices,
657 const ANeuralNetworksDevice* device, const char* extensionName,
DNeuralNetworksShim.h745 ANeuralNetworksDevice** device) { in ANeuralNetworks_getDevice()
766 inline int ANeuralNetworksDevice_getName(const ANeuralNetworksDevice* device, in ANeuralNetworksDevice_getName()
803 inline int ANeuralNetworksDevice_getVersion(const ANeuralNetworksDevice* device, in ANeuralNetworksDevice_getVersion()
828 const ANeuralNetworksDevice* device, int64_t* featureLevel) { in ANeuralNetworksDevice_getFeatureLevel()
854 const ANeuralNetworksDevice* const* devices, uint32_t numDevices, in ANeuralNetworksModel_getSupportedOperationsForDevices()
879 ANeuralNetworksModel* model, const ANeuralNetworksDevice* const* devices, in ANeuralNetworksCompilation_createForDevices()
1169 const ANeuralNetworksDevice* device, const char* extensionName, in ANeuralNetworksDevice_getExtensionSupport()
Dnnapi_util.cc56 ANeuralNetworksDevice* device = nullptr; in GetDeviceNamesList()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_device_selection_test.cc84 [](uint32_t devIndex, ANeuralNetworksDevice** device) -> int { in SetUp()
85 *device = reinterpret_cast<ANeuralNetworksDevice*>(devIndex + 1); in SetUp()
89 [](const ANeuralNetworksDevice* device, const char** name) -> int { in SetUp()
90 if (device == reinterpret_cast<ANeuralNetworksDevice*>(1)) { in SetUp()
92 } else if (device == reinterpret_cast<ANeuralNetworksDevice*>(2)) { in SetUp()
101 const ANeuralNetworksDevice* const* devices, uint32_t numDevices, in SetUp()
120 const ANeuralNetworksDevice* const* devices, uint32_t numDevices, in TEST_F()
137 const ANeuralNetworksDevice* const* devices, uint32_t numDevices, in TEST_F()
140 EXPECT_EQ(devices[0], reinterpret_cast<ANeuralNetworksDevice*>(1)); in TEST_F()
142 devices[0] != reinterpret_cast<ANeuralNetworksDevice*>(1)) { in TEST_F()
[all …]
Dnnapi_delegate_kernel.h319 std::vector<ANeuralNetworksDevice*> nnapi_devices_;
Dnnapi_delegate.cc481 ANeuralNetworksDevice** result, int* nnapi_errno) { in GetDeviceHandle()
489 ANeuralNetworksDevice* device = nullptr; in GetDeviceHandle()
545 const std::vector<ANeuralNetworksDevice*>& device_handles, in GetTargetSdkVersion()
611 std::vector<ANeuralNetworksDevice*>* result) { in GetTargetDevices()
621 ANeuralNetworksDevice* nnapi_device = nullptr; in GetTargetDevices()
631 ANeuralNetworksDevice* device = nullptr; in GetTargetDevices()
5069 std::vector<ANeuralNetworksDevice*> devices; in DoPrepare()
/external/tensorflow/tensorflow/lite/experimental/acceleration/configuration/
Dnnapi_plugin_test.cc67 const ANeuralNetworksDevice* const* devices, uint32_t numDevices, in SetUp()
186 const ANeuralNetworksDevice* const* devices, uint32_t numDevices, in TEST_F()
202 const ANeuralNetworksDevice* const* devices, uint32_t numDevices, in TEST_F()