Home
last modified time | relevance | path

Searched refs:getInputInfo (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/common/
DCpuExecutor.cpp83 const RunTimeOperandInfo* getInputInfo(uint32_t index) const;
93 const RunTimeOperandInfo* OperationExecutionContext::getInputInfo(uint32_t index) const { in getInputInfo() function in android::nn::__anonec8350e20111::OperationExecutionContext
109 return getInputInfo(index)->type; in getInputType()
113 return getInputInfo(index)->shape(); in getInputShape()
117 return getInputInfo(index)->buffer; in getInputBuffer()
121 return getInputInfo(index)->extraParams; in getInputExtraParams()
235 return getInputInfo(index)->lifetime == OperandLifeTime::NO_VALUE; in isOmittedInput()
257 for (uint32_t j = 0; j < getInputInfo(i)->dimensions.size(); j++) { in checkNoZeroSizedInput()
258 NN_RET_CHECK_NE(getInputInfo(i)->dimensions[j], 0) in checkNoZeroSizedInput()
/frameworks/ml/nn/runtime/
DExecutionBuilder.h116 const ModelArgumentInfo& getInputInfo(uint32_t index) const { return mInputs[index]; } in getInputInfo() function
DExecutionPlan.cpp1060 const ModelArgumentInfo& info = controller->mExecutionBuilder->getInputInfo(it->second); in getBuffer()