Home
last modified time | relevance | path

Searched refs:outExpectedTypes (Results 1 – 25 of 28) sorted by relevance

12

/packages/modules/NeuralNetworks/common/types/operations/src/
DReshape.cpp36 std::vector<OperandType> outExpectedTypes; in validateDepthToSpace() local
40 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validateDepthToSpace()
44 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validateDepthToSpace()
48 outExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM}; in validateDepthToSpace()
52 outExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM_SIGNED}; in validateDepthToSpace()
63 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validateDepthToSpace()
76 std::vector<OperandType> outExpectedTypes; in validateSpaceToDepth() local
80 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validateSpaceToDepth()
84 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validateSpaceToDepth()
88 outExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM}; in validateSpaceToDepth()
[all …]
DGenerateProposals.cpp60 std::vector<OperandType> outExpectedTypes; in validate() local
67 outExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate()
74 outExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate()
87 outExpectedTypes = {inputType, OperandType::TENSOR_QUANT16_ASYMM, OperandType::TENSOR_INT32, in validate()
93 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
109 std::vector<OperandType> outExpectedTypes; in validate() local
123 outExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate()
137 outExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate()
152 outExpectedTypes = {inputType, OperandType::TENSOR_QUANT16_ASYMM, in validate()
158 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
[all …]
DHeatmapMaxKeypoint.cpp30 std::vector<OperandType> outExpectedTypes; in validate() local
35 outExpectedTypes = {inputType, inputType}; in validate()
39 outExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM, OperandType::TENSOR_QUANT16_ASYMM}; in validate()
43 outExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM_SIGNED, in validate()
50 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
DUnidirectionalSequenceLSTM.cpp32 std::vector<OperandType> outExpectedTypes; in validate() local
48 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validate()
64 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validate()
73 outExpectedTypes.insert(outExpectedTypes.end(), {inputType, inputType}); in validate()
76 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
DQLSTM.cpp65 std::vector<OperandType> outExpectedTypes; in validate() local
67 outExpectedTypes.push_back(OperandType::TENSOR_QUANT8_ASYMM_SIGNED); in validate()
69 outExpectedTypes.push_back(OperandType::TENSOR_QUANT16_SYMM); in validate()
71 outExpectedTypes.push_back(OperandType::TENSOR_QUANT8_ASYMM_SIGNED); in validate()
72 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
DGroupedConv2D.cpp35 std::vector<OperandType> outExpectedTypes; in validate() local
41 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validate()
47 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validate()
63 outExpectedTypes = {inputType}; in validate()
81 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
DLogSoftmax.cpp31 std::vector<OperandType> outExpectedTypes; in validate() local
34 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validate()
37 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validate()
42 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
DRNN.cpp32 std::vector<OperandType> outExpectedTypes; in validate() local
40 outExpectedTypes = { in validate()
51 outExpectedTypes = { in validate()
59 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
DCast.cpp37 std::vector<OperandType> outExpectedTypes; in validate() local
45 outExpectedTypes = {outputType}; in validate()
53 outExpectedTypes = {inputType}; // Only identity CAST is supported. in validate()
69 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
DPow.cpp31 std::vector<OperandType> outExpectedTypes; in validate() local
34 outExpectedTypes = {inputType}; in validate()
45 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
DArgMinMax.cpp29 std::vector<OperandType> outExpectedTypes; in validate() local
34 outExpectedTypes = {OperandType::TENSOR_INT32}; in validate()
39 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
DTile.cpp31 std::vector<OperandType> outExpectedTypes; in validate() local
36 outExpectedTypes = {inputType}; in validate()
47 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
DExpandDims.cpp31 std::vector<OperandType> outExpectedTypes; in validate() local
36 outExpectedTypes = {inputType}; in validate()
47 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
DMaximumMinimum.cpp30 std::vector<OperandType> outExpectedTypes; in validate() local
36 outExpectedTypes = {inputType}; in validate()
47 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
DLocalResponseNormalization.cpp33 std::vector<OperandType> outExpectedTypes; in validate() local
41 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validate()
48 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validate()
DLSTM.cpp34 std::vector<OperandType> outExpectedTypes; in validate() local
54 outExpectedTypes = {inputType, inputType, inputType, inputType}; in validate()
63 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
DMultinomial.cpp37 std::vector<OperandType> outExpectedTypes = {OperandType::TENSOR_INT32}; in validate() local
38 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
DHashtableLookup.cpp36 std::vector<OperandType> outExpectedTypes = {inputType, OperandType::TENSOR_QUANT8_ASYMM}; in validate() local
37 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
DSVDF.cpp43 std::vector<OperandType> outExpectedTypes = {inputType, inputType}; in validate() local
44 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
DQuantizedLSTM.cpp38 std::vector<OperandType> outExpectedTypes = {OperandType::TENSOR_QUANT16_SYMM, in validate() local
40 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
DSplit.cpp44 std::vector<OperandType> outExpectedTypes(context->getNumOutputs(), inputType); in validate() local
45 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
DSimpleMath.cpp46 std::vector<OperandType> outExpectedTypes = {inputType}; in validate() local
47 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
DEmbeddingLookup.cpp38 std::vector<OperandType> outExpectedTypes = {inputType}; in validate() local
48 NN_TRY(context->validateOperationOperandTypes(inExpectedTypes, outExpectedTypes)); in validate()
DBidirectionalSequenceRNN.cpp44 std::vector<OperandType> outExpectedTypes(numOutputs, inputType); in validate() local
45 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
/packages/modules/NeuralNetworks/common/
DLegacyUtils.cpp740 std::vector<OperandType> outExpectedTypes; in validateOperation() local
744 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validateOperation()
748 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validateOperation()
752 outExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM}; in validateOperation()
757 outExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM_SIGNED}; in validateOperation()
761 outExpectedTypes = {OperandType::TENSOR_INT32}; in validateOperation()
773 outExpectedTypes); in validateOperation()
784 std::vector<OperandType> outExpectedTypes; in validateOperation() local
788 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validateOperation()
792 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validateOperation()
[all …]

12