Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/
DUtils.cpp502 const std::vector<OperandType>& inExpectedTypes, in validateOperationOperandTypes() argument
505 if (inOperandCount != static_cast<uint32_t>(inExpectedTypes.size()) || in validateOperationOperandTypes()
507 LOG(ERROR) << "Wrong operand count: expected " << inExpectedTypes.size() << " inputs and " in validateOperationOperandTypes()
513 if (operands[inOperandIndexes[i]].type != inExpectedTypes[i]) { in validateOperationOperandTypes()
516 << " for input " << i << ", expected " << toString(inExpectedTypes[i]); in validateOperationOperandTypes()
581 std::vector<OperandType> inExpectedTypes; in validateOperation() local
585 inExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validateOperation()
589 inExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validateOperation()
598 inExpectedTypes, in validateOperation()
613 std::vector<OperandType> inExpectedTypes; in validateOperation() local
[all …]
/frameworks/ml/nn/common/operations/
DLogSoftmax.cpp76 std::vector<OperandType> inExpectedTypes; in validate() local
79 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::FLOAT32, OperandType::INT32}; in validate()
82 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::FLOAT16, OperandType::INT32}; in validate()
88 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
DInstanceNormalization.cpp95 std::vector<OperandType> inExpectedTypes; in validate() local
98 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::FLOAT32, OperandType::FLOAT32, in validate()
101 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::FLOAT16, OperandType::FLOAT16, in validate()
107 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
DPooling.cpp255 std::vector<OperandType> inExpectedTypes; in validate() local
258 inExpectedTypes = { in validate()
264 inExpectedTypes = { in validate()
272 inExpectedTypes = { in validate()
288 inExpectedTypes.insert(inExpectedTypes.end(), explicitScalarTypes.begin(), in validate()
292 inExpectedTypes.push_back(OperandType::BOOL); in validate()
297 return validateInputTypes(context, inExpectedTypes) && in validate()
DFullyConnected.cpp145 std::vector<OperandType> inExpectedTypes; in validate() local
149 inExpectedTypes = { in validate()
157 inExpectedTypes = { in validate()
178 inExpectedTypes = { in validate()
188 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
DSoftmax.cpp231 std::vector<OperandType> inExpectedTypes; in validate() local
234 inExpectedTypes = {inputType, OperandType::FLOAT32}; in validate()
237 inExpectedTypes = {inputType, OperandType::FLOAT16}; in validate()
243 inExpectedTypes.push_back(OperandType::INT32); in validate()
250 return validateInputTypes(context, inExpectedTypes) && in validate()
DConv2D.cpp405 std::vector<OperandType> inExpectedTypes; in validate() local
407 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate()
412 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate()
419 inExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM, in validate()
460 inExpectedTypes.insert(inExpectedTypes.end(), explicitScalarTypes.begin(), in validate()
466 inExpectedTypes.push_back(OperandType::BOOL); in validate()
473 inExpectedTypes.push_back(OperandType::INT32); in validate()
474 inExpectedTypes.push_back(OperandType::INT32); in validate()
485 return validateInputTypes(context, inExpectedTypes) && in validate()
DRoiPooling.cpp172 std::vector<OperandType> inExpectedTypes; in validate() local
175 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate()
180 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate()
185 inExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM, in validate()
197 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
DConcatenation.cpp112 std::vector<OperandType> inExpectedTypes(inputCount - 1, inputType); in validate() local
113 inExpectedTypes.push_back(OperandType::INT32); in validate()
123 return validateInputTypes(context, inExpectedTypes) && in validate()
DHeatmapMaxKeypoint.cpp205 std::vector<OperandType> inExpectedTypes; in validate() local
209 inExpectedTypes = {inputType, inputType, OperandType::BOOL}; in validate()
212 inExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM, OperandType::TENSOR_QUANT16_ASYMM, in validate()
219 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
DL2Normalization.cpp151 std::vector<OperandType> inExpectedTypes = {inputType}; in validate() local
160 inExpectedTypes.push_back(OperandType::INT32); in validate()
165 return validateInputTypes(context, inExpectedTypes) && in validate()
DResizeImageOps.cpp115 std::vector<OperandType> inExpectedTypes = {inputType, scalarType, scalarType}; in validate() local
134 inExpectedTypes.push_back(OperandType::BOOL); in validate()
139 return validateInputTypes(context, inExpectedTypes) && in validate()
DTransposeConv2D.cpp432 std::vector<OperandType> inExpectedTypes; in validate() local
434 inExpectedTypes = {inputType, inputType, inputType}; in validate()
445 inExpectedTypes = {inputType, filterType, OperandType::TENSOR_INT32}; in validate()
459 inExpectedTypes.insert(inExpectedTypes.end(), argExpectedTypes.begin(), argExpectedTypes.end()); in validate()
461 return validateInputTypes(context, inExpectedTypes) && in validate()
DGenerateProposals.cpp177 std::vector<OperandType> inExpectedTypes; in validate() local
180 inExpectedTypes = {inputType, inputType, OperandType::TENSOR_INT32, inputType}; in validate()
182 inExpectedTypes = {OperandType::TENSOR_QUANT16_ASYMM, OperandType::TENSOR_QUANT8_ASYMM, in validate()
188 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
635 std::vector<OperandType> inExpectedTypes; in validate() local
639 inExpectedTypes = { in validate()
646 inExpectedTypes = { in validate()
653 inExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM, in validate()
667 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
1111 std::vector<OperandType> inExpectedTypes; in validate() local
[all …]
DRoiAlign.cpp333 std::vector<OperandType> inExpectedTypes; in validate() local
336 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate()
342 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate()
348 inExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM, in validate()
362 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()
DUnidirectionalSequenceLSTM.cpp115 std::vector<OperandType> inExpectedTypes; in validate() local
118 inExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate()
134 inExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate()
154 NN_RET_CHECK(validateInputTypes(context, inExpectedTypes)); in validate()