Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/
DCpuExecutor.cpp84 const RunTimeOperandInfo* getOutputInfo(uint32_t index) const;
85 RunTimeOperandInfo* getOutputInfo(uint32_t index);
98 const RunTimeOperandInfo* OperationExecutionContext::getOutputInfo(uint32_t index) const { in getOutputInfo() function in android::nn::__anonec8350e20111::OperationExecutionContext
103 RunTimeOperandInfo* OperationExecutionContext::getOutputInfo(uint32_t index) { in getOutputInfo() function in android::nn::__anonec8350e20111::OperationExecutionContext
125 return getOutputInfo(index)->type; in getOutputType()
129 return getOutputInfo(index)->shape(); in getOutputShape()
133 return getOutputInfo(index)->buffer; in getOutputBuffer()
231 return setInfoAndAllocateIfNeeded(getOutputInfo(index), shape, &result); in setOutputShape()
239 return getOutputInfo(index)->lifetime == OperandLifeTime::NO_VALUE; in isOmittedOutput()
/frameworks/ml/nn/runtime/
DExecutionBuilder.h117 const ModelArgumentInfo& getOutputInfo(uint32_t index) const { return mOutputs[index]; } in getOutputInfo() function
DExecutionPlan.cpp1064 const ModelArgumentInfo& info = controller->mExecutionBuilder->getOutputInfo(it->second); in getBuffer()