Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/driver/sample/
DCanonicalDevice.cpp186 std::vector<RunTimePoolInfo> poolInfos; in prepareModel() local
187 if (!setRunTimePoolInfosFromCanonicalMemories(&poolInfos, model.pools)) { in prepareModel()
193 kBufferTracker, std::move(poolInfos)); in prepareModel()
DCanonicalPreparedModel.cpp119 std::vector<RunTimePoolInfo> poolInfos) in PreparedModel() argument
125 kPoolInfos(std::move(poolInfos)) { in PreparedModel()
DCanonicalPreparedModel.h40 std::vector<RunTimePoolInfo> poolInfos);
/packages/modules/NeuralNetworks/driver/sample_hidl/
DSampleDriver.cpp452 const std::vector<RunTimePoolInfo>& poolInfos, const OptionalTimePoint& deadline, in asyncExecute() argument
477 int n = executor.run(uncheckedConvert(model), uncheckedConvert(request), poolInfos, in asyncExecute()
508 const std::vector<RunTimePoolInfo>& poolInfos, in executeBase() argument
534 std::thread([&model, &driver, preparedModel, &poolInfos, request, measure, driverStart, in executeBase()
536 asyncExecute(request, measure, driverStart, model, driver, preparedModel, poolInfos, in executeBase()
572 const std::vector<RunTimePoolInfo>& poolInfos, in executeSynchronouslyBase() argument
609 int n = executor.run(uncheckedConvert(model), uncheckedConvert(request), poolInfos, in executeSynchronouslyBase()
770 const std::vector<RunTimePoolInfo>& poolInfos) in BurstExecutorWithCache() argument
771 : mModel(model), mDriver(driver), mModelPoolInfos(poolInfos) {} in BurstExecutorWithCache()
DSampleDriverFloatXNNPACK.cpp1956 std::vector<RunTimePoolInfo> poolInfos; in getSupportedOperationsImpl() local
1957 setRunTimePoolInfosFromCanonicalMemories(&poolInfos, uncheckedConvert(model.pools)); in getSupportedOperationsImpl()
1958 auto operands = initializeRunTimeInfo(model.main, poolInfos, &model.operandValues); in getSupportedOperationsImpl()
/packages/modules/NeuralNetworks/common/
DCpuExecutor.cpp365 bool setRunTimePoolInfosFromCanonicalMemories(std::vector<RunTimePoolInfo>* poolInfos, in setRunTimePoolInfosFromCanonicalMemories() argument
367 CHECK(poolInfos != nullptr); in setRunTimePoolInfosFromCanonicalMemories()
368 poolInfos->clear(); in setRunTimePoolInfosFromCanonicalMemories()
369 poolInfos->reserve(pools.size()); in setRunTimePoolInfosFromCanonicalMemories()
372 poolInfos->push_back(*poolInfo); in setRunTimePoolInfosFromCanonicalMemories()
375 poolInfos->clear(); in setRunTimePoolInfosFromCanonicalMemories()
382 bool setRunTimePoolInfosFromMemoryPools(std::vector<RunTimePoolInfo>* poolInfos, in setRunTimePoolInfosFromMemoryPools() argument
384 CHECK(poolInfos != nullptr); in setRunTimePoolInfosFromMemoryPools()
385 poolInfos->clear(); in setRunTimePoolInfosFromMemoryPools()
386 poolInfos->reserve(pools.size()); in setRunTimePoolInfosFromMemoryPools()
[all …]
DLegacyHalUtils.cpp315 bool setRunTimePoolInfosFromHidlMemories(std::vector<RunTimePoolInfo>* poolInfos, in setRunTimePoolInfosFromHidlMemories() argument
317 return setRunTimePoolInfosFromCanonicalMemories(poolInfos, uncheckedConvert(pools)); in setRunTimePoolInfosFromHidlMemories()
/packages/modules/NeuralNetworks/common/include/
DCpuExecutor.h123 bool setRunTimePoolInfosFromCanonicalMemories(std::vector<RunTimePoolInfo>* poolInfos,
126 bool setRunTimePoolInfosFromMemoryPools(std::vector<RunTimePoolInfo>* poolInfos,
DLegacyHalUtils.h191 bool setRunTimePoolInfosFromHidlMemories(std::vector<RunTimePoolInfo>* poolInfos,
/packages/modules/NeuralNetworks/runtime/
DManager.cpp960 CpuPreparedModel(Model model, std::vector<RunTimePoolInfo> poolInfos) in CpuPreparedModel() argument
961 : mModel(std::move(model)), mModelPoolInfos(std::move(poolInfos)) {} in CpuPreparedModel()
1064 std::vector<RunTimePoolInfo> poolInfos; in create() local
1065 if (!setRunTimePoolInfosFromCanonicalMemories(&poolInfos, model.pools)) { in create()
1070 std::make_shared<CpuPreparedModel>(std::move(model), std::move(poolInfos)); in create()