Searched refs:bufferWrappers (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/driver/sample/ |
| D | CanonicalPreparedModel.cpp | 42 std::vector<std::shared_ptr<ManagedBuffer>> bufferWrappers; in createRunTimePoolInfos() local
44 bufferWrappers.reserve(request.pools.size()); in createRunTimePoolInfos()
54 bufferWrappers.push_back(nullptr); in createRunTimePoolInfos()
66 bufferWrappers.push_back(std::move(bufferWrapper)); in createRunTimePoolInfos()
69 return std::make_pair(std::move(requestPoolInfos), std::move(bufferWrappers)); in createRunTimePoolInfos()
73 const std::vector<std::shared_ptr<ManagedBuffer>>& bufferWrappers, in updateDeviceMemories() argument
81 if (!bufferWrappers[poolIndex]->updateDimensions(outputShapes[i].dimensions)) { in updateDeviceMemories()
91 bufferWrappers[poolIndex]->setInitialized(true); in updateDeviceMemories()
149 const auto [requestPoolInfos, bufferWrappers] = in execute()
171 updateDeviceMemories(executionStatus, request, bufferWrappers, outputShapes); in execute()
[all …]
|
| /packages/modules/NeuralNetworks/driver/sample_aidl/ |
| D | SampleDriverAidl.cpp | 319 std::vector<std::shared_ptr<AidlManagedBuffer>> bufferWrappers; in createRunTimePoolInfos() local
321 bufferWrappers.reserve(request.pools.size()); in createRunTimePoolInfos()
331 bufferWrappers.push_back(nullptr); in createRunTimePoolInfos()
346 bufferWrappers.push_back(std::move(bufferWrapper)); in createRunTimePoolInfos()
352 return {aidl_hal::ErrorStatus::NONE, std::move(requestPoolInfos), std::move(bufferWrappers)}; in createRunTimePoolInfos()
357 const std::vector<std::shared_ptr<AidlManagedBuffer>>& bufferWrappers, in updateDeviceMemories() argument
365 if (!bufferWrappers[poolIndex]->updateDimensions(unsignedDimensions)) { in updateDeviceMemories()
374 bufferWrappers[poolIndex]->setInitialized(true); in updateDeviceMemories()
435 const auto [poolStatus, requestPoolInfos, bufferWrappers] = in executeSynchronously()
463 updateDeviceMemories(executionStatus, request, bufferWrappers, outputShapes); in executeSynchronously()
[all …]
|
| /packages/modules/NeuralNetworks/driver/sample_hidl/ |
| D | SampleDriver.cpp | 374 std::vector<std::shared_ptr<HalManagedBuffer>> bufferWrappers; in createRunTimePoolInfos() local
376 bufferWrappers.reserve(request.pools.size()); in createRunTimePoolInfos()
388 bufferWrappers.push_back(nullptr); in createRunTimePoolInfos()
401 bufferWrappers.push_back(std::move(bufferWrapper)); in createRunTimePoolInfos()
405 return {V1_3::ErrorStatus::NONE, std::move(requestPoolInfos), std::move(bufferWrappers)}; in createRunTimePoolInfos()
410 const std::vector<std::shared_ptr<HalManagedBuffer>>& bufferWrappers, in updateDeviceMemories() argument
417 if (!bufferWrappers[poolIndex]->updateDimensions(outputShapes[i].dimensions)) { in updateDeviceMemories()
426 bufferWrappers[poolIndex]->setInitialized(true); in updateDeviceMemories()
458 const auto [poolStatus, requestPoolInfos, bufferWrappers] = in asyncExecute()
486 updateDeviceMemories(executionStatus, request, bufferWrappers, outputShapes); in asyncExecute()
[all …]
|