Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/
DUtils.cpp900 std::vector<OperandType> outExpectedTypes; in validateOperation() local
904 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validateOperation()
908 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validateOperation()
912 outExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM}; in validateOperation()
917 outExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM_SIGNED}; in validateOperation()
931 outExpectedTypes); in validateOperation()
942 std::vector<OperandType> outExpectedTypes; in validateOperation() local
946 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validateOperation()
950 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validateOperation()
954 outExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM}; in validateOperation()
[all …]
/frameworks/ml/nn/common/operations/
DLogSoftmax.cpp79 std::vector<OperandType> outExpectedTypes; in validate() local
82 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validate()
85 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validate()
91 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
DHeatmapMaxKeypoint.cpp234 std::vector<OperandType> outExpectedTypes; in validate() local
239 outExpectedTypes = {inputType, inputType}; in validate()
243 outExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM, OperandType::TENSOR_QUANT16_ASYMM}; in validate()
247 outExpectedTypes = {OperandType::TENSOR_QUANT8_ASYMM_SIGNED, in validate()
255 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
DLocalResponseNormalization.cpp143 std::vector<OperandType> outExpectedTypes; in validate() local
150 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validate()
157 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validate()
DUnidirectionalSequenceLSTM.cpp124 std::vector<OperandType> outExpectedTypes; in validate() local
140 outExpectedTypes = {OperandType::TENSOR_FLOAT32}; in validate()
156 outExpectedTypes = {OperandType::TENSOR_FLOAT16}; in validate()
165 outExpectedTypes.insert(outExpectedTypes.end(), {inputType, inputType}); in validate()
168 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
DQLSTM.cpp145 std::vector<OperandType> outExpectedTypes; in validate() local
147 outExpectedTypes.push_back(OperandType::TENSOR_QUANT8_ASYMM_SIGNED); in validate()
149 outExpectedTypes.push_back(OperandType::TENSOR_QUANT16_SYMM); in validate()
151 outExpectedTypes.push_back(OperandType::TENSOR_QUANT8_ASYMM_SIGNED); in validate()
152 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
DGenerateProposals.cpp713 std::vector<OperandType> outExpectedTypes; in validate() local
720 outExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate()
727 outExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate()
740 outExpectedTypes = {inputType, OperandType::TENSOR_QUANT16_ASYMM, OperandType::TENSOR_INT32, in validate()
746 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
1223 std::vector<OperandType> outExpectedTypes; in validate() local
1237 outExpectedTypes = {OperandType::TENSOR_FLOAT16, OperandType::TENSOR_FLOAT16, in validate()
1251 outExpectedTypes = {OperandType::TENSOR_FLOAT32, OperandType::TENSOR_FLOAT32, in validate()
1266 outExpectedTypes = {inputType, OperandType::TENSOR_QUANT16_ASYMM, in validate()
1272 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
[all …]
DBidirectionalSequenceRNN.cpp338 std::vector<OperandType> outExpectedTypes(numOutputs, inputType); in validate() local
339 NN_RET_CHECK(validateOutputTypes(context, outExpectedTypes)); in validate()
DFullyConnected.cpp228 std::vector<OperandType> outExpectedTypes; in validate() local