Searched refs:getInputOperand (Results 1 – 7 of 7) sorted by relevance
| /frameworks/ml/nn/runtime/ |
| D | ModelBuilder.h | 92 const hal::Operand& getInputOperand(uint32_t i) const { in getInputOperand() function
|
| D | ExecutionBuilder.cpp | 143 if (!checkDimensionInfo(mModel->getInputOperand(index), type, in setInput()
159 mModel->getInputOperand(index), type, const_cast<void*>(buffer), l); in setInput()
178 if (!checkDimensionInfo(mModel->getInputOperand(index), type, in setInputFromMemory()
204 mModel->getInputOperand(index), type, poolIndex, offset, length); in setInputFromMemory()
|
| D | ModelBuilder.cpp | 385 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()
|
| D | ExecutionBuilder.h | 258 return setInputOrOutputFromMemory(mModel->getInputOperand(inputIndex), memory, offset, in setInputFromMemory()
|
| D | Memory.cpp | 358 operand = model->getInputOperand(index); in addRole()
|
| /frameworks/ml/nn/common/ |
| D | ValidateHal.cpp | 469 auto getInputOperand = [&getSubgraph](const Operand& modelOperand, in validateOperations() local
492 .getSubgraphInputOperand = getInputOperand, in validateOperations()
|
| D | Utils.cpp | 221 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()
|