Home
last modified time | relevance | path

Searched refs:mTokenToBuffers (Results 1 – 6 of 6) sorted by relevance

/packages/modules/NeuralNetworks/shim_and_sl/
DShimBufferTracker.cpp42 token = mTokenToBuffers.size(); in add()
43 mTokenToBuffers.push_back(std::move(buffer)); in add()
47 mTokenToBuffers[token] = std::move(buffer); in add()
54 if (mTokenToBuffers.size() <= token || mTokenToBuffers[token] == nullptr) { in get()
58 return mTokenToBuffers[token]; in get()
63 CHECK_LT(token, mTokenToBuffers.size()); in free()
64 CHECK(mTokenToBuffers[token] != nullptr); in free()
65 mTokenToBuffers[token] = nullptr; in free()
/packages/modules/NeuralNetworks/common/
DBufferTracker.cpp195 mTokenToBuffers.reserve(kPreallocatedElements); in BufferTracker()
196 mTokenToBuffers.emplace_back(); in BufferTracker()
206 token = static_cast<Request::MemoryDomainToken>(mTokenToBuffers.size()); in add()
207 mTokenToBuffers.push_back(std::move(buffer)); in add()
212 mTokenToBuffers[index] = std::move(buffer); in add()
221 if (mTokenToBuffers.size() <= index || mTokenToBuffers[index] == nullptr) { in get()
225 return mTokenToBuffers[index]; in get()
231 CHECK_LT(index, mTokenToBuffers.size()); in free()
232 CHECK(mTokenToBuffers[index] != nullptr); in free()
234 mTokenToBuffers[index] = nullptr; in free()
DHalBufferTracker.cpp199 token = mTokenToBuffers.size(); in add()
200 mTokenToBuffers.push_back(std::move(buffer)); in add()
204 mTokenToBuffers[token] = std::move(buffer); in add()
212 if (mTokenToBuffers.size() <= token || mTokenToBuffers[token] == nullptr) { in get()
216 return mTokenToBuffers[token]; in get()
221 CHECK_LT(token, mTokenToBuffers.size()); in free()
222 CHECK(mTokenToBuffers[token] != nullptr); in free()
224 mTokenToBuffers[token] = nullptr; in free()
/packages/modules/NeuralNetworks/shim_and_sl/include/
DShimBufferTracker.h67 ShimBufferTracker() : mTokenToBuffers(1) {} in ShimBufferTracker()
81 std::vector<std::shared_ptr<::android::nn::sl_wrapper::Memory>> mTokenToBuffers
/packages/modules/NeuralNetworks/common/include/
DHalBufferTracker.h102 HalBufferTracker() : mTokenToBuffers(1) {} in HalBufferTracker()
116 std::vector<std::shared_ptr<HalManagedBuffer>> mTokenToBuffers; variable
DBufferTracker.h115 std::vector<std::shared_ptr<ManagedBuffer>> mTokenToBuffers GUARDED_BY(mMutex);