Lines Matching refs:mNnApi

104     std::unique_ptr<const NnApiSupportLibrary> mNnApi;  member in android::nn::generated_tests::GeneratedTests
140 EXPECT_EQ(mNnApi->getFL5()->ANeuralNetworksModel_getSupportedOperationsForDevices( in checkSupported()
200 EXPECT_EQ(mNnApi->getFL5()->SL_ANeuralNetworksDevice_getNumberOfCacheFilesNeeded( in compileModel()
211 auto resultCompilation1 = Compilation::createForDevice(mNnApi.get(), &model, device); in compileModel()
224 auto resultCompilation2 = Compilation::createForDevice(mNnApi.get(), &model, device); in compileModel()
239 auto resultCompilation = Compilation::createForDevice(mNnApi.get(), &model, device); in compileModel()
266 EXPECT_EQ(mNnApi->getFL5()->ANeuralNetworksMemoryDesc_create(&desc), ANEURALNETWORKS_NO_ERROR); in createDeviceMemoryForInput()
267 EXPECT_EQ(mNnApi->getFL5()->ANeuralNetworksMemoryDesc_addInputRole( in createDeviceMemoryForInput()
270 EXPECT_EQ(mNnApi->getFL5()->ANeuralNetworksMemoryDesc_finish(desc), ANEURALNETWORKS_NO_ERROR); in createDeviceMemoryForInput()
272 mNnApi->getFL5()->ANeuralNetworksMemory_createFromDesc(desc, &memory); in createDeviceMemoryForInput()
273 mNnApi->getFL5()->ANeuralNetworksMemoryDesc_free(desc); in createDeviceMemoryForInput()
280 EXPECT_EQ(mNnApi->getFL5()->ANeuralNetworksMemoryDesc_create(&desc), ANEURALNETWORKS_NO_ERROR); in createDeviceMemoryForOutput()
281 EXPECT_EQ(mNnApi->getFL5()->ANeuralNetworksMemoryDesc_addOutputRole( in createDeviceMemoryForOutput()
284 EXPECT_EQ(mNnApi->getFL5()->ANeuralNetworksMemoryDesc_finish(desc), ANEURALNETWORKS_NO_ERROR); in createDeviceMemoryForOutput()
286 mNnApi->getFL5()->ANeuralNetworksMemory_createFromDesc(desc, &memory); in createDeviceMemoryForOutput()
287 mNnApi->getFL5()->ANeuralNetworksMemoryDesc_free(desc); in createDeviceMemoryForOutput()
318 auto& wrapperMemory = inputMemories.emplace_back(Memory(mNnApi.get(), memory)); in computeWithDeviceMemories()
321 auto ashmem = TestAshmem::createFrom(mNnApi.get(), operand.data); in computeWithDeviceMemories()
323 EXPECT_EQ(mNnApi->getFL5()->ANeuralNetworksMemory_copy(ashmem->get()->get(), memory), in computeWithDeviceMemories()
335 auto& wrapperMemory = outputMemories.emplace_back(Memory(mNnApi.get(), memory)); in computeWithDeviceMemories()
349 auto ashmem = TestAshmem::createFrom(mNnApi.get(), output); in computeWithDeviceMemories()
351 EXPECT_EQ(mNnApi->getFL5()->ANeuralNetworksMemory_copy(outputMemories[i].get(), in computeWithDeviceMemories()
364 Execution execution(mNnApi.get(), &compilation); in executeWithCompilation()
408 mNnApi->getFL5()->ANeuralNetworks_getDeviceCount(&numDevices); in executeOnce()
412 mNnApi->getFL5()->ANeuralNetworks_getDevice(i, &device); in executeOnce()
414 mNnApi->getFL5()->ANeuralNetworksDevice_getName(device, &deviceName); in executeOnce()
455 mNnApi->getFL5()->ANeuralNetworks_getDeviceCount(&numDevices); in executeMultithreadedSharedCompilation()
459 mNnApi->getFL5()->ANeuralNetworks_getDevice(i, &device); in executeMultithreadedSharedCompilation()
461 mNnApi->getFL5()->ANeuralNetworksDevice_getName(device, &deviceName); in executeMultithreadedSharedCompilation()
492 GeneratedModel model(mNnApi.get()); in execute()
493 createModel(mNnApi.get(), testModel, mTestDynamicOutputShape, &model); in execute()
542 mNnApi = loadNnApiSupportLibrary(SUPPORT_LIBRARY_NAME); in SetUp()
560 mNnApi.reset(nullptr); in TearDown()