Searched refs:requestPoolInfos (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/driver/sample/ |
| D | SampleDriver.cpp | 233 std::vector<RunTimePoolInfo> requestPoolInfos; in asyncExecute() local
234 if (!setRunTimePoolInfosFromHidlMemories(&requestPoolInfos, request.pools)) { in asyncExecute()
244 int n = executor.run(model, request, poolInfos, requestPoolInfos); in asyncExecute()
321 std::vector<RunTimePoolInfo> requestPoolInfos; in executeSynchronously() local
322 if (!setRunTimePoolInfosFromHidlMemories(&requestPoolInfos, request.pools)) { in executeSynchronously()
331 int n = executor.run(mModel, request, mPoolInfos, requestPoolInfos); in executeSynchronously()
398 std::vector<RunTimePoolInfo> requestPoolInfos; in execute() local
399 requestPoolInfos.reserve(slots.size()); in execute()
400 std::transform(slots.begin(), slots.end(), std::back_inserter(requestPoolInfos), in execute()
406 int n = executor.run(mModel, request, mModelPoolInfos, requestPoolInfos); in execute()
|
| /frameworks/ml/nn/common/include/ |
| D | CpuExecutor.h | 145 const std::vector<RunTimePoolInfo>& requestPoolInfos);
154 const std::vector<RunTimePoolInfo>& requestPoolInfos);
|
| /frameworks/ml/nn/runtime/ |
| D | ExecutionBuilder.cpp | 997 const std::vector<RunTimePoolInfo>& requestPoolInfos, in computeOnCpu() argument
1001 int err = executor.run(model, request, modelPoolInfos, requestPoolInfos); in computeOnCpu()
1026 std::vector<RunTimePoolInfo> requestPoolInfos; in startComputeOnCpu() local
1027 requestPoolInfos.reserve(mMemories.size()); in startComputeOnCpu()
1031 requestPoolInfos.emplace_back(*poolInfo); in startComputeOnCpu()
1037 auto fixPointerArguments = [&requestPoolInfos](std::vector<ModelArgumentInfo>& argumentInfos) { in startComputeOnCpu()
1041 static_cast<uint32_t>(requestPoolInfos.size()); in startComputeOnCpu()
1043 requestPoolInfos.emplace_back(RunTimePoolInfo::createFromExistingBuffer( in startComputeOnCpu()
1056 computeOnCpu(model, request, modelPoolInfos, requestPoolInfos, executionCallback); in startComputeOnCpu()
1060 std::move(requestPoolInfos), executionCallback); in startComputeOnCpu()
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 537 const std::vector<RunTimePoolInfo>& requestPoolInfos) { in run() argument
548 initializeRunTimeInfo(modelPoolInfos, requestPoolInfos); in run()
560 for (auto& runtimeInfo : requestPoolInfos) { in run()
569 const std::vector<RunTimePoolInfo>& requestPoolInfos) { in initializeRunTimeInfo() argument
616 auto updateForArguments = [this, &requestPoolInfos]( in initializeRunTimeInfo()
638 nnAssert(poolIndex < requestPoolInfos.size()); in initializeRunTimeInfo()
639 auto& r = requestPoolInfos[poolIndex]; in initializeRunTimeInfo()
|