Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/
DCpuExecutor.cpp97 const RunTimeOperandInfo* getInputInfo(uint32_t index) const;
107 const RunTimeOperandInfo* OperationExecutionContext::getInputInfo(uint32_t index) const { in getInputInfo() function in android::nn::__anon47058d8f0111::OperationExecutionContext
123 return getInputInfo(index)->type; in getInputType()
127 return getInputInfo(index)->shape(); in getInputShape()
131 return getInputInfo(index)->buffer; in getInputBuffer()
135 return getInputInfo(index)->extraParams; in getInputExtraParams()
249 return getInputInfo(index)->lifetime == Operand::LifeTime::NO_VALUE; in isOmittedInput()
271 for (uint32_t j = 0; j < getInputInfo(i)->dimensions.size(); j++) { in checkNoZeroSizedInput()
272 NN_RET_CHECK_NE(getInputInfo(i)->dimensions[j], 0) in checkNoZeroSizedInput()
/packages/modules/NeuralNetworks/runtime/
DExecutionBuilder.h133 const ModelArgumentInfo& getInputInfo(uint32_t index) const { return mInputs[index]; } in getInputInfo() function
DExecutionPlan.cpp1474 const ModelArgumentInfo& info = controller->mExecutionBuilder->getInputInfo(it->second); in getBuffer()