Searched refs:ManagedBuffer (Results 1 – 6 of 6) sorted by relevance
| /packages/modules/NeuralNetworks/common/ |
| D | BufferTracker.cpp | 35 std::shared_ptr<ManagedBuffer> ManagedBuffer::create(uint32_t size, in create()
46 return std::make_shared<ManagedBuffer>(std::move(buffer), size, std::move(roles), operand); in create()
49 ManagedBuffer::ManagedBuffer(std::unique_ptr<uint8_t[]> buffer, uint32_t size, in ManagedBuffer() function in android::nn::ManagedBuffer
60 ErrorStatus ManagedBuffer::validateRequest(uint32_t poolIndex, const Request& request, in validateRequest()
114 ErrorStatus ManagedBuffer::validateCopyFrom(const std::vector<uint32_t>& dimensions, in validateCopyFrom()
157 ErrorStatus ManagedBuffer::validateCopyTo(uint32_t size) const { in validateCopyTo()
171 bool ManagedBuffer::updateDimensions(const std::vector<uint32_t>& dimensions) { in updateDimensions()
183 void ManagedBuffer::setInitialized(bool initialized) { in setInitialized()
199 std::unique_ptr<BufferTracker::Token> BufferTracker::add(std::shared_ptr<ManagedBuffer> buffer) { in add()
218 std::shared_ptr<ManagedBuffer> BufferTracker::get(Request::MemoryDomainToken token) const { in get()
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | BufferTracker.h | 39 class ManagedBuffer {
41 static std::shared_ptr<ManagedBuffer> create(uint32_t size, std::set<PreparedModelRole> roles,
45 ManagedBuffer(std::unique_ptr<uint8_t[]> buffer, uint32_t size,
102 std::unique_ptr<Token> add(std::shared_ptr<ManagedBuffer> buffer);
103 std::shared_ptr<ManagedBuffer> get(Request::MemoryDomainToken token) const;
115 std::vector<std::shared_ptr<ManagedBuffer>> mTokenToBuffers GUARDED_BY(mMutex);
|
| /packages/modules/NeuralNetworks/driver/sample/ |
| D | CanonicalBuffer.h | 32 Buffer(std::shared_ptr<ManagedBuffer> buffer, std::unique_ptr<BufferTracker::Token> token);
41 const std::shared_ptr<ManagedBuffer> kBuffer;
|
| D | CanonicalBuffer.cpp | 40 const std::shared_ptr<ManagedBuffer>& bufferWrapper) { in copyFromInternal()
60 Buffer::Buffer(std::shared_ptr<ManagedBuffer> buffer, std::unique_ptr<BufferTracker::Token> token) in Buffer()
|
| D | CanonicalPreparedModel.cpp | 38 GeneralResult<std::pair<std::vector<RunTimePoolInfo>, std::vector<std::shared_ptr<ManagedBuffer>>>>
42 std::vector<std::shared_ptr<ManagedBuffer>> bufferWrappers; in createRunTimePoolInfos()
82 const std::vector<std::shared_ptr<ManagedBuffer>>& bufferWrappers, in updateDeviceMemories()
|
| D | CanonicalDevice.cpp | 241 auto bufferWrapper = ManagedBuffer::create(size, std::move(roles), operand); in allocate()
|