tequilaOS/platform_hardware_interfaces
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge "Add TENSOR_QUANT8_ASYMM_SIGNED support for ARG(MIN|MAX)"

Browse source
This commit is contained in:
Przemyslaw Szczepaniak 2019-12-11 10:56:20 +00:00 committed by Android (Google) Code Review
commit 463f1a18fd
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
neuralnetworks/1.3/vts/functional/ValidateModel.cpp
View file

@ -340,7 +340,8 @@ static bool mutateOperationOperandTypeSkip(size_t operand, OperandType type, con
case OperationType::ARGMAX:
case OperationType::ARGMIN: {
if (type == OperandType::TENSOR_FLOAT16 || type == OperandType::TENSOR_FLOAT32 ||
type == OperandType::TENSOR_INT32 || type == OperandType::TENSOR_QUANT8_ASYMM) {
type == OperandType::TENSOR_INT32 || type == OperandType::TENSOR_QUANT8_ASYMM ||
type == OperandType::TENSOR_QUANT8_ASYMM_SIGNED) {
return true;
}
} break;