Home
last modified time | relevance | path

Searched refs:nnapi_ (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/nnapi/
Dnnapi_handler.cc50 *nnapi_ = *NnApiPassthroughInstance(); in Reset()
55 nnapi_->android_sdk_version = version; in SetAndroidSdkVersion()
62 nnapi_->ANeuralNetworks_getDeviceCount = nullptr; in SetAndroidSdkVersion()
63 nnapi_->ANeuralNetworks_getDevice = nullptr; in SetAndroidSdkVersion()
64 nnapi_->ANeuralNetworksDevice_getName = nullptr; in SetAndroidSdkVersion()
65 nnapi_->ANeuralNetworksDevice_getVersion = nullptr; in SetAndroidSdkVersion()
66 nnapi_->ANeuralNetworksDevice_getFeatureLevel = nullptr; in SetAndroidSdkVersion()
67 nnapi_->ANeuralNetworksDevice_getType = nullptr; in SetAndroidSdkVersion()
68 nnapi_->ANeuralNetworksModel_getSupportedOperationsForDevices = nullptr; in SetAndroidSdkVersion()
69 nnapi_->ANeuralNetworksCompilation_createForDevices = nullptr; in SetAndroidSdkVersion()
[all …]
Dnnapi_handler.h49 nnapi_->ANeuralNetworks_getDeviceCount = [](uint32_t* numDevices) -> int { in GetDeviceCountReturns()
57 nnapi_->ANeuralNetworks_getDeviceCount = [](uint32_t* numDevices) -> int { in GetDeviceCountReturnsCount()
64 nnapi_->ANeuralNetworks_getDeviceCount = stub; in StubGetDeviceCountWith()
69 nnapi_->ANeuralNetworks_getDevice = in GetDeviceReturns()
78 nnapi_->ANeuralNetworks_getDevice = stub; in StubGetDeviceWith()
83 nnapi_->ANeuralNetworksDevice_getName = in GetDeviceNameReturns()
94 nnapi_->ANeuralNetworksDevice_getName = stub; in StubGetDeviceNameWith()
104 nnapi_->ANeuralNetworksModel_create = [](ANeuralNetworksModel** model) { in ModelCreateReturns()
111 nnapi_->ANeuralNetworksModel_create = stub; in StubModelCreateWith()
116 nnapi_->ANeuralNetworksModel_addOperand = in AddOperandReturns()
[all …]
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_mock_test.h43 nnapi_->nnapi_exists = true; in NnApiHandler()
44 nnapi_->android_sdk_version = android_sdk_version; in NnApiHandler()
46 nnapi_->ANeuralNetworksCompilation_free = in NnApiHandler()
48 nnapi_->ANeuralNetworksMemory_free = [](ANeuralNetworksMemory* memory) {}; in NnApiHandler()
49 nnapi_->ANeuralNetworksModel_free = [](ANeuralNetworksModel* model) {}; in NnApiHandler()
50 nnapi_->ANeuralNetworksExecution_free = in NnApiHandler()
52 nnapi_->ASharedMemory_create = [](const char* name, size_t size) -> int { in NnApiHandler()
55 nnapi_->ANeuralNetworksEvent_free = [](ANeuralNetworksEvent* event) {}; in NnApiHandler()
85 nnapi_ = *NnApiImplementation(); in SetUp()
86 nnapi_mock_ = absl::make_unique<NnApiMock>(&nnapi_); in SetUp()
[all …]
Dnnapi_delegate_kernel.h124 explicit NNFreeModel(const NnApi* nnapi) : nnapi_(nnapi) {} in NNFreeModel()
126 nnapi_->ANeuralNetworksModel_free(model); in operator()
131 const NnApi* nnapi_;
136 explicit NNFreeCompilation(const NnApi* nnapi) : nnapi_(nnapi) {} in NNFreeCompilation()
138 nnapi_->ANeuralNetworksCompilation_free(model); in operator()
143 const NnApi* nnapi_;
148 explicit NNFreeExecution(const NnApi* nnapi) : nnapi_(nnapi) {} in NNFreeExecution()
150 nnapi_->ANeuralNetworksExecution_free(execution); in operator()
155 const NnApi* nnapi_;
171 const NnApi* nnapi_;
[all …]
Dnnapi_delegate.cc656 nnapi_ = nnapi; in NNMemory()
658 fd_ = nnapi_->ASharedMemory_create(name, size); in NNMemory()
661 nnapi_->ANeuralNetworksMemory_createFromFd(size, PROT_READ | PROT_WRITE, in NNMemory()
668 : nnapi_(nullptr) {}
677 nnapi_->ANeuralNetworksMemory_free(nn_memory_handle_); in ~NNMemory()
719 : nnapi_(nnapi), in NNAPIOpBuilder()
991 nnapi_->ANeuralNetworksModel_addOperation( in AddOperationToModel()
1018 nnapi_->ANeuralNetworksModel_addOperand(nn_model_, &operand_type), in AddDequantize()
1065 nnapi_->ANeuralNetworksModel_addOperand(nn_model_, &operand_type), in AddSingleValueTensorAsScalarOperand()
1123 nnapi_->ANeuralNetworksModel_addOperand(nn_model_, &operand_type), in AddNewInputConstantTensor()
[all …]
/external/tensorflow/tensorflow/lite/experimental/acceleration/configuration/
Dnnapi_plugin_test.cc63 nnapi_ = const_cast<NnApi*>(NnApiImplementation()); in SetUp()
64 nnapi_mock_ = absl::make_unique<NnApiMock>(nnapi_); in SetUp()
65 nnapi_->ANeuralNetworksModel_getSupportedOperationsForDevices = in SetUp()
78 nnapi_->ANeuralNetworksCompilation_setPreference = in CheckExecutionPreference()
95 nnapi_->ANeuralNetworksCompilation_setPriority = in CheckExecutionPriority()
120 NnApi* nnapi_; member in tflite::__anonf33248460111::NNAPIPluginTest
153 nnapi_->android_sdk_version = in TEST_F()
166 nnapi_->ANeuralNetworksCompilation_setCaching = in TEST_F()
184 nnapi_->ANeuralNetworksModel_getSupportedOperationsForDevices = in TEST_F()
200 nnapi_->ANeuralNetworksModel_getSupportedOperationsForDevices = in TEST_F()