Home
last modified time | relevance | path

Searched refs:MemoryDomainToken (Results 1 – 12 of 12) sorted by relevance

/packages/modules/NeuralNetworks/common/include/
DBufferTracker.h85 Token(Request::MemoryDomainToken token, std::shared_ptr<BufferTracker> tracker) in Token()
88 Request::MemoryDomainToken get() const { return kToken; } in get()
91 const Request::MemoryDomainToken kToken;
103 std::shared_ptr<ManagedBuffer> get(Request::MemoryDomainToken token) const;
106 void free(Request::MemoryDomainToken token);
109 std::stack<Request::MemoryDomainToken, std::vector<Request::MemoryDomainToken>> mFreeTokens
/packages/modules/NeuralNetworks/common/
DBufferTracker.cpp63 CHECK(std::holds_alternative<Request::MemoryDomainToken>(request.pools[poolIndex])); in validateRequest()
190 using StackSpace = std::vector<Request::MemoryDomainToken>; in BufferTracker()
191 using Stack = std::stack<Request::MemoryDomainToken, StackSpace>; in BufferTracker()
204 auto token = Request::MemoryDomainToken{0}; in add()
206 token = static_cast<Request::MemoryDomainToken>(mTokenToBuffers.size()); in add()
218 std::shared_ptr<ManagedBuffer> BufferTracker::get(Request::MemoryDomainToken token) const { in get()
228 void BufferTracker::free(Request::MemoryDomainToken token) { in free()
DHalBufferTracker.cpp63 CHECK(std::holds_alternative<Request::MemoryDomainToken>(request.pools[poolIndex])); in validateRequest()
/packages/modules/NeuralNetworks/driver/sample/
DCanonicalPreparedModel.cpp55 } else if (const auto* maybeToken = std::get_if<Request::MemoryDomainToken>(&pool)) { in createRunTimePoolInfos()
80 if (std::holds_alternative<Request::MemoryDomainToken>(pool)) { in updateDeviceMemories()
90 if (std::holds_alternative<Request::MemoryDomainToken>(pool)) { in updateDeviceMemories()
102 if (std::holds_alternative<Request::MemoryDomainToken>(pool)) { in updateDeviceMemories()
263 if (std::holds_alternative<Request::MemoryDomainToken>(pool)) { in executeFenced()
DCanonicalBuffer.cpp66 Request::MemoryDomainToken Buffer::getToken() const { in getToken()
67 return Request::MemoryDomainToken{kToken->get()}; in getToken()
DCanonicalBuffer.h34 Request::MemoryDomainToken getToken() const override;
/packages/modules/NeuralNetworks/common/types/include/nnapi/
DIBuffer.h37 virtual Request::MemoryDomainToken getToken() const = 0;
DTypes.h918 enum class MemoryDomainToken : uint32_t {}; enum
923 using MemoryPool = std::variant<SharedMemory, MemoryDomainToken, SharedBuffer>;
/packages/modules/NeuralNetworks/driver/sample_aidl/
DSampleDriverAidl.cpp332 } else if (const auto* token = std::get_if<Request::MemoryDomainToken>(&pool)) { in createRunTimePoolInfos()
363 if (std::holds_alternative<Request::MemoryDomainToken>(pool)) { in updateDeviceMemories()
373 if (std::holds_alternative<Request::MemoryDomainToken>(pool)) { in updateDeviceMemories()
384 if (std::holds_alternative<Request::MemoryDomainToken>(pool)) { in updateDeviceMemories()
579 if (std::holds_alternative<Request::MemoryDomainToken>(pool)) { in executeFenced()
/packages/modules/NeuralNetworks/common/types/src/
DTypeUtils.cpp701 } else if (std::holds_alternative<Request::MemoryDomainToken>(memoryPool)) { in operator <<()
702 const auto& token = std::get<Request::MemoryDomainToken>(memoryPool); in operator <<()
703 if (token == Request::MemoryDomainToken{}) { in operator <<()
DValidation.cpp69 constexpr auto kInvalidMemoryDomainToken = Request::MemoryDomainToken{};
1068 if (std::holds_alternative<Request::MemoryDomainToken>(memoryPool)) { in validateRequestMemoryPool()
1069 NN_RET_CHECK(std::get<Request::MemoryDomainToken>(memoryPool) != kInvalidMemoryDomainToken); in validateRequestMemoryPool()
/packages/modules/NeuralNetworks/runtime/test/
DTestCompliance.cpp191 const Request::MemoryPool deviceMemoryPool = Request::MemoryDomainToken(1); in TEST_F()