Searched refs:getOutputOperand (Results 1 – 7 of 7) sorted by relevance
| /frameworks/ml/nn/runtime/ |
| D | ExecutionBuilder.cpp | 218 if (!checkDimensionInfo(mModel->getOutputOperand(index), type, in setOutput()
234 ModelArgumentInfo::createFromPointer(mModel->getOutputOperand(index), type, buffer, l); in setOutput()
253 if (!checkDimensionInfo(mModel->getOutputOperand(index), type, in setOutputFromMemory()
278 mModel->getOutputOperand(index), type, poolIndex, offset, length); in setOutputFromMemory()
762 !checkDimensionInfo(mModel->getOutputOperand(i), nullptr, in computeFenced()
905 const OperandType operandType = mModel->getOutputOperand(i).type; in updateOutputShapes()
|
| D | ModelBuilder.h | 102 const hal::Operand& getOutputOperand(uint32_t i) const { in getOutputOperand() function
|
| D | ModelBuilder.cpp | 388 auto getOutputOperand = [this](const Operand& modelOperand, uint32_t index) -> const Operand* { in addOperation() local
389 return &getReferencedModel(modelOperand)->getOutputOperand(index); in addOperation()
397 .getSubgraphOutputOperand = getOutputOperand, in addOperation()
|
| D | ExecutionBuilder.h | 262 return setInputOrOutputFromMemory(mModel->getOutputOperand(outputIndex), memory, offset, in setOutputFromMemory()
|
| D | Memory.cpp | 364 operand = model->getOutputOperand(index); in addRole()
|
| /frameworks/ml/nn/common/ |
| D | ValidateHal.cpp | 475 auto getOutputOperand = [&getSubgraph](const Operand& modelOperand, in validateOperations() local
492 .getSubgraphOutputOperand = getOutputOperand, in validateOperations()
|
| D | Utils.cpp | 222 const Operand* getOutputOperand(uint32_t index) const;
246 const Operand* OperationValidationContext::getOutputOperand(uint32_t index) const { in getOutputOperand() function in android::nn::__anon8d54713c0111::OperationValidationContext
274 return getOutputOperand(index)->type; in getOutputType()
278 const Operand* operand = getOutputOperand(index); in getOutputShape()
|