Home
last modified time | relevance | path

Searched refs:getOutputOperand (Results 1 – 7 of 7) sorted by relevance

/packages/modules/NeuralNetworks/runtime/
DExecutionBuilder.cpp77 const Operand& outputOperand = model->getOutputOperand(outputIndex); in validateOutputShapesFromDriver()
271 if (!checkDimensionInfo(mModel->getOutputOperand(index), type, in setOutput()
287 mModel->getOutputOperand(index), type, buffer, l, mInputAndOutputPaddingEnabled); in setOutput()
308 if (!checkDimensionInfo(mModel->getOutputOperand(index), type, in setOutputFromMemory()
333 ModelArgumentInfo::createFromMemory(mModel->getOutputOperand(index), type, poolIndex, in setOutputFromMemory()
577 TypeManager::get()->isTensorType(mModel->getOutputOperand(i).type) && in areOutputsFullySpecified()
578 tensorHasUnspecifiedDimensions(mModel->getOutputOperand(i).type, in areOutputsFullySpecified()
1132 const OperandType operandType = mModel->getOutputOperand(i).type; in updateOutputShapes()
DModelBuilder.h102 const Operand& getOutputOperand(uint32_t i) const { in getOutputOperand() function
DExecutionBuilder.h376 return setInputOrOutputFromMemory(mModel->getOutputOperand(outputIndex), memory, offset,
DMemory.cpp349 operand = model->getOutputOperand(index); in addRole()
/packages/modules/NeuralNetworks/common/
DValidation.cpp1364 const Operand* getOutputOperand(uint32_t index) const;
1380 const Operand* OperationValidationContext::getOutputOperand(uint32_t index) const { in getOutputOperand() function in android::nn::__anonb795b7630111::OperationValidationContext
1411 return getOutputOperand(index)->type; in getOutputType()
1415 const Operand* operand = getOutputOperand(index); in getOutputShape()
1609 const Operand& getOutputOperand(const std::vector<Model::Subgraph>& subgraphs, in getOutputOperand() function
1664 const Operand& innerOperand = getOutputOperand(subgraphs, branchModelOperand, i); in validateIfOperation()
1729 NN_TRY(validateConditionOperand(getOutputOperand(subgraphs, condModelOperand, 0))); in validateWhileOperation()
1757 const Operand& innerOperand = getOutputOperand(subgraphs, bodyModelOperand, i); in validateWhileOperation()
1765 const Operand& outputOperand = getOutputOperand(subgraphs, bodyModelOperand, i); in validateWhileOperation()
DLegacyUtils.cpp204 const Operand* getOutputOperand(uint32_t index) const;
223 const Operand* OperationValidationContext::getOutputOperand(uint32_t index) const { in getOutputOperand() function in android::nn::__anon3bac189e0111::OperationValidationContext
251 return getOutputOperand(index)->type; in getOutputType()
255 const Operand* operand = getOutputOperand(index); in getOutputShape()
DValidateHal.cpp479 auto getOutputOperand = [&getSubgraph](const Operand& modelOperand, in validateOperations() local
497 .getSubgraphOutputOperand = getOutputOperand, in validateOperations()