Home
last modified time | relevance | path

Searched refs:RunTimePoolInfo (Results 1 – 5 of 5) sorted by relevance

/frameworks/ml/nn/common/include/
DCpuExecutor.h104 class RunTimePoolInfo {
106 static std::optional<RunTimePoolInfo> createFromHidlMemory(const hidl_memory& hidlMemory);
107 static RunTimePoolInfo createFromExistingBuffer(uint8_t* buffer);
115 RunTimePoolInfo(const std::shared_ptr<const RunTimePoolInfoImpl>& impl);
120 bool setRunTimePoolInfosFromHidlMemories(std::vector<RunTimePoolInfo>* poolInfos,
144 const std::vector<RunTimePoolInfo>& modelPoolInfos,
145 const std::vector<RunTimePoolInfo>& requestPoolInfos);
153 bool initializeRunTimeInfo(const std::vector<RunTimePoolInfo>& modelPoolInfos,
154 const std::vector<RunTimePoolInfo>& requestPoolInfos);
/frameworks/ml/nn/driver/sample/
DSampleDriver.cpp229 const std::vector<RunTimePoolInfo>& poolInfos, in asyncExecute()
233 std::vector<RunTimePoolInfo> requestPoolInfos; in asyncExecute()
267 const std::vector<RunTimePoolInfo>& poolInfos, in executeBase()
321 std::vector<RunTimePoolInfo> requestPoolInfos; in executeSynchronously()
356 const std::vector<RunTimePoolInfo>& poolInfos) in BurstExecutorWithCache()
365 mMemoryCache[slot] = RunTimePoolInfo::createFromHidlMemory(memory); in addCacheEntry()
398 std::vector<RunTimePoolInfo> requestPoolInfos; in execute()
426 const std::vector<RunTimePoolInfo> mModelPoolInfos;
427 std::map<int32_t, std::optional<RunTimePoolInfo>> mMemoryCache; // cached requestPoolInfos
DSampleDriver.h102 std::vector<RunTimePoolInfo> mPoolInfos;
/frameworks/ml/nn/common/
DCpuExecutor.cpp252 class RunTimePoolInfo::RunTimePoolInfoImpl {
278 RunTimePoolInfo::RunTimePoolInfoImpl::RunTimePoolInfoImpl(const hidl_memory& hidlMemory, in RunTimePoolInfoImpl()
284 RunTimePoolInfo::RunTimePoolInfoImpl::~RunTimePoolInfoImpl() { in ~RunTimePoolInfoImpl()
307 bool RunTimePoolInfo::RunTimePoolInfoImpl::update() const { in update()
326 std::optional<RunTimePoolInfo> RunTimePoolInfo::createFromHidlMemory( in createFromHidlMemory()
381 return {RunTimePoolInfo(impl)}; in createFromHidlMemory()
384 RunTimePoolInfo RunTimePoolInfo::createFromExistingBuffer(uint8_t* buffer) { in createFromExistingBuffer()
390 RunTimePoolInfo::RunTimePoolInfo(const std::shared_ptr<const RunTimePoolInfoImpl>& impl) in RunTimePoolInfo() function in android::nn::RunTimePoolInfo
393 uint8_t* RunTimePoolInfo::getBuffer() const { in getBuffer()
397 bool RunTimePoolInfo::update() const { in update()
[all …]
/frameworks/ml/nn/runtime/
DExecutionBuilder.cpp996 const std::vector<RunTimePoolInfo>& modelPoolInfos, in computeOnCpu()
997 const std::vector<RunTimePoolInfo>& requestPoolInfos, in computeOnCpu()
1021 std::vector<RunTimePoolInfo> modelPoolInfos; in startComputeOnCpu()
1026 std::vector<RunTimePoolInfo> requestPoolInfos; in startComputeOnCpu()
1029 if (std::optional<RunTimePoolInfo> poolInfo = in startComputeOnCpu()
1030 RunTimePoolInfo::createFromHidlMemory(mem->getHidlMemory())) { in startComputeOnCpu()
1043 requestPoolInfos.emplace_back(RunTimePoolInfo::createFromExistingBuffer( in startComputeOnCpu()