Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/
DModelArgumentInfo.cpp35 static const std::pair<int, ModelArgumentInfo> kBadDataModelArgumentInfo{ANEURALNETWORKS_BAD_DATA,
38 std::pair<int, ModelArgumentInfo> ModelArgumentInfo::createFromPointer( in createFromPointer()
48 ModelArgumentInfo ret; in createFromPointer()
50 ret.mState = ModelArgumentInfo::HAS_NO_VALUE; in createFromPointer()
53 return {n, ModelArgumentInfo()}; in createFromPointer()
64 ret.mState = ModelArgumentInfo::POINTER; in createFromPointer()
71 std::pair<int, ModelArgumentInfo> ModelArgumentInfo::createFromMemory( in createFromMemory()
74 ModelArgumentInfo ret; in createFromMemory()
76 return {n, ModelArgumentInfo()}; in createFromMemory()
89 ret.mState = ModelArgumentInfo::MEMORY; in createFromMemory()
[all …]
DModelArgumentInfo.h36 class ModelArgumentInfo {
38 ModelArgumentInfo() {} in ModelArgumentInfo() function
40 static std::pair<int, ModelArgumentInfo> createFromPointer(
43 static std::pair<int, ModelArgumentInfo> createFromMemory(
112 const std::vector<ModelArgumentInfo>& argumentInfos,
DExecutionBuilder.h116 const ModelArgumentInfo& getInputInfo(uint32_t index) const { return mInputs[index]; } in getInputInfo()
117 const ModelArgumentInfo& getOutputInfo(uint32_t index) const { return mOutputs[index]; } in getOutputInfo()
160 std::vector<ModelArgumentInfo> mInputs;
161 std::vector<ModelArgumentInfo> mOutputs;
284 void mapInputOrOutput(const ModelArgumentInfo& builderInputOrOutput,
285 ModelArgumentInfo* executorInputOrOutput);
288 uint32_t offset, ModelArgumentInfo* inputOrOutputInfo);
316 std::vector<ModelArgumentInfo> mInputs;
317 std::vector<ModelArgumentInfo> mOutputs;
DExecutionBuilder.cpp156 std::tie(n, mInputs[index]) = ModelArgumentInfo::createFromPointer( in setInput()
201 std::tie(n, mInputs[index]) = ModelArgumentInfo::createFromMemory( in setInputFromMemory()
234 ModelArgumentInfo::createFromPointer(mModel->getOutputOperand(index), type, buffer, l); in setOutput()
277 std::tie(n, mOutputs[index]) = ModelArgumentInfo::createFromMemory( in setOutputFromMemory()
747 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in computeFenced()
754 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in computeFenced()
761 if (mOutputs[i].state() != ModelArgumentInfo::HAS_NO_VALUE && in computeFenced()
804 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in compute()
807 } else if (p.state() == ModelArgumentInfo::MEMORY) { in compute()
815 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in compute()
[all …]
DManager.cpp135 const std::vector<ModelArgumentInfo>& inputs,
136 const std::vector<ModelArgumentInfo>& outputs,
143 const std::vector<ModelArgumentInfo>& inputs,
144 const std::vector<ModelArgumentInfo>& outputs,
281 allocatePointerArgumentsToPool(const std::vector<ModelArgumentInfo>& args, in allocatePointerArgumentsToPool()
288 if (info.state() == ModelArgumentInfo::POINTER) { in allocatePointerArgumentsToPool()
323 const std::vector<ModelArgumentInfo>& inputs, const std::vector<ModelArgumentInfo>& outputs, in execute()
353 if (info.state() == ModelArgumentInfo::POINTER) { in execute()
417 if (info.state() == ModelArgumentInfo::POINTER) { in execute()
431 const std::vector<ModelArgumentInfo>& inputs, const std::vector<ModelArgumentInfo>& outputs, in executeFenced()
[all …]
DManager.h42 class ModelArgumentInfo; variable
58 const std::vector<ModelArgumentInfo>& inputs,
59 const std::vector<ModelArgumentInfo>& outputs,
69 const std::vector<ModelArgumentInfo>& inputs,
70 const std::vector<ModelArgumentInfo>& outputs,
DAndroid.bp57 "ModelArgumentInfo.cpp",
DExecutionPlan.cpp1022 const ModelArgumentInfo& info, const ExecutionBuilder* executionBuilder) const { in getBufferFromModelArgumentInfo()
1024 case ModelArgumentInfo::POINTER: { in getBufferFromModelArgumentInfo()
1027 case ModelArgumentInfo::MEMORY: { in getBufferFromModelArgumentInfo()
1036 case ModelArgumentInfo::HAS_NO_VALUE: { in getBufferFromModelArgumentInfo()
1060 const ModelArgumentInfo& info = controller->mExecutionBuilder->getInputInfo(it->second); in getBuffer()
1064 const ModelArgumentInfo& info = controller->mExecutionBuilder->getOutputInfo(it->second); in getBuffer()
DExecutionPlan.h608 const ModelArgumentInfo& info, const ExecutionBuilder* executionBuilder) const;