Home
last modified time | relevance | path

Searched refs:inExpectedTypes (Results 1 – 19 of 19) sorted by relevance

/frameworks/ml/nn/common/
DUtils.cpp654 const std::vector<OperandType>& inExpectedTypes, in validateOperationOperandTypes() argument
657 if (inOperandCount != static_cast<uint32_t>(inExpectedTypes.size()) || in validateOperationOperandTypes()
659 LOG(ERROR) << "Wrong operand count: expected " << inExpectedTypes.size() << " inputs and " in validateOperationOperandTypes()
665 if (operands[inOperandIndexes[i]].type != inExpectedTypes[i]) { in validateOperationOperandTypes()
668 << ", expected " << toString(inExpectedTypes[i]); in validateOperationOperandTypes()
899 std::vector<OperandType> inExpectedTypes; in validateOperation() local
903 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_INT32}; in validateOperation()
907 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_INT32}; in validateOperation()
911 inExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM, OperandType::TENSOR_INT32}; in validateOperation()
915 inExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM_SIGNED, in validateOperation()
[all …]
/frameworks/ml/nn/common/operations/
DLogSoftmax.cpp78 std::vector<OperandType> inExpectedTypes; in validate() local
81 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::FLOAT32, OperandType::INT32}; in validate()
84 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::FLOAT16, OperandType::INT32}; in validate()
90 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
DLocalResponseNormalization.cpp142 std::vector<OperandType> inExpectedTypes; in validate() local
146 inExpectedTypes = { in validate()
153 inExpectedTypes = { in validate()
163 inExpectedTypes.push_back(OperandType::INT32); in validate()
173 return validateInputTypes(context, inExpectedTypes) && in validate()
DInstanceNormalization.cpp108 std::vector<OperandType> inExpectedTypes; in validate() local
111 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::FLOAT32, OperandType::FLOAT32, in validate()
114 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::FLOAT16, OperandType::FLOAT16, in validate()
120 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
DPooling.cpp298 std::vector<OperandType> inExpectedTypes; in validate() local
301 inExpectedTypes = { in validate()
307 inExpectedTypes = { in validate()
315 inExpectedTypes = { in validate()
327 inExpectedTypes = { in validate()
343 inExpectedTypes.insert(inExpectedTypes.end(), explicitScalarTypes.begin(), in validate()
347 inExpectedTypes.push_back(OperandType::BOOL); in validate()
352 return validateInputTypes(context, inExpectedTypes) && in validate()
DQLSTM.cpp110 std::vector<OperandType> inExpectedTypes; in validate() local
112 inExpectedTypes.push_back(OperandType::TENSOR_QUANT8_ASYMM_SIGNED); in validate()
115 inExpectedTypes.push_back(OperandType::TENSOR_QUANT8_SYMM); in validate()
119 inExpectedTypes.push_back(OperandType::TENSOR_QUANT16_SYMM); in validate()
123 inExpectedTypes.push_back(OperandType::TENSOR_INT32); in validate()
126 inExpectedTypes.push_back(OperandType::TENSOR_QUANT8_SYMM); in validate()
127 inExpectedTypes.push_back(OperandType::TENSOR_INT32); in validate()
129 inExpectedTypes.push_back(OperandType::TENSOR_QUANT8_ASYMM_SIGNED); in validate()
131 inExpectedTypes.push_back(OperandType::TENSOR_QUANT16_SYMM); in validate()
134 inExpectedTypes.push_back(OperandType::TENSOR_QUANT16_SYMM); in validate()
[all …]
DFullyConnected.cpp227 std::vector<OperandType> inExpectedTypes; in validate() local
231 inExpectedTypes = { in validate()
239 inExpectedTypes = { in validate()
260 inExpectedTypes = { in validate()
269 inExpectedTypes = { in validate()
279 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
DSoftmax.cpp237 std::vector<OperandType> inExpectedTypes; in validate() local
240 inExpectedTypes = {inputType, OperandType::FLOAT32}; in validate()
243 inExpectedTypes = {inputType, OperandType::FLOAT16}; in validate()
246 inExpectedTypes = {inputType, OperandType::FLOAT32}; in validate()
256 inExpectedTypes.push_back(OperandType::INT32); in validate()
262 return validateInputTypes(context, inExpectedTypes) && in validate()
DResizeImageOps.cpp186 std::vector<OperandType> inExpectedTypes = {inputType, scalarType, scalarType}; in validate() local
212 inExpectedTypes.push_back(OperandType::BOOL); in validate()
215 while (inExpectedTypes.size() < numInputs) { in validate()
216 inExpectedTypes.push_back(OperandType::BOOL); in validate()
220 return validateInputTypes(context, inExpectedTypes) && in validate()
DDepthwiseConv2D.cpp425 std::vector<OperandType> inExpectedTypes; in validate() local
427 inExpectedTypes = { in validate()
434 inExpectedTypes = { in validate()
451 inExpectedTypes = { in validate()
479 inExpectedTypes.insert(inExpectedTypes.end(), explicitScalarTypes.begin(), in validate()
485 inExpectedTypes.push_back(OperandType::BOOL); in validate()
492 inExpectedTypes.push_back(OperandType::INT32); in validate()
493 inExpectedTypes.push_back(OperandType::INT32); in validate()
507 return validateInputTypes(context, inExpectedTypes) && in validate()
DRoiPooling.cpp193 std::vector<OperandType> inExpectedTypes; in validate() local
196 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate()
201 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate()
207 inExpectedTypes = {inputType, in validate()
219 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
DHeatmapMaxKeypoint.cpp233 std::vector<OperandType> inExpectedTypes; in validate() local
238 inExpectedTypes = {inputType, inputType, OperandType::BOOL}; in validate()
241 inExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM, OperandType::TENSOR_QUANT16_ASYMM, in validate()
245 inExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM_SIGNED, in validate()
254 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
DConv2D.cpp548 std::vector<OperandType> inExpectedTypes; in validate() local
550 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate()
555 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate()
564 inExpectedTypes = {inputType, filterType, OperandType::TENSOR_INT32, in validate()
597 inExpectedTypes.insert(inExpectedTypes.end(), explicitScalarTypes.begin(), in validate()
603 inExpectedTypes.push_back(OperandType::BOOL); in validate()
610 inExpectedTypes.push_back(OperandType::INT32); in validate()
611 inExpectedTypes.push_back(OperandType::INT32); in validate()
625 return validateInputTypes(context, inExpectedTypes) && in validate()
DConcatenation.cpp155 std::vector<OperandType> inExpectedTypes(inputCount - 1, inputType); in validate() local
156 inExpectedTypes.push_back(OperandType::INT32); in validate()
172 return validateInputTypes(context, inExpectedTypes) && in validate()
DTransposeConv2D.cpp445 std::vector<OperandType> inExpectedTypes; in validate() local
448 inExpectedTypes = {inputType, inputType, inputType}; in validate()
460 inExpectedTypes = {inputType, filterType, OperandType::TENSOR_INT32}; in validate()
477 inExpectedTypes.insert(inExpectedTypes.end(), argExpectedTypes.begin(), argExpectedTypes.end()); in validate()
479 return validateInputTypes(context, inExpectedTypes) && in validate()
DGenerateProposals.cpp206 std::vector<OperandType> inExpectedTypes; in validate() local
210 inExpectedTypes = {inputType, inputType, OperandType::TENSOR_INT32, inputType}; in validate()
214 inExpectedTypes = {OperandType::TENSOR_QUANT16_ASYMM, deltaInputType, in validate()
224 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
712 std::vector<OperandType> inExpectedTypes; in validate() local
716 inExpectedTypes = { in validate()
723 inExpectedTypes = { in validate()
731 inExpectedTypes = {inputType, in validate()
745 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
1222 std::vector<OperandType> inExpectedTypes; in validate() local
[all …]
DRoiAlign.cpp345 std::vector<OperandType> inExpectedTypes; in validate() local
348 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate()
354 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate()
361 inExpectedTypes = {inputType, in validate()
375 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
DL2Normalization.cpp208 std::vector<OperandType> inExpectedTypes = {inputType}; in validate() local
219 inExpectedTypes.push_back(OperandType::INT32); in validate()
228 return validateInputTypes(context, inExpectedTypes) && in validate()
DUnidirectionalSequenceLSTM.cpp123 std::vector<OperandType> inExpectedTypes; in validate() local
126 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate()
142 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate()
167 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()