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 …]
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,
DManager.h44 class ModelArgumentInfo; variable
80 const std::vector<ModelArgumentInfo>& inputs,
81 const std::vector<ModelArgumentInfo>& outputs,
91 const std::vector<ModelArgumentInfo>& inputs,
92 const std::vector<ModelArgumentInfo>& outputs,
101 const std::vector<ModelArgumentInfo>& inputs,
102 const std::vector<ModelArgumentInfo>& outputs,
DExecutionBuilder.h150 const ModelArgumentInfo& getInputInfo(uint32_t index) const { return mInputs[index]; } in getInputInfo()
151 const ModelArgumentInfo& getOutputInfo(uint32_t index) const { return mOutputs[index]; } in getOutputInfo()
205 std::vector<ModelArgumentInfo> mInputs;
206 std::vector<ModelArgumentInfo> mOutputs;
425 void mapInputOrOutput(const ModelArgumentInfo& builderInputOrOutput,
426 ModelArgumentInfo* executorInputOrOutput,
436 ModelArgumentInfo* inputOrOutputInfo);
469 std::vector<ModelArgumentInfo> mInputs;
470 std::vector<ModelArgumentInfo> mOutputs;
DManager.cpp194 const std::vector<ModelArgumentInfo>& inputs,
195 const std::vector<ModelArgumentInfo>& outputs,
202 const std::vector<ModelArgumentInfo>& inputs,
203 const std::vector<ModelArgumentInfo>& outputs,
211 const std::vector<ModelArgumentInfo>& inputs,
212 const std::vector<ModelArgumentInfo>& outputs,
557 static Request createDriverRequest(const std::vector<ModelArgumentInfo>& inputs, in createDriverRequest()
558 const std::vector<ModelArgumentInfo>& outputs, in createDriverRequest()
580 const std::vector<ModelArgumentInfo>& inputs, const std::vector<ModelArgumentInfo>& outputs, in execute()
636 const std::vector<ModelArgumentInfo>& inputs, const std::vector<ModelArgumentInfo>& outputs, in executeFenced()
[all …]
DExecutionBuilder.cpp205 std::tie(n, mInputs[index]) = ModelArgumentInfo::createFromPointer( in setInput()
254 ModelArgumentInfo::createFromMemory(mModel->getInputOperand(index), type, poolIndex, in setInputFromMemory()
287 std::tie(n, mOutputs[index]) = ModelArgumentInfo::createFromPointer( in setOutput()
334 ModelArgumentInfo::createFromMemory(mModel->getOutputOperand(index), type, poolIndex, in setOutputFromMemory()
587 static int validateRequest(const std::vector<ModelArgumentInfo>& inputs, in validateRequest()
588 const std::vector<ModelArgumentInfo>& outputs) { in validateRequest()
590 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in validateRequest()
596 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in validateRequest()
615 if (mOutputs[i].state() != ModelArgumentInfo::HAS_NO_VALUE && in areOutputsFullySpecified()
1070 if (p.state() == ModelArgumentInfo::MEMORY) { in compute()
[all …]
DAndroid.bp97 "ModelArgumentInfo.cpp",
293 "ModelArgumentInfo.cpp",
DExecutionPlan.cpp1487 const ModelArgumentInfo& info, const ExecutionBuilder* executionBuilder) const { in getBufferFromModelArgumentInfo()
1489 case ModelArgumentInfo::POINTER: { in getBufferFromModelArgumentInfo()
1492 case ModelArgumentInfo::MEMORY: { in getBufferFromModelArgumentInfo()
1501 case ModelArgumentInfo::HAS_NO_VALUE: { in getBufferFromModelArgumentInfo()
1526 const ModelArgumentInfo& info = controller->mExecutionBuilder->getInputInfo(it->second); in getBuffer()
1530 const ModelArgumentInfo& info = controller->mExecutionBuilder->getOutputInfo(it->second); in getBuffer()
DExecutionPlan.h810 const ModelArgumentInfo& info, const ExecutionBuilder* executionBuilder) const;