Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/
DExecutionBuilder.cpp141 if (!checkDimensionInfo(mModel->getInputOperand(index), type, in setInput()
157 mModel->getInputOperand(index), type, const_cast<void*>(buffer), l); in setInput()
176 if (!checkDimensionInfo(mModel->getInputOperand(index), type, in setInputFromMemory()
202 mModel->getInputOperand(index), type, poolIndex, offset, length); in setInputFromMemory()
DModelBuilder.h92 const hal::Operand& getInputOperand(uint32_t i) const { in getInputOperand() function
DModelBuilder.cpp385 auto getInputOperand = [this](const Operand& modelOperand, uint32_t index) -> const Operand* { in addOperation() local
386 return &getReferencedModel(modelOperand)->getInputOperand(index); in addOperation()
396 .getSubgraphInputOperand = getInputOperand, in addOperation()
DExecutionBuilder.h258 return setInputOrOutputFromMemory(mModel->getInputOperand(inputIndex), memory, offset, in setInputFromMemory()
DMemory.cpp358 operand = model->getInputOperand(index); in addRole()
/frameworks/ml/nn/common/
DValidateHal.cpp469 auto getInputOperand = [&getSubgraph](const Operand& modelOperand, in validateOperations() local
491 .getSubgraphInputOperand = getInputOperand, in validateOperations()
DUtils.cpp221 const Operand* getInputOperand(uint32_t index) const;
241 const Operand* OperationValidationContext::getInputOperand(uint32_t index) const { in getInputOperand() function in android::nn::__anon8d54713c0111::OperationValidationContext
260 return getInputOperand(index)->type; in getInputType()
264 const Operand* operand = getInputOperand(index); in getInputShape()
270 return getInputOperand(index)->extraParams; in getInputExtraParams()