Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/
DUtils.cpp582 std::vector<OperandType> outExpectedTypes; in validateOperation() local
586 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validateOperation()
590 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validateOperation()
600 outExpectedTypes); in validateOperation()
614 std::vector<OperandType> outExpectedTypes; in validateOperation() local
622 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validateOperation()
630 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validateOperation()
649 outExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM}; in validateOperation()
708 outExpectedTypes); in validateOperation()
719 std::vector<OperandType> outExpectedTypes; in validateOperation() local
[all …]
/frameworks/ml/nn/common/operations/
DLogSoftmax.cpp77 std::vector<OperandType> outExpectedTypes; in validate() local
80 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validate()
83 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validate()
89 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
DHeatmapMaxKeypoint.cpp206 std::vector<OperandType> outExpectedTypes; in validate() local
210 outExpectedTypes = {inputType, inputType}; in validate()
214 outExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM, OperandType::TENSOR_QUANT16_ASYMM}; in validate()
220 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
DUnidirectionalSequenceLSTM.cpp116 std::vector<OperandType> outExpectedTypes; in validate() local
132 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validate()
148 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validate()
155 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
DGenerateProposals.cpp636 std::vector<OperandType> outExpectedTypes; in validate() local
643 outExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate()
650 outExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate()
662 outExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM, OperandType::TENSOR_QUANT16_ASYMM, in validate()
668 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
1112 std::vector<OperandType> outExpectedTypes; in validate() local
1126 outExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate()
1140 outExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate()
1154 outExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM, OperandType::TENSOR_QUANT16_ASYMM, in validate()
1160 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
[all …]
DFullyConnected.cpp146 std::vector<OperandType> outExpectedTypes; in validate() local