Searched refs:RunTimePoolInfo (Results 1 – 11 of 11) sorted by relevance
| /frameworks/ml/nn/common/include/ |
| D | CpuExecutor.h | 106 class RunTimePoolInfo {
108 static std::optional<RunTimePoolInfo> createFromHidlMemory(const hal::hidl_memory& hidlMemory);
109 static RunTimePoolInfo createFromExistingBuffer(uint8_t* buffer, uint32_t size = 0);
118 RunTimePoolInfo(const std::shared_ptr<const RunTimePoolInfoImpl>& impl);
123 bool setRunTimePoolInfosFromHidlMemories(std::vector<RunTimePoolInfo>* poolInfos,
126 bool setRunTimePoolInfosFromMemoryPools(std::vector<RunTimePoolInfo>* poolInfos,
150 const std::vector<RunTimePoolInfo>& modelPoolInfos,
151 const std::vector<RunTimePoolInfo>& requestPoolInfos);
168 const std::vector<RunTimePoolInfo>& requestPoolInfos,
183 const std::vector<RunTimePoolInfo>* mModelPoolInfos = nullptr;
|
| D | BufferTracker.h | 46 RunTimePoolInfo createRunTimePoolInfo() const { in createRunTimePoolInfo()
47 return RunTimePoolInfo::createFromExistingBuffer(kBuffer.get(), kSize); in createRunTimePoolInfo()
|
| /frameworks/ml/nn/runtime/ |
| D | Memory.h | 180 virtual std::optional<RunTimePoolInfo> getRunTimePoolInfo() const;
226 mutable std::optional<RunTimePoolInfo> mCachedRunTimePoolInfo;
290 std::optional<RunTimePoolInfo> getRunTimePoolInfo() const override { in getRunTimePoolInfo()
291 return RunTimePoolInfo::createFromExistingBuffer(getPointer(), kHidlMemory.size()); in getRunTimePoolInfo()
345 std::optional<RunTimePoolInfo> getRunTimePoolInfo() const override { in getRunTimePoolInfo()
346 return RunTimePoolInfo::createFromExistingBuffer(getPointer(), kHidlMemory.size()); in getRunTimePoolInfo()
|
| D | Manager.cpp | 628 CpuPreparedModel(Model model, std::vector<RunTimePoolInfo> poolInfos) in CpuPreparedModel()
633 const std::vector<RunTimePoolInfo> mModelPoolInfos;
681 std::vector<RunTimePoolInfo> poolInfos; in create()
693 const std::vector<RunTimePoolInfo>& modelPoolInfos, in computeOnCpu()
694 const std::vector<RunTimePoolInfo>& requestPoolInfos, in computeOnCpu()
762 std::vector<RunTimePoolInfo> requestPoolInfos; in execute()
765 if (std::optional<RunTimePoolInfo> poolInfo = mem->getRunTimePoolInfo()) { in execute()
781 requestPoolInfos.emplace_back(RunTimePoolInfo::createFromExistingBuffer( in execute()
|
| D | Memory.cpp | 214 std::optional<RunTimePoolInfo> Memory::getRunTimePoolInfo() const { in getRunTimePoolInfo()
217 mCachedRunTimePoolInfo = RunTimePoolInfo::createFromHidlMemory(kHidlMemory); in getRunTimePoolInfo()
232 static int copyHidlMemories(const std::optional<RunTimePoolInfo>& src, in copyHidlMemories()
233 const std::optional<RunTimePoolInfo>& dst) { in copyHidlMemories()
|
| D | ExecutionBuilder.h | 119 std::optional<RunTimePoolInfo> getRunTimePoolInfo(uint32_t poolIndex) const { in getRunTimePoolInfo()
|
| D | ExecutionPlan.h | 594 Buffer(RunTimePoolInfo info, uint32_t offset);
600 RunTimePoolInfo mInfo;
|
| D | ExecutionPlan.cpp | 1003 : mInfo(RunTimePoolInfo::createFromExistingBuffer(reinterpret_cast<uint8_t*>(pointer), size)), in Buffer()
1006 ExecutionPlan::Buffer::Buffer(RunTimePoolInfo info, uint32_t offset) in Buffer()
1028 if (std::optional<RunTimePoolInfo> poolInfo = in getBufferFromModelArgumentInfo()
1069 const std::optional<RunTimePoolInfo> info = location.memory->getRunTimePoolInfo(); in getBuffer()
|
| /frameworks/ml/nn/driver/sample/ |
| D | SampleDriver.cpp | 313 static void copyRunTimePoolInfos(const RunTimePoolInfo& srcPool, const RunTimePoolInfo& dstPool) { in copyRunTimePoolInfos()
322 const auto dstPool = RunTimePoolInfo::createFromHidlMemory(dst); in copyTo()
339 const auto srcPool = RunTimePoolInfo::createFromHidlMemory(src); in copyFromInternal()
370 static std::tuple<ErrorStatus, std::vector<RunTimePoolInfo>,
374 std::vector<RunTimePoolInfo> requestPoolInfos; in createRunTimePoolInfos()
382 auto buffer = RunTimePoolInfo::createFromHidlMemory(pool.hidlMemory()); in createRunTimePoolInfos()
452 const std::vector<RunTimePoolInfo>& poolInfos, in asyncExecute()
506 const std::vector<RunTimePoolInfo>& poolInfos, in executeBase()
567 const std::vector<RunTimePoolInfo>& poolInfos, const OptionalTimePoint& halDeadline, in executeSynchronouslyBase()
759 const std::vector<RunTimePoolInfo>& poolInfos) in BurstExecutorWithCache()
[all …]
|
| D | SampleDriver.h | 177 std::vector<RunTimePoolInfo> mPoolInfos;
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 274 class RunTimePoolInfo::RunTimePoolInfoImpl {
302 RunTimePoolInfo::RunTimePoolInfoImpl::RunTimePoolInfoImpl(const hidl_memory& hidlMemory, in RunTimePoolInfoImpl()
313 RunTimePoolInfo::RunTimePoolInfoImpl::~RunTimePoolInfoImpl() { in ~RunTimePoolInfoImpl()
340 bool RunTimePoolInfo::RunTimePoolInfoImpl::flush() const { in flush()
355 std::optional<RunTimePoolInfo> RunTimePoolInfo::createFromHidlMemory( in createFromHidlMemory()
421 return {RunTimePoolInfo(impl)}; in createFromHidlMemory()
424 RunTimePoolInfo RunTimePoolInfo::createFromExistingBuffer(uint8_t* buffer, uint32_t size) { in createFromExistingBuffer()
430 RunTimePoolInfo::RunTimePoolInfo(const std::shared_ptr<const RunTimePoolInfoImpl>& impl) in RunTimePoolInfo() function in android::nn::RunTimePoolInfo
433 uint8_t* RunTimePoolInfo::getBuffer() const { in getBuffer()
437 uint32_t RunTimePoolInfo::getSize() const { in getSize()
[all …]
|