Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/
DCpuExecutor.cpp98 const RunTimeOperandInfo* getOutputInfo(uint32_t index) const;
99 RunTimeOperandInfo* getOutputInfo(uint32_t index);
112 const RunTimeOperandInfo* OperationExecutionContext::getOutputInfo(uint32_t index) const { in getOutputInfo() function in android::nn::__anon47058d8f0111::OperationExecutionContext
117 RunTimeOperandInfo* OperationExecutionContext::getOutputInfo(uint32_t index) { in getOutputInfo() function in android::nn::__anon47058d8f0111::OperationExecutionContext
139 return getOutputInfo(index)->type; in getOutputType()
143 return getOutputInfo(index)->shape(); in getOutputShape()
147 return getOutputInfo(index)->buffer; in getOutputBuffer()
245 return setInfoAndAllocateIfNeeded(getOutputInfo(index), shape, &result); in setOutputShape()
253 return getOutputInfo(index)->lifetime == Operand::LifeTime::NO_VALUE; in isOmittedOutput()
/packages/modules/NeuralNetworks/runtime/
DExecutionBuilder.h134 const ModelArgumentInfo& getOutputInfo(uint32_t index) const { return mOutputs[index]; } in getOutputInfo() function
DExecutionPlan.cpp1478 const ModelArgumentInfo& info = controller->mExecutionBuilder->getOutputInfo(it->second); in getBuffer()