Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/runtime/
DModelArgumentInfo.cpp33 static const std::pair<int, ModelArgumentInfo> kBadDataModelArgumentInfo{ANEURALNETWORKS_BAD_DATA,
36 std::pair<int, ModelArgumentInfo> ModelArgumentInfo::createFromPointer( in createFromPointer()
46 ModelArgumentInfo ret; in createFromPointer()
49 ret.mState = ModelArgumentInfo::HAS_NO_VALUE; in createFromPointer()
52 return {n, ModelArgumentInfo()}; in createFromPointer()
62 ret.mState = ModelArgumentInfo::POINTER; 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()
[all …]
DManager.cpp151 const std::vector<ModelArgumentInfo>& inputs,
152 const std::vector<ModelArgumentInfo>& outputs,
158 const std::vector<ModelArgumentInfo>& inputs,
159 const std::vector<ModelArgumentInfo>& outputs,
166 const std::vector<ModelArgumentInfo>& inputs,
167 const std::vector<ModelArgumentInfo>& outputs,
502 static Request createDriverRequest(const std::vector<ModelArgumentInfo>& inputs, in createDriverRequest()
503 const std::vector<ModelArgumentInfo>& outputs, in createDriverRequest()
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()
[all …]
DExecutionBuilder.cpp204 std::tie(n, mInputs[index]) = ModelArgumentInfo::createFromPointer( in setInput()
253 ModelArgumentInfo::createFromMemory(mModel->getInputOperand(index), type, poolIndex, in setInputFromMemory()
286 std::tie(n, mOutputs[index]) = ModelArgumentInfo::createFromPointer( in setOutput()
333 ModelArgumentInfo::createFromMemory(mModel->getOutputOperand(index), type, poolIndex, in setOutputFromMemory()
548 static int validateRequest(const std::vector<ModelArgumentInfo>& inputs, in validateRequest()
549 const std::vector<ModelArgumentInfo>& outputs) { in validateRequest()
551 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in validateRequest()
557 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in validateRequest()
576 if (mOutputs[i].state() != ModelArgumentInfo::HAS_NO_VALUE && in areOutputsFullySpecified()
1026 if (p.state() == ModelArgumentInfo::MEMORY) { in compute()
[all …]
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;
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,
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,
DAndroid.bp93 "ModelArgumentInfo.cpp",
214 "ModelArgumentInfo.cpp",
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()
DExecutionPlan.h810 const ModelArgumentInfo& info, const ExecutionBuilder* executionBuilder) const;