/packages/modules/NeuralNetworks/common/types/operations/src/ |
D | Reshape.cpp | 35 std::vector<OperandType> inExpectedTypes; in validateDepthToSpace() local 39 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::INT32}; in validateDepthToSpace() 43 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::INT32}; in validateDepthToSpace() 47 inExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM, OperandType::INT32}; in validateDepthToSpace() 51 inExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM_SIGNED, OperandType::INT32}; in validateDepthToSpace() 58 inExpectedTypes.push_back(OperandType::BOOL); in validateDepthToSpace() 63 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validateDepthToSpace() 75 std::vector<OperandType> inExpectedTypes; in validateSpaceToDepth() local 79 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::INT32}; in validateSpaceToDepth() 83 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::INT32}; in validateSpaceToDepth() [all …]
|
D | QLSTM.cpp | 30 std::vector<OperandType> inExpectedTypes; in validate() local 32 inExpectedTypes.push_back(OperandType::TENSOR_QUANT8_ASYMM_SIGNED); in validate() 35 inExpectedTypes.push_back(OperandType::TENSOR_QUANT8_SYMM); in validate() 39 inExpectedTypes.push_back(OperandType::TENSOR_QUANT16_SYMM); in validate() 43 inExpectedTypes.push_back(OperandType::TENSOR_INT32); in validate() 46 inExpectedTypes.push_back(OperandType::TENSOR_QUANT8_SYMM); in validate() 47 inExpectedTypes.push_back(OperandType::TENSOR_INT32); in validate() 49 inExpectedTypes.push_back(OperandType::TENSOR_QUANT8_ASYMM_SIGNED); in validate() 51 inExpectedTypes.push_back(OperandType::TENSOR_QUANT16_SYMM); in validate() 54 inExpectedTypes.push_back(OperandType::TENSOR_QUANT16_SYMM); in validate() [all …]
|
D | GenerateProposals.cpp | 31 std::vector<OperandType> inExpectedTypes; in validate() local 35 inExpectedTypes = {inputType, inputType, OperandType::TENSOR_INT32, inputType}; in validate() 39 inExpectedTypes = {OperandType::TENSOR_QUANT16_ASYMM, deltaInputType, in validate() 47 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate() 59 std::vector<OperandType> inExpectedTypes; in validate() local 63 inExpectedTypes = { in validate() 70 inExpectedTypes = { in validate() 78 inExpectedTypes = {inputType, in validate() 92 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate() 108 std::vector<OperandType> inExpectedTypes; in validate() local [all …]
|
D | BidirectionalSequenceLSTM.cpp | 40 std::vector<OperandType> inExpectedTypes; in validate() local 46 inExpectedTypes = {}; in validate() 48 inExpectedTypes.push_back(inputType); in validate() 50 inExpectedTypes.push_back(OperandType::INT32); in validate() 51 inExpectedTypes.push_back(inputType == OperandType::TENSOR_FLOAT32 ? OperandType::FLOAT32 in validate() 53 inExpectedTypes.push_back(inputType == OperandType::TENSOR_FLOAT32 ? OperandType::FLOAT32 in validate() 55 inExpectedTypes.push_back(OperandType::BOOL); in validate() 56 inExpectedTypes.push_back(OperandType::BOOL); in validate() 58 inExpectedTypes.push_back(inputType); in validate() 67 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
|
D | DepthwiseConv2D.cpp | 33 std::vector<OperandType> inExpectedTypes; in validate() local 35 inExpectedTypes = { in validate() 42 inExpectedTypes = { in validate() 61 inExpectedTypes = { in validate() 89 inExpectedTypes.insert(inExpectedTypes.end(), explicitScalarTypes.begin(), in validate() 95 inExpectedTypes.push_back(OperandType::BOOL); in validate() 102 inExpectedTypes.push_back(OperandType::INT32); in validate() 103 inExpectedTypes.push_back(OperandType::INT32); in validate() 118 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
|
D | LSTM.cpp | 33 std::vector<OperandType> inExpectedTypes; in validate() local 41 inExpectedTypes = {inputType, inputType, inputType, inputType, inputType, inputType, in validate() 46 inExpectedTypes.push_back(OperandType::FLOAT32); in validate() 47 inExpectedTypes.push_back(OperandType::FLOAT32); in validate() 50 inExpectedTypes.push_back(OperandType::FLOAT16); in validate() 51 inExpectedTypes.push_back(OperandType::FLOAT16); in validate() 60 inExpectedTypes.push_back(inputType); in validate() 63 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
|
D | Conv2D.cpp | 42 std::vector<OperandType> inExpectedTypes; in validate() local 44 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate() 49 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate() 58 inExpectedTypes = {inputType, filterType, OperandType::TENSOR_INT32, in validate() 93 inExpectedTypes.insert(inExpectedTypes.end(), explicitScalarTypes.begin(), in validate() 99 inExpectedTypes.push_back(OperandType::BOOL); in validate() 106 inExpectedTypes.push_back(OperandType::INT32); in validate() 107 inExpectedTypes.push_back(OperandType::INT32); in validate() 122 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
|
D | Pooling.cpp | 32 std::vector<OperandType> inExpectedTypes; in validate() local 36 inExpectedTypes = { in validate() 42 inExpectedTypes = { in validate() 50 inExpectedTypes = { in validate() 62 inExpectedTypes = { in validate() 77 inExpectedTypes.insert(inExpectedTypes.end(), explicitScalarTypes.begin(), in validate() 81 inExpectedTypes.push_back(OperandType::BOOL); in validate() 86 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
|
D | GroupedConv2D.cpp | 34 std::vector<OperandType> inExpectedTypes; in validate() local 37 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate() 43 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate() 60 inExpectedTypes = {inputType, filterType, OperandType::TENSOR_INT32, in validate() 71 inExpectedTypes.insert(inExpectedTypes.end(), explicitScalarTypes.begin(), in validate() 74 inExpectedTypes.push_back(OperandType::BOOL); in validate() 81 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
|
D | Softmax.cpp | 32 std::vector<OperandType> inExpectedTypes; in validate() local 36 inExpectedTypes = {inputType, OperandType::FLOAT32}; in validate() 39 inExpectedTypes = {inputType, OperandType::FLOAT16}; in validate() 42 inExpectedTypes = {inputType, OperandType::FLOAT32}; in validate() 52 inExpectedTypes.push_back(OperandType::INT32); in validate() 58 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
|
D | TransposeConv2D.cpp | 32 std::vector<OperandType> inExpectedTypes; in validate() local 35 inExpectedTypes = {inputType, inputType, inputType}; in validate() 49 inExpectedTypes = {inputType, filterType, OperandType::TENSOR_INT32}; in validate() 66 inExpectedTypes.insert(inExpectedTypes.end(), argExpectedTypes.begin(), argExpectedTypes.end()); in validate() 67 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
|
D | HeatmapMaxKeypoint.cpp | 29 std::vector<OperandType> inExpectedTypes; in validate() local 34 inExpectedTypes = {inputType, inputType, OperandType::BOOL}; in validate() 37 inExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM, OperandType::TENSOR_QUANT16_ASYMM, in validate() 41 inExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM_SIGNED, in validate() 49 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
|
D | RoiPooling.cpp | 29 std::vector<OperandType> inExpectedTypes; in validate() local 32 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate() 37 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate() 43 inExpectedTypes = {inputType, in validate() 54 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
|
D | LocalResponseNormalization.cpp | 32 std::vector<OperandType> inExpectedTypes; in validate() local 37 inExpectedTypes = { in validate() 44 inExpectedTypes = { in validate() 54 inExpectedTypes.push_back(OperandType::INT32); in validate() 64 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
|
D | RoiAlign.cpp | 29 std::vector<OperandType> inExpectedTypes; in validate() local 32 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate() 38 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate() 45 inExpectedTypes = {inputType, in validate() 58 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
|
D | FullyConnected.cpp | 68 std::vector<OperandType> inExpectedTypes; in validate() local 73 inExpectedTypes = { in validate() 81 inExpectedTypes = { in validate() 102 inExpectedTypes = { in validate() 111 inExpectedTypes = { in validate() 120 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
|
D | InstanceNormalization.cpp | 29 std::vector<OperandType> inExpectedTypes; in validate() local 32 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::FLOAT32, OperandType::FLOAT32, in validate() 35 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::FLOAT16, OperandType::FLOAT16, in validate() 40 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
|
D | ResizeImageOps.cpp | 39 std::vector<OperandType> inExpectedTypes = {inputType, scalarType, scalarType}; in validate() local 66 inExpectedTypes.push_back(OperandType::BOOL); in validate() 69 while (inExpectedTypes.size() < numInputs) { in validate() 70 inExpectedTypes.push_back(OperandType::BOOL); in validate() 74 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
|
D | LogSoftmax.cpp | 30 std::vector<OperandType> inExpectedTypes; in validate() local 33 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::FLOAT32, OperandType::INT32}; in validate() 36 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::FLOAT16, OperandType::INT32}; in validate() 41 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
|
D | LSHProjection.cpp | 36 std::vector<OperandType> inExpectedTypes; in validate() local 39 inExpectedTypes = { in validate() 47 inExpectedTypes = { in validate() 58 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
|
D | RNN.cpp | 31 std::vector<OperandType> inExpectedTypes; in validate() local 35 inExpectedTypes = { in validate() 46 inExpectedTypes = { in validate() 59 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
|
D | Cast.cpp | 36 std::vector<OperandType> inExpectedTypes; in validate() local 44 inExpectedTypes = {inputType}; in validate() 52 inExpectedTypes = {inputType}; in validate() 69 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
|
D | UnidirectionalSequenceLSTM.cpp | 31 std::vector<OperandType> inExpectedTypes; in validate() local 34 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate() 50 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate() 75 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
|
/packages/modules/NeuralNetworks/common/ |
D | LegacyUtils.cpp | 512 const std::vector<OperandType>& inExpectedTypes, in validateOperationOperandTypes() argument 515 if (inOperandCount != static_cast<uint32_t>(inExpectedTypes.size()) || in validateOperationOperandTypes() 517 LOG(ERROR) << "Wrong operand count: expected " << inExpectedTypes.size() << " inputs and " in validateOperationOperandTypes() 523 if (operands[inOperandIndexes[i]].type != inExpectedTypes[i]) { in validateOperationOperandTypes() 525 << " for input " << i << ", expected " << inExpectedTypes[i]; in validateOperationOperandTypes() 739 std::vector<OperandType> inExpectedTypes; in validateOperation() local 743 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_INT32}; in validateOperation() 747 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_INT32}; in validateOperation() 751 inExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM, OperandType::TENSOR_INT32}; in validateOperation() 755 inExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM_SIGNED, in validateOperation() [all …]
|
/packages/modules/NeuralNetworks/common/types/src/ |
D | OperationsValidationUtils.cpp | 56 const std::vector<OperandType>& inExpectedTypes, in validateOperationOperandTypes() argument 58 NN_RET_CHECK_EQ(getNumInputs(), inExpectedTypes.size()) in validateOperationOperandTypes() 59 << "Wrong operand count: expected " << inExpectedTypes.size() << " inputs, got " in validateOperationOperandTypes() 65 NN_RET_CHECK_EQ(getInputType(i), inExpectedTypes[i]) in validateOperationOperandTypes() 67 << ", expected " << inExpectedTypes[i]; in validateOperationOperandTypes()
|