Searched defs:requestPoolInfos (Results 1 - 3 of 3) sorted by relevance

/frameworks/ml/nn/driver/sample/
H A DSampleDriver.cpp116 std::vector<RunTimePoolInfo> requestPoolInfos; local
117 if (!setRunTimePoolInfosFromHidlMemories(&requestPoolInfos, request.pools)) {
123 int n = executor.run(mModel, request, mPoolInfos, requestPoolInfos);
/frameworks/ml/nn/common/
H A DCpuExecutor.cpp191 const std::vector<RunTimePoolInfo>& requestPoolInfos) {
192 return run(convertToV1_1(model), request, modelPoolInfos, requestPoolInfos);
197 const std::vector<RunTimePoolInfo>& requestPoolInfos) {
205 initializeRunTimeInfo(modelPoolInfos, requestPoolInfos);
216 for (auto& runtimeInfo : requestPoolInfos) {
226 const std::vector<RunTimePoolInfo>& requestPoolInfos) {
272 auto updateForArguments = [this, &requestPoolInfos](const std::vector<uint32_t>& indexes,
292 nnAssert(poolIndex < requestPoolInfos.size());
293 auto& r = requestPoolInfos[poolIndex];
189 run(const V1_0::Model& model, const Request& request, const std::vector<RunTimePoolInfo>& modelPoolInfos, const std::vector<RunTimePoolInfo>& requestPoolInfos) argument
195 run(const V1_1::Model& model, const Request& request, const std::vector<RunTimePoolInfo>& modelPoolInfos, const std::vector<RunTimePoolInfo>& requestPoolInfos) argument
225 initializeRunTimeInfo(const std::vector<RunTimePoolInfo>& modelPoolInfos, const std::vector<RunTimePoolInfo>& requestPoolInfos) argument
/frameworks/ml/nn/runtime/
H A DExecutionBuilder.cpp683 const std::vector<RunTimePoolInfo>& requestPoolInfos,
686 int err = executor.run(model, request, modelPoolInfos, requestPoolInfos);
708 std::vector<RunTimePoolInfo> requestPoolInfos; local
709 requestPoolInfos.reserve(mMemories.size());
712 requestPoolInfos.emplace_back(mem->getHidlMemory(), &fail);
718 auto fixPointerArguments = [&requestPoolInfos](std::vector<ModelArgumentInfo>& argumentInfos) {
722 static_cast<uint32_t>(requestPoolInfos.size());
724 requestPoolInfos.emplace_back(static_cast<uint8_t*>(argumentInfo.buffer));
737 std::move(modelPoolInfos), std::move(requestPoolInfos),
681 asyncStartComputeOnCpu(const Model& model, const Request& request, const std::vector<RunTimePoolInfo>& modelPoolInfos, const std::vector<RunTimePoolInfo>& requestPoolInfos, const sp<IExecutionCallback>& executionCallback) argument

Completed in 140 milliseconds