Searched refs:getInputOperand (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/runtime/ |
| D | ModelBuilder.h | 85 const Operand& getInputOperand(uint32_t i) const { return mOperands[getInputOperandIndex(i)]; } in getInputOperand() function
|
| D | ExecutionBuilder.cpp | 184 if (!checkDimensionInfo(mModel->getInputOperand(index), type, in setInput()
193 return mInputs[index].setFromPointer(mModel->getInputOperand(index), type, in setInput()
212 if (!checkDimensionInfo(mModel->getInputOperand(index), type, in setInputFromMemory()
226 return mInputs[index].setFromMemory(mModel->getInputOperand(index), type, poolIndex, offset, in setInputFromMemory()
|
| D | ExecutionBuilder.h | 208 return setInputOrOutputFromTemporaryMemory(mModel->getInputOperand(inputIndex), in setInputFromTemporaryMemory()
|
| /frameworks/ml/nn/common/ |
| D | Utils.cpp | 144 const Operand* getInputOperand(uint32_t index) const;
159 const Operand* OperationValidationContext::getInputOperand(uint32_t index) const { in getInputOperand() function in android::nn::__anon6bc5d5a50111::OperationValidationContext
178 return getInputOperand(index)->type; in getInputType()
182 const Operand* operand = getInputOperand(index); in getInputShape()
188 return getInputOperand(index)->extraParams; in getInputExtraParams()
|