Home
last modified time | relevance | path

Searched refs:ModelArgumentInfo (Results 1 – 9 of 9) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/runtime/
H A DModelArgumentInfo.cpp36 std::pair<int, ModelArgumentInfo> ModelArgumentInfo::createFromPointer( in createFromPointer()
46 ModelArgumentInfo ret; in createFromPointer()
52 return {n, ModelArgumentInfo()}; in createFromPointer()
79 std::pair<int, ModelArgumentInfo> ModelArgumentInfo::createFromMemory( in createFromMemory()
82 ModelArgumentInfo ret; in createFromMemory()
84 return {n, ModelArgumentInfo()}; in createFromMemory()
106 ret.mState = ModelArgumentInfo::MEMORY; in createFromMemory()
128 case ModelArgumentInfo::POINTER: { in createRequestArgument()
135 case ModelArgumentInfo::MEMORY: in createRequestArgument()
141 case ModelArgumentInfo::UNSPECIFIED: in createRequestArgument()
[all …]
H A DModelArgumentInfo.h36 class ModelArgumentInfo {
38 ModelArgumentInfo() {} in ModelArgumentInfo() function
40 static std::pair<int, ModelArgumentInfo> createFromPointer(
44 static std::pair<int, ModelArgumentInfo> createFromMemory(
142 const std::vector<ModelArgumentInfo>& argumentInfos,
H A DManager.h43 class ModelArgumentInfo; variable
79 const std::vector<ModelArgumentInfo>& inputs,
80 const std::vector<ModelArgumentInfo>& outputs,
89 const std::vector<ModelArgumentInfo>& inputs,
90 const std::vector<ModelArgumentInfo>& outputs,
98 const std::vector<ModelArgumentInfo>& inputs,
99 const std::vector<ModelArgumentInfo>& outputs,
H A DExecutionBuilder.h133 const ModelArgumentInfo& getInputInfo(uint32_t index) const { return mInputs[index]; } in getInputInfo()
134 const ModelArgumentInfo& getOutputInfo(uint32_t index) const { return mOutputs[index]; } in getOutputInfo()
186 std::vector<ModelArgumentInfo> mInputs;
187 std::vector<ModelArgumentInfo> mOutputs;
403 void mapInputOrOutput(const ModelArgumentInfo& builderInputOrOutput,
404 ModelArgumentInfo* executorInputOrOutput,
414 ModelArgumentInfo* inputOrOutputInfo);
447 std::vector<ModelArgumentInfo> mInputs;
448 std::vector<ModelArgumentInfo> mOutputs;
H A DManager.cpp151 const std::vector<ModelArgumentInfo>& inputs,
152 const std::vector<ModelArgumentInfo>& outputs,
158 const std::vector<ModelArgumentInfo>& inputs,
166 const std::vector<ModelArgumentInfo>& inputs,
525 const std::vector<ModelArgumentInfo>& inputs, const std::vector<ModelArgumentInfo>& outputs, in execute()
579 const std::vector<ModelArgumentInfo>& inputs, const std::vector<ModelArgumentInfo>& outputs, in executeFenced()
650 const std::vector<ModelArgumentInfo>& inputs, const std::vector<ModelArgumentInfo>& outputs, in createReusableExecution()
1046 const std::vector<ModelArgumentInfo>& inputs, const std::vector<ModelArgumentInfo>& outputs, in executeFenced()
1077 const std::vector<ModelArgumentInfo>& inputs, const std::vector<ModelArgumentInfo>& outputs, in createCpuRequest()
1122 const std::vector<ModelArgumentInfo>& inputs, const std::vector<ModelArgumentInfo>& outputs, in execute()
[all …]
H A DExecutionBuilder.cpp551 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in validateRequest()
557 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in validateRequest()
1026 if (p.state() == ModelArgumentInfo::MEMORY) { in compute()
1358 case ModelArgumentInfo::HAS_NO_VALUE: in mapInputOrOutput()
1359 case ModelArgumentInfo::UNSPECIFIED: in mapInputOrOutput()
1361 case ModelArgumentInfo::POINTER: in mapInputOrOutput()
1364 case ModelArgumentInfo::MEMORY: { in mapInputOrOutput()
1416 case ModelArgumentInfo::POINTER: in logArguments()
1420 case ModelArgumentInfo::MEMORY: in logArguments()
1426 case ModelArgumentInfo::HAS_NO_VALUE: in logArguments()
[all …]
H A DAndroid.bp93 "ModelArgumentInfo.cpp",
214 "ModelArgumentInfo.cpp",
H A DExecutionPlan.cpp1435 const ModelArgumentInfo& info, const ExecutionBuilder* executionBuilder) const { in getBufferFromModelArgumentInfo()
1437 case ModelArgumentInfo::POINTER: { in getBufferFromModelArgumentInfo()
1440 case ModelArgumentInfo::MEMORY: { in getBufferFromModelArgumentInfo()
1449 case ModelArgumentInfo::HAS_NO_VALUE: { in getBufferFromModelArgumentInfo()
1474 const ModelArgumentInfo& info = controller->mExecutionBuilder->getInputInfo(it->second); in getBuffer()
1478 const ModelArgumentInfo& info = controller->mExecutionBuilder->getOutputInfo(it->second); in getBuffer()
H A DExecutionPlan.h810 const ModelArgumentInfo& info, const ExecutionBuilder* executionBuilder) const;