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()
82 const std::vector<std::shared_ptr<ManagedBuffer>>& bufferWrappers, in updateDeviceMemories() argument
89 if (!bufferWrappers[poolIndex]->updateDimensions(outputShapes[i].dimensions)) { in updateDeviceMemories()
98 bufferWrappers[poolIndex]->setInitialized(true); in updateDeviceMemories()
154 const auto [requestPoolInfos, bufferWrappers] = in execute()
176 updateDeviceMemories(executionStatus, request, bufferWrappers, outputShapes); in execute()
[all …]
|
| D | SampleDriver.cpp | 385 std::vector<std::shared_ptr<HalManagedBuffer>> bufferWrappers; in createRunTimePoolInfos() local
387 bufferWrappers.reserve(request.pools.size()); in createRunTimePoolInfos()
399 bufferWrappers.push_back(nullptr); in createRunTimePoolInfos()
412 bufferWrappers.push_back(std::move(bufferWrapper)); in createRunTimePoolInfos()
416 return {V1_3::ErrorStatus::NONE, std::move(requestPoolInfos), std::move(bufferWrappers)}; in createRunTimePoolInfos()
421 const std::vector<std::shared_ptr<HalManagedBuffer>>& bufferWrappers, in updateDeviceMemories() argument
428 if (!bufferWrappers[poolIndex]->updateDimensions(outputShapes[i].dimensions)) { in updateDeviceMemories()
437 bufferWrappers[poolIndex]->setInitialized(true); in updateDeviceMemories()
469 const auto [poolStatus, requestPoolInfos, bufferWrappers] = in asyncExecute()
497 updateDeviceMemories(executionStatus, request, bufferWrappers, outputShapes); in asyncExecute()
[all …]
|
| /packages/modules/NeuralNetworks/driver/sample_aidl/ |
| D | SampleDriver.cpp | 311 std::vector<std::shared_ptr<AidlManagedBuffer>> bufferWrappers; in createRunTimePoolInfos() local
313 bufferWrappers.reserve(request.pools.size()); in createRunTimePoolInfos()
323 bufferWrappers.push_back(nullptr); in createRunTimePoolInfos()
337 bufferWrappers.push_back(std::move(bufferWrapper)); in createRunTimePoolInfos()
343 return {aidl_hal::ErrorStatus::NONE, std::move(requestPoolInfos), std::move(bufferWrappers)}; in createRunTimePoolInfos()
348 const std::vector<std::shared_ptr<AidlManagedBuffer>>& bufferWrappers, in updateDeviceMemories() argument
356 if (!bufferWrappers[poolIndex]->updateDimensions(unsignedDimensions)) { in updateDeviceMemories()
365 bufferWrappers[poolIndex]->setInitialized(true); in updateDeviceMemories()
426 const auto [poolStatus, requestPoolInfos, bufferWrappers] = in executeSynchronously()
454 updateDeviceMemories(executionStatus, request, bufferWrappers, outputShapes); in executeSynchronously()
[all …]
|