Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/runtime/
H A DModelBuilder.h92 const Operand& getInputOperand(uint32_t i) const { in getInputOperand() function
H A DExecutionBuilder.cpp189 if (!checkDimensionInfo(mModel->getInputOperand(index), type, in setInput()
205 mModel->getInputOperand(index), type, const_cast<void*>(buffer), l, in setInput()
227 if (!checkDimensionInfo(mModel->getInputOperand(index), type, in setInputFromMemory()
253 ModelArgumentInfo::createFromMemory(mModel->getInputOperand(index), type, poolIndex, in setInputFromMemory()
H A DExecutionBuilder.h371 return setInputOrOutputFromMemory(mModel->getInputOperand(inputIndex), memory, offset,
H A DMemory.cpp343 operand = model->getInputOperand(index); in addRole()
/aosp12/packages/modules/NeuralNetworks/common/
H A DValidation.cpp1363 const Operand* getInputOperand(uint32_t index) const;
1376 const Operand* OperationValidationContext::getInputOperand(uint32_t index) const { in getInputOperand() function in android::nn::__anonef200d880110::OperationValidationContext
1397 return getInputOperand(index)->type; in getInputType()
1401 const Operand* operand = getInputOperand(index); in getInputShape()
1407 return getInputOperand(index)->extraParams; in getInputExtraParams()
1604 const Operand& getInputOperand(const std::vector<Model::Subgraph>& subgraphs, in getInputOperand() function
1659 const Operand& innerOperand = getInputOperand(subgraphs, branchModelOperand, i); in validateIfOperation()
1721 const Operand& innerOperand = getInputOperand(subgraphs, condModelOperand, i); in validateWhileOperation()
1748 const Operand& innerOperand = getInputOperand(subgraphs, bodyModelOperand, i); in validateWhileOperation()
1764 const Operand& inputOperand = getInputOperand(subgraphs, bodyModelOperand, i); in validateWhileOperation()
H A DLegacyUtils.cpp203 const Operand* getInputOperand(uint32_t index) const;
218 const Operand* OperationValidationContext::getInputOperand(uint32_t index) const { in getInputOperand() function in android::nn::__anon648133630110::OperationValidationContext
237 return getInputOperand(index)->type; in getInputType()
241 const Operand* operand = getInputOperand(index); in getInputShape()
247 return getInputOperand(index)->extraParams; in getInputExtraParams()
H A DValidateHal.cpp473 auto getInputOperand = [&getSubgraph](const Operand& modelOperand, in validateOperations() local
496 .getSubgraphInputOperand = getInputOperand, in validateOperations()