Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/
H A DLegacyUtils.cpp772 std::vector<OperandType> outExpectedTypes; in validateOperation() local
1127 outExpectedTypes = { in validateOperation()
1138 outExpectedTypes = { in validateOperation()
1320 outExpectedTypes = {inputType}; in validateOperation()
1370 outExpectedTypes = {inputType}; in validateOperation()
1404 outExpectedTypes = {outputType}; in validateOperation()
1504 outExpectedTypes = {inputType}; in validateOperation()
1560 outExpectedTypes = {inputType}; in validateOperation()
1618 outExpectedTypes = {inputType}; in validateOperation()
1653 outExpectedTypes = {inputType}; in validateOperation()
[all …]
H A DValidation.cpp1821 std::vector<OperandType> outExpectedTypes; in validateOperationButNotOperandsImpl() local
2141 outExpectedTypes = { in validateOperationButNotOperandsImpl()
2152 outExpectedTypes = { in validateOperationButNotOperandsImpl()
2325 outExpectedTypes = {inputType}; in validateOperationButNotOperandsImpl()
2371 outExpectedTypes = {inputType}; in validateOperationButNotOperandsImpl()
2402 outExpectedTypes = {outputType}; in validateOperationButNotOperandsImpl()
2487 outExpectedTypes = {inputType}; in validateOperationButNotOperandsImpl()
2538 outExpectedTypes = {inputType}; in validateOperationButNotOperandsImpl()
2590 outExpectedTypes = {inputType}; in validateOperationButNotOperandsImpl()
2623 outExpectedTypes = {inputType}; in validateOperationButNotOperandsImpl()
[all …]
/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DLogSoftmax.cpp78 std::vector<OperandType> outExpectedTypes; in validate() local
81 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validate()
84 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validate()
89 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
H A DHeatmapMaxKeypoint.cpp236 std::vector<OperandType> outExpectedTypes; in validate() local
241 outExpectedTypes = {inputType, inputType}; in validate()
245 outExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM, OperandType::TENSOR_QUANT16_ASYMM}; in validate()
249 outExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM_SIGNED, in validate()
256 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
H A DUnidirectionalSequenceLSTM.cpp126 std::vector<OperandType> outExpectedTypes; in validate() local
142 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validate()
158 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validate()
167 outExpectedTypes.insert(outExpectedTypes.end(), {inputType, inputType}); in validate()
170 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
H A DLocalResponseNormalization.cpp146 std::vector<OperandType> outExpectedTypes; in validate() local
154 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validate()
161 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validate()
H A DQLSTM.cpp146 std::vector<OperandType> outExpectedTypes; in validate() local
148 outExpectedTypes.push_back(OperandType::TENSOR_QUANT8_ASYMM_SIGNED); in validate()
150 outExpectedTypes.push_back(OperandType::TENSOR_QUANT16_SYMM); in validate()
152 outExpectedTypes.push_back(OperandType::TENSOR_QUANT8_ASYMM_SIGNED); in validate()
153 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
H A DGenerateProposals.cpp717 std::vector<OperandType> outExpectedTypes; in validate() local
724 outExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate()
731 outExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate()
744 outExpectedTypes = {inputType, OperandType::TENSOR_QUANT16_ASYMM, OperandType::TENSOR_INT32, in validate()
750 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
1231 std::vector<OperandType> outExpectedTypes; in validate() local
1245 outExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate()
1259 outExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate()
1274 outExpectedTypes = {inputType, OperandType::TENSOR_QUANT16_ASYMM, in validate()
1280 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
[all …]
H A DBidirectionalSequenceRNN.cpp336 std::vector<OperandType> outExpectedTypes(numOutputs, inputType); in validate() local
337 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
H A DFullyConnected.cpp230 std::vector<OperandType> outExpectedTypes; in validate() local