Lines Matching refs:operand
47 static bool checkDimensionInfo(const Operand& operand, const ANeuralNetworksOperandType* newType, in checkDimensionInfo() argument
51 if (isExtensionOperandType(operand.type)) { in checkDimensionInfo()
52 NN_RET_CHECK(TypeManager::get()->getExtensionOperandTypeInfo(operand.type, &info)); in checkDimensionInfo()
59 if (operand.dimensions.size() == 0) { in checkDimensionInfo()
62 if (operand.dimensions.size() != newType->dimensionCount) { in checkDimensionInfo()
67 if (operand.dimensions[i] != newType->dimensions[i] && operand.dimensions[i] != 0) { in checkDimensionInfo()
73 if (!allowUnspecified && TypeManager::get()->isTensorType(operand.type) && in checkDimensionInfo()
74 tensorHasUnspecifiedDimensions(operand)) { in checkDimensionInfo()
82 int ModelArgumentInfo::setFromPointer(const Operand& operand, in setFromPointer() argument
94 NN_RETURN_IF_ERROR(updateDimensionInfo(operand, type)); in setFromPointer()
95 if (operand.type != OperandType::OEM) { in setFromPointer()
96 uint32_t neededLength = TypeManager::get()->getSizeOfData(operand.type, dimensions); in setFromPointer()
110 int ModelArgumentInfo::setFromMemory(const Operand& operand, const ANeuralNetworksOperandType* type, in setFromMemory() argument
112 NN_RETURN_IF_ERROR(updateDimensionInfo(operand, type)); in setFromMemory()
113 if (operand.type != OperandType::OEM) { in setFromMemory()
114 uint32_t neededLength = TypeManager::get()->getSizeOfData(operand.type, dimensions); in setFromMemory()
128 int ModelArgumentInfo::setFromTemporaryMemory(const Operand& operand, uint32_t poolIndex, in setFromTemporaryMemory() argument
130 NN_RETURN_IF_ERROR(updateDimensionInfo(operand, nullptr)); in setFromTemporaryMemory()
131 if (operand.type != OperandType::OEM) { in setFromTemporaryMemory()
132 uint32_t neededLength = TypeManager::get()->getSizeOfData(operand.type, dimensions); in setFromTemporaryMemory()
150 int ModelArgumentInfo::updateDimensionInfo(const Operand& operand, in updateDimensionInfo() argument
153 dimensions = operand.dimensions; in updateDimensionInfo()