Lines Matching refs:outputCount
197 const uint32_t* inputIndexes, uint32_t outputCount, in OperationValidationContext() argument
203 outputCount(outputCount), in OperationValidationContext()
227 uint32_t outputCount; member in android::nn::__anon8d54713c0111::OperationValidationContext
247 CHECK(index < static_cast<uint32_t>(outputCount)); in getOutputOperand()
256 return outputCount; in getNumOutputs()
730 static bool validateIfOperation(uint32_t inputCount, const uint32_t* inputs, uint32_t outputCount, in validateIfOperation() argument
736 NN_RET_CHECK_GE(outputCount, 1u) << "ANEURALNETWORKS_IF must have at least 1 output"; in validateIfOperation()
743 NN_RET_CHECK_EQ(outputCount, branchModelOutputCount); in validateIfOperation()
775 uint32_t outputCount, const uint32_t* outputs, in validateWhileOperation() argument
789 NN_RET_CHECK_GE(outputCount, 1u) << "ANEURALNETWORKS_WHILE must have at least 1 output"; in validateWhileOperation()
814 NN_RET_CHECK_GE(bodyModelOutputCount, outputCount); in validateWhileOperation()
816 const uint32_t inputOutputCount = outputCount; in validateWhileOperation()
848 const uint32_t* inputIndexes, uint32_t outputCount, in validateOperation() argument
857 return validateOperation(opType, inputCount, inputIndexes, outputCount, outputIndexes, operands, in validateOperation()
862 const uint32_t* inputIndexes, uint32_t outputCount, in validateOperation() argument
868 NN_RETURN_IF_ERROR(validateOperandList(outputCount, outputIndexes, in validateOperation()
883 auto logInvalidInOutNumber = [opType, inputCount, outputCount](int expIn, int expOut) { in validateOperation()
885 << ") or output operands (" << outputCount << ", expected " << expOut in validateOperation()
894 if (inputCount != 2 || outputCount != 1) { in validateOperation()
930 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
934 if ((inputCount != 3 && inputCount != 2) || outputCount != 1) { in validateOperation()
936 << ", expected 3 or 2) or output operands (" << outputCount in validateOperation()
971 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
975 if ((inputCount != 3 && inputCount != 2) || outputCount != 1) { in validateOperation()
977 << ", expected 3 or 2) or output operands (" << outputCount in validateOperation()
1012 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1016 if (inputCount != 2 || outputCount != 1) { in validateOperation()
1042 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1046 if (inputCount != 3 || outputCount != 2) { in validateOperation()
1064 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1068 if (inputCount != 4 || outputCount != 1) { in validateOperation()
1106 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1115 (outputCount != kNumOutputs && outputCount != kNumOutputsMerged && in validateOperation()
1116 outputCount != kNumOutputsWithState && in validateOperation()
1117 outputCount != kNumOutputsMergedWithState)) { in validateOperation()
1119 << ", expected 61) or output operands (" << outputCount in validateOperation()
1151 if (outputCount == kNumOutputsWithState || outputCount == kNumOutputsMergedWithState) { in validateOperation()
1155 std::vector<OperandType> outExpectedTypes(outputCount, inputType); in validateOperation()
1157 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1162 if ((inputCount != 23 && inputCount != 27) || outputCount != 4) { in validateOperation()
1164 << ", expected 23 or 27) or output operands (" << outputCount in validateOperation()
1202 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1206 if (inputCount != 15 || outputCount != 2) { in validateOperation()
1223 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1227 if (inputCount != 3 || outputCount != 1) { in validateOperation()
1248 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1252 if (inputCount != 6 || outputCount != 2) { in validateOperation()
1287 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1291 if (inputCount != 7 || outputCount != 2) { in validateOperation()
1312 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1316 if ((inputCount != 3 && inputCount != 2) || outputCount != 1) { in validateOperation()
1318 << ", expected 3 or 2) or output operands (" << outputCount in validateOperation()
1363 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1367 if ((inputCount != 4 && inputCount != 3) || outputCount != 1) { in validateOperation()
1369 << ", expected 4 or 3) or output operands (" << outputCount in validateOperation()
1421 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1425 if (inputCount != 2 || outputCount != 1) { in validateOperation()
1476 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1480 if (inputCount != 3 || outputCount != 1) { in validateOperation()
1528 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1532 if (inputCount != 1 || outputCount != 1) { in validateOperation()
1579 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1583 if (inputCount != 3 || outputCount != 1) { in validateOperation()
1611 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1616 if (inputCount != 2 || outputCount != 1) { in validateOperation()
1637 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1641 if (inputCount != 2 || outputCount != 1) { in validateOperation()
1666 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1692 std::vector<OperandType> outExpectedTypes(outputCount, inputType); in validateOperation()
1694 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1699 if (inputCount != 2 || outputCount != 1) { in validateOperation()
1724 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1728 if ((inputCount != 12 && inputCount != 9) || outputCount != 1) { in validateOperation()
1730 << ", expected 12 or 9) or output operands (" << outputCount in validateOperation()
1789 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1793 if (inputCount != 2 || outputCount != 1) { in validateOperation()
1818 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1822 if (inputCount != 2 || outputCount != 1) { in validateOperation()
1844 inExpectedTypes, outputCount, outputIndexes, in validateOperation()
1849 return validateIfOperation(inputCount, inputIndexes, outputCount, outputIndexes, in validateOperation()
1856 return validateWhileOperation(inputCount, inputIndexes, outputCount, outputIndexes, in validateOperation()
1879 inputIndexes, outputCount, outputIndexes, in validateOperation()