Home
last modified time | relevance | path

Searched refs:bufferWrappers (Results 1 – 3 of 3) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/driver/sample/
H A DCanonicalPreparedModel.cpp42 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()
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()
236 const auto [requestPoolInfos, bufferWrappers] = in executeFenced()
[all …]
H A DSampleDriver.cpp385 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()
421 const std::vector<std::shared_ptr<HalManagedBuffer>>& bufferWrappers, in updateDeviceMemories() argument
437 bufferWrappers[poolIndex]->setInitialized(true); in updateDeviceMemories()
469 const auto [poolStatus, requestPoolInfos, bufferWrappers] = in asyncExecute()
497 updateDeviceMemories(executionStatus, request, bufferWrappers, outputShapes); in asyncExecute()
603 const auto [poolStatus, requestPoolInfos, bufferWrappers] = in executeSynchronouslyBase()
715 const auto [poolStatus, requestPoolInfos, bufferWrappers] = in executeFenced()
[all …]
/aosp12/packages/modules/NeuralNetworks/driver/sample_aidl/
H A DSampleDriver.cpp311 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()
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()
541 const auto [poolStatus, requestPoolInfos, bufferWrappers] = in executeFenced()
[all …]