mutateOperationOperandTypeSkip for new DEPTHWISE_CONV_2D variant.
Bug: 119255406
Test: Vts NNAPI tests.
Change-Id: Iaeb1ef1095dde51e9324f526a2126ed5d0491586
Merged-In: Iaeb1ef1095dde51e9324f526a2126ed5d0491586
(cherry picked from commit 47b9141609
)
This commit is contained in:
parent
c3ea0775ba
commit
725f68487a
1 changed files with 2 additions and 0 deletions
|
@ -327,6 +327,7 @@ static bool mutateOperationOperandTypeSkip(size_t operand, OperandType type, con
|
|||
// - CAST's argument can be any of TENSOR_(FLOAT16|FLOAT32|INT32|QUANT8_ASYMM).
|
||||
// - 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
|
||||
switch (operation.type) {
|
||||
case OperationType::LSH_PROJECTION: {
|
||||
if (operand == operation.inputs[1]) {
|
||||
|
@ -346,6 +347,7 @@ static bool mutateOperationOperandTypeSkip(size_t operand, OperandType type, con
|
|||
return true;
|
||||
}
|
||||
} break;
|
||||
case OperationType::DEPTHWISE_CONV_2D:
|
||||
case OperationType::CONV_2D: {
|
||||
if (operand == 1 && (type == OperandType::TENSOR_QUANT8_ASYMM ||
|
||||
type == OperandType::TENSOR_QUANT8_SYMM_PER_CHANNEL)) {
|
||||
|
|
Loading…
Reference in a new issue