Home
last modified time | relevance | path

Searched refs:poolInfos (Results 1 – 4 of 4) sorted by relevance

/frameworks/ml/nn/driver/sample/
DSampleDriver.cpp452 const std::vector<RunTimePoolInfo>& poolInfos, in asyncExecute() argument
478 int n = executor.run(model, request, poolInfos, requestPoolInfos); in asyncExecute()
506 const std::vector<RunTimePoolInfo>& poolInfos, in executeBase() argument
532 std::thread([&model, &driver, preparedModel, &poolInfos, request, measure, driverStart, in executeBase()
534 asyncExecute(request, measure, driverStart, model, driver, preparedModel, poolInfos, in executeBase()
567 const std::vector<RunTimePoolInfo>& poolInfos, const OptionalTimePoint& halDeadline, in executeSynchronouslyBase() argument
603 int n = executor.run(model, request, poolInfos, requestPoolInfos); in executeSynchronouslyBase()
759 const std::vector<RunTimePoolInfo>& poolInfos) in BurstExecutorWithCache() argument
760 : mModel(model), mDriver(driver), mModelPoolInfos(poolInfos) {} in BurstExecutorWithCache()
/frameworks/ml/nn/common/include/
DCpuExecutor.h123 bool setRunTimePoolInfosFromHidlMemories(std::vector<RunTimePoolInfo>* poolInfos,
126 bool setRunTimePoolInfosFromMemoryPools(std::vector<RunTimePoolInfo>* poolInfos,
/frameworks/ml/nn/common/
DCpuExecutor.cpp449 bool setRunTimePoolInfosFromHidlMemories(std::vector<RunTimePoolInfo>* poolInfos, in setRunTimePoolInfosFromHidlMemories() argument
451 CHECK(poolInfos != nullptr); in setRunTimePoolInfosFromHidlMemories()
452 poolInfos->clear(); in setRunTimePoolInfosFromHidlMemories()
453 poolInfos->reserve(pools.size()); in setRunTimePoolInfosFromHidlMemories()
456 poolInfos->push_back(*poolInfo); in setRunTimePoolInfosFromHidlMemories()
459 poolInfos->clear(); in setRunTimePoolInfosFromHidlMemories()
466 bool setRunTimePoolInfosFromMemoryPools(std::vector<RunTimePoolInfo>* poolInfos, in setRunTimePoolInfosFromMemoryPools() argument
468 CHECK(poolInfos != nullptr); in setRunTimePoolInfosFromMemoryPools()
469 poolInfos->clear(); in setRunTimePoolInfosFromMemoryPools()
470 poolInfos->reserve(pools.size()); in setRunTimePoolInfosFromMemoryPools()
[all …]
/frameworks/ml/nn/runtime/
DManager.cpp628 CpuPreparedModel(Model model, std::vector<RunTimePoolInfo> poolInfos) in CpuPreparedModel() argument
629 : mModel(std::move(model)), mModelPoolInfos(std::move(poolInfos)) {} in CpuPreparedModel()
681 std::vector<RunTimePoolInfo> poolInfos; in create() local
682 if (!setRunTimePoolInfosFromHidlMemories(&poolInfos, hidlModel.pools)) { in create()
687 std::make_shared<CpuPreparedModel>(std::move(hidlModel), std::move(poolInfos)); in create()