mutateOperationOperandTypeSkip for new GROUPED_CONV_2D variant
Bug: 119255406 Test: Vts NNAPI tests. Change-Id: I234727f5abba546c1392f7096b7d411d36566bbf
This commit is contained in:
parent
647ad0c56a
commit
b0762ccb2f
1 changed files with 2 additions and 0 deletions
|
@ -336,6 +336,7 @@ static bool mutateOperationOperandTypeSkip(size_t operand, OperandType type, con
|
|||
// - RANDOM_MULTINOMIAL's argument can be either TENSOR_FLOAT16 or TENSOR_FLOAT32.
|
||||
// - CONV_2D filter type (arg 1) can be QUANT8_ASYMM or QUANT8_SYMM_PER_CHANNEL
|
||||
// - DEPTHWISE_CONV_2D filter type (arg 1) can be QUANT8_ASYMM or QUANT8_SYMM_PER_CHANNEL
|
||||
// - GROUPED_CONV_2D filter type (arg 1) can be QUANT8_ASYMM or QUANT8_SYMM_PER_CHANNEL
|
||||
switch (operation.type) {
|
||||
case OperationType::LSH_PROJECTION: {
|
||||
if (operand == operation.inputs[1]) {
|
||||
|
@ -355,6 +356,7 @@ static bool mutateOperationOperandTypeSkip(size_t operand, OperandType type, con
|
|||
return true;
|
||||
}
|
||||
} break;
|
||||
case OperationType::GROUPED_CONV_2D:
|
||||
case OperationType::DEPTHWISE_CONV_2D:
|
||||
case OperationType::CONV_2D: {
|
||||
if (operand == 1 && (type == OperandType::TENSOR_QUANT8_ASYMM ||
|
||||
|
|
Loading…
Reference in a new issue