Searched refs:getOutputOperand (Results 1 – 8 of 8) sorted by relevance
| /packages/modules/NeuralNetworks/runtime/ |
| D | ExecutionBuilder.cpp | 78 const Operand& outputOperand = model->getOutputOperand(outputIndex); in validateOutputShapesFromDriver()
272 if (!checkDimensionInfo(mModel->getOutputOperand(index), type, in setOutput()
288 mModel->getOutputOperand(index), type, buffer, l, mInputAndOutputPaddingEnabled); in setOutput()
309 if (!checkDimensionInfo(mModel->getOutputOperand(index), type, in setOutputFromMemory()
334 ModelArgumentInfo::createFromMemory(mModel->getOutputOperand(index), type, poolIndex, in setOutputFromMemory()
616 TypeManager::get()->isTensorType(mModel->getOutputOperand(i).type) && in areOutputsFullySpecified()
617 tensorHasUnspecifiedDimensions(mModel->getOutputOperand(i).type, in areOutputsFullySpecified()
1177 const OperandType operandType = mModel->getOutputOperand(i).type; in updateOutputShapes()
|
| D | ModelBuilder.h | 104 const Operand& getOutputOperand(uint32_t i) const { in getOutputOperand() function
|
| D | Telemetry.cpp | 116 result = evalDataClass(m->getOutputOperand(i).type, result); in evalOutputDataClass()
|
| D | ExecutionBuilder.h | 398 return setInputOrOutputFromMemory(mModel->getOutputOperand(outputIndex), memory, offset,
|
| D | Memory.cpp | 357 operand = model->getOutputOperand(index); in addRole()
|
| /packages/modules/NeuralNetworks/common/types/src/ |
| D | Validation.cpp | 1343 const Operand* getOutputOperand(uint32_t index) const;
1359 const Operand* OperationValidationContext::getOutputOperand(uint32_t index) const { in getOutputOperand() function in android::nn::__anon7d7df9830111::OperationValidationContext
1390 return getOutputOperand(index)->type; in getOutputType()
1394 const Operand* operand = getOutputOperand(index); in getOutputShape()
1563 const Operand& getOutputOperand(const std::vector<Model::Subgraph>& subgraphs, in getOutputOperand() function
1618 const Operand& innerOperand = getOutputOperand(subgraphs, branchModelOperand, i); in validateIfOperation()
1684 NN_TRY(validateConditionOperand(getOutputOperand(subgraphs, condModelOperand, 0))); in validateWhileOperation()
1712 const Operand& innerOperand = getOutputOperand(subgraphs, bodyModelOperand, i); in validateWhileOperation()
1720 const Operand& outputOperand = getOutputOperand(subgraphs, bodyModelOperand, i); in validateWhileOperation()
|
| /packages/modules/NeuralNetworks/common/ |
| D | LegacyUtils.cpp | 181 const Operand* getOutputOperand(uint32_t index) const;
200 const Operand* OperationValidationContext::getOutputOperand(uint32_t index) const { in getOutputOperand() function in android::nn::__anon0471dca30111::OperationValidationContext
228 return getOutputOperand(index)->type; in getOutputType()
232 const Operand* operand = getOutputOperand(index); in getOutputShape()
|
| D | ValidateHal.cpp | 479 auto getOutputOperand = [&getSubgraph](const Operand& modelOperand, in validateOperations() local
497 .getSubgraphOutputOperand = getOutputOperand, in validateOperations()
|