Home
last modified time | relevance | path

Searched refs:ModelArgumentInfo (Results 1 – 8 of 8) 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.cpp158 std::tie(n, mInputs[index]) = ModelArgumentInfo::createFromPointer( in setInput()
203 std::tie(n, mInputs[index]) = ModelArgumentInfo::createFromMemory( in setInputFromMemory()
236 ModelArgumentInfo::createFromPointer(mModel->getOutputOperand(index), type, buffer, l); in setOutput()
279 std::tie(n, mOutputs[index]) = ModelArgumentInfo::createFromMemory( in setOutputFromMemory()
749 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in computeFenced()
756 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in computeFenced()
763 if (mOutputs[i].state() != ModelArgumentInfo::HAS_NO_VALUE && in computeFenced()
806 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in compute()
809 } else if (p.state() == ModelArgumentInfo::MEMORY) { in compute()
817 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in compute()
[all …]
DManager.cpp136 const std::vector<ModelArgumentInfo>& inputs,
137 const std::vector<ModelArgumentInfo>& outputs,
144 const std::vector<ModelArgumentInfo>& inputs,
145 const std::vector<ModelArgumentInfo>& outputs,
282 allocatePointerArgumentsToPool(const std::vector<ModelArgumentInfo>& args, in allocatePointerArgumentsToPool()
289 if (info.state() == ModelArgumentInfo::POINTER) { in allocatePointerArgumentsToPool()
324 const std::vector<ModelArgumentInfo>& inputs, const std::vector<ModelArgumentInfo>& outputs, in execute()
354 if (info.state() == ModelArgumentInfo::POINTER) { in execute()
418 if (info.state() == ModelArgumentInfo::POINTER) { in execute()
432 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.bp51 "ModelArgumentInfo.cpp",
DExecutionPlan.h608 const ModelArgumentInfo& info, const ExecutionBuilder* executionBuilder) const;