Searched refs:ManagedBuffer (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/common/ |
| D | BufferTracker.cpp | 36 std::shared_ptr<ManagedBuffer> ManagedBuffer::create(uint32_t size, in create()
47 return std::make_shared<ManagedBuffer>(std::move(buffer), size, std::move(roles), operand); in create()
50 ManagedBuffer::ManagedBuffer(std::unique_ptr<uint8_t[]> buffer, uint32_t size, in ManagedBuffer() function in android::nn::ManagedBuffer
61 ErrorStatus ManagedBuffer::validateRequest(uint32_t poolIndex, const Request& request, in validateRequest()
116 ErrorStatus ManagedBuffer::validateCopyFrom(const std::vector<uint32_t>& dimensions, in validateCopyFrom()
159 ErrorStatus ManagedBuffer::validateCopyTo(uint32_t size) const { in validateCopyTo()
173 bool ManagedBuffer::updateDimensions(const std::vector<uint32_t>& dimensions) { in updateDimensions()
185 void ManagedBuffer::setInitialized(bool initialized) { in setInitialized()
190 std::unique_ptr<BufferTracker::Token> BufferTracker::add(std::shared_ptr<ManagedBuffer> buffer) { in add()
208 std::shared_ptr<ManagedBuffer> BufferTracker::get(uint32_t token) const { in get()
|
| /frameworks/ml/nn/common/include/ |
| D | BufferTracker.h | 37 class ManagedBuffer {
39 static std::shared_ptr<ManagedBuffer> create(uint32_t size, std::set<PreparedModelRole> roles,
43 ManagedBuffer(std::unique_ptr<uint8_t[]> buffer, uint32_t size,
100 std::unique_ptr<Token> add(std::shared_ptr<ManagedBuffer> buffer);
101 std::shared_ptr<ManagedBuffer> get(uint32_t token) const;
112 std::vector<std::shared_ptr<ManagedBuffer>> mTokenToBuffers;
|
| /frameworks/ml/nn/driver/sample/ |
| D | SampleDriver.h | 41 SampleBuffer(std::shared_ptr<ManagedBuffer> buffer, std::unique_ptr<BufferTracker::Token> token) in SampleBuffer()
51 const std::shared_ptr<ManagedBuffer> kBuffer;
|
| D | SampleDriver.cpp | 281 auto bufferWrapper = ManagedBuffer::create(size, std::move(roles), std::move(operand)); in allocate()
337 const std::shared_ptr<ManagedBuffer>& bufferWrapper) { in copyFromInternal()
371 std::vector<std::shared_ptr<ManagedBuffer>>>
375 std::vector<std::shared_ptr<ManagedBuffer>> bufferWrappers; in createRunTimePoolInfos()
410 const std::vector<std::shared_ptr<ManagedBuffer>>& bufferWrappers, in updateDeviceMemories()
|