Lines Matching refs:inputCount
181 OperationValidationContext(const char* operationName, uint32_t inputCount, in OperationValidationContext() argument
185 inputCount(inputCount), in OperationValidationContext()
207 uint32_t inputCount; member in android::nn::__anon648133630110::OperationValidationContext
219 CHECK(index < static_cast<uint32_t>(inputCount)); in getInputOperand()
229 return inputCount; in getNumInputs()
617 static bool validateIfOperation(uint32_t inputCount, const uint32_t* inputs, uint32_t outputCount, in validateIfOperation() argument
622 NN_RET_CHECK_GE(inputCount, 3u) << "ANEURALNETWORKS_IF must have at least 3 inputs"; in validateIfOperation()
629 NN_RET_CHECK_EQ(inputCount, op::kFirstInput + branchModelInputCount); in validateIfOperation()
660 static bool validateWhileOperation(uint32_t inputCount, const uint32_t* inputs, in validateWhileOperation() argument
674 NN_RET_CHECK_GE(inputCount, 3u) << "ANEURALNETWORKS_WHILE must have at least 3 inputs"; in validateWhileOperation()
681 NN_RET_CHECK_EQ(inputCount, op::kFirstInput + condModelInputCount); in validateWhileOperation()
699 NN_RET_CHECK_EQ(inputCount, op::kFirstInput + bodyModelInputCount); in validateWhileOperation()
733 int validateOperation(ANeuralNetworksOperationType opType, uint32_t inputCount, in validateOperation() argument
737 NN_RETURN_IF_ERROR(validateOperandList(inputCount, inputIndexes, in validateOperation()
755 auto logInvalidInOutNumber = [opType, inputCount, outputCount](int expIn, int expOut) { in validateOperation()
756 LOG(ERROR) << "Invalid number of input operands (" << inputCount << ", expected " << expIn in validateOperation()
766 if (inputCount != 2 || outputCount != 1) { in validateOperation()
799 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
804 if ((inputCount != 3 && inputCount != 2) || outputCount != 1) { in validateOperation()
805 LOG(ERROR) << "Invalid number of input operands (" << inputCount in validateOperation()
833 if (inputCount == 3) { in validateOperation()
839 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
844 if ((inputCount != 3 && inputCount != 2) || outputCount != 1) { in validateOperation()
845 LOG(ERROR) << "Invalid number of input operands (" << inputCount in validateOperation()
873 if (inputCount == 3) { in validateOperation()
879 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
884 if (inputCount != 2 || outputCount != 1) { in validateOperation()
908 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
913 if (inputCount != 3 || outputCount != 2) { in validateOperation()
929 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
934 if (inputCount != 4 || outputCount != 1) { in validateOperation()
969 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
978 if (inputCount != 61 || in validateOperation()
982 LOG(ERROR) << "Invalid number of input operands (" << inputCount in validateOperation()
1019 auto status = validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
1025 if ((inputCount != 23 && inputCount != 27) || outputCount != 4) { in validateOperation()
1026 LOG(ERROR) << "Invalid number of input operands (" << inputCount in validateOperation()
1055 if (inputCount == 23) { in validateOperation()
1063 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
1068 if (inputCount != 15 || outputCount != 2) { in validateOperation()
1084 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
1089 if (inputCount != 3 || outputCount != 1) { in validateOperation()
1108 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
1113 if (inputCount != 6 || outputCount != 2) { in validateOperation()
1146 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
1151 if (inputCount != 7 || outputCount != 2) { in validateOperation()
1170 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
1175 if ((inputCount != 3 && inputCount != 2) || outputCount != 1) { in validateOperation()
1176 LOG(ERROR) << "Invalid number of input operands (" << inputCount in validateOperation()
1214 if (inputCount == 3) { in validateOperation()
1220 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
1225 if ((inputCount != 4 && inputCount != 3) || outputCount != 1) { in validateOperation()
1226 LOG(ERROR) << "Invalid number of input operands (" << inputCount in validateOperation()
1271 if (inputCount == 4) { in validateOperation()
1277 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
1282 if (inputCount != 2 || outputCount != 1) { in validateOperation()
1330 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
1335 if (inputCount != 3 || outputCount != 1) { in validateOperation()
1380 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
1385 if (inputCount != 1 || outputCount != 1) { in validateOperation()
1431 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
1436 if (inputCount != 3 || outputCount != 1) { in validateOperation()
1461 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
1467 if (inputCount != 2 || outputCount != 1) { in validateOperation()
1486 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
1491 if (inputCount != 2 || outputCount != 1) { in validateOperation()
1514 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
1519 if (inputCount != 3) { in validateOperation()
1520 LOG(ERROR) << "Invalid number of input operands (" << inputCount << ", expected 3)" in validateOperation()
1541 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
1547 if (inputCount != 2 || outputCount != 1) { in validateOperation()
1570 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
1575 if ((inputCount != 12 && inputCount != 9) || outputCount != 1) { in validateOperation()
1576 LOG(ERROR) << "Invalid number of input operands (" << inputCount in validateOperation()
1624 if (inputCount == 12) { in validateOperation()
1635 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
1640 if (inputCount != 2 || outputCount != 1) { in validateOperation()
1663 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
1668 if (inputCount != 2 || outputCount != 1) { in validateOperation()
1688 return validateOperationOperandTypes(operands, inputCount, inputIndexes, in validateOperation()
1694 return validateIfOperation(inputCount, inputIndexes, outputCount, outputIndexes, in validateOperation()
1701 return validateWhileOperation(inputCount, inputIndexes, outputCount, outputIndexes, in validateOperation()
1723 OperationValidationContext context(operationRegistration->name, inputCount, in validateOperation()