Searched refs:requestPoolInfos (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/driver/sample/ |
| D | SampleDriver.cpp | 374 std::vector<RunTimePoolInfo> requestPoolInfos; in createRunTimePoolInfos() local
376 requestPoolInfos.reserve(request.pools.size()); in createRunTimePoolInfos()
387 requestPoolInfos.push_back(std::move(*buffer)); in createRunTimePoolInfos()
400 requestPoolInfos.push_back(bufferWrapper->createRunTimePoolInfo()); in createRunTimePoolInfos()
405 return {ErrorStatus::NONE, std::move(requestPoolInfos), std::move(bufferWrappers)}; in createRunTimePoolInfos()
459 const auto [poolStatus, requestPoolInfos, bufferWrappers] = in asyncExecute()
478 int n = executor.run(model, request, poolInfos, requestPoolInfos); in asyncExecute()
586 const auto [poolStatus, requestPoolInfos, bufferWrappers] = in executeSynchronouslyBase()
603 int n = executor.run(model, request, poolInfos, requestPoolInfos); in executeSynchronouslyBase()
694 const auto [poolStatus, requestPoolInfos, bufferWrappers] = in executeFenced()
[all …]
|
| /frameworks/ml/nn/runtime/ |
| D | Manager.cpp | 694 const std::vector<RunTimePoolInfo>& requestPoolInfos, in computeOnCpu() argument
706 int err = executor.run(model, request, modelPoolInfos, requestPoolInfos); in computeOnCpu()
762 std::vector<RunTimePoolInfo> requestPoolInfos; in execute() local
763 requestPoolInfos.reserve(memories.size()); in execute()
766 requestPoolInfos.emplace_back(*poolInfo); in execute()
773 [&requestPoolInfos](const std::vector<ModelArgumentInfo>& argumentInfos) { in execute()
778 {.poolIndex = static_cast<uint32_t>(requestPoolInfos.size()), in execute()
781 requestPoolInfos.emplace_back(RunTimePoolInfo::createFromExistingBuffer( in execute()
799 std::thread([this, &request, &requestPoolInfos, &deadline, &loopTimeoutDuration, &result] { in execute()
800 result = computeOnCpu(mModel, request, mModelPoolInfos, requestPoolInfos, deadline, in execute()
[all …]
|
| /frameworks/ml/nn/common/include/ |
| D | CpuExecutor.h | 151 const std::vector<RunTimePoolInfo>& requestPoolInfos);
168 const std::vector<RunTimePoolInfo>& requestPoolInfos,
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 642 const std::vector<RunTimePoolInfo>& requestPoolInfos) { in run() argument
655 updateForArguments(model.main.inputIndexes, request.inputs, requestPoolInfos, operands.data()); in run()
656 updateForArguments(model.main.outputIndexes, request.outputs, requestPoolInfos, in run()
663 for (auto& runtimeInfo : requestPoolInfos) { in run()
742 const std::vector<RunTimePoolInfo>& requestPoolInfos, in updateForArguments() argument
763 CHECK_LT(poolIndex, requestPoolInfos.size()); in updateForArguments()
764 auto& r = requestPoolInfos[poolIndex]; in updateForArguments()
|