summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorTreeHugger Robot <treehugger-gerrit@google.com>2019-01-22 18:40:08 +0000
committerAndroid (Google) Code Review <android-gerrit@google.com>2019-01-22 18:40:08 +0000
commitc02a133f5af6af95da6598673a1b4bb87c76a12f (patch)
tree41922a7e6c24ca22f8f746b1bbd3fe7f2a526e5d
parentf52680bd048c55f413df47de276beb376c77d72b (diff)
parent1509a2664cef4ab883a6ddf4556fbd544c8ca22f (diff)
Merge "mutateOperationOperandTypeSkip for new TRANSPOSE_CONV_2D variant"
-rw-r--r--neuralnetworks/1.2/vts/functional/ValidateModel.cpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/neuralnetworks/1.2/vts/functional/ValidateModel.cpp b/neuralnetworks/1.2/vts/functional/ValidateModel.cpp
index 11284ce0e7..f3e528e40c 100644
--- a/neuralnetworks/1.2/vts/functional/ValidateModel.cpp
+++ b/neuralnetworks/1.2/vts/functional/ValidateModel.cpp
@@ -337,6 +337,7 @@ static bool mutateOperationOperandTypeSkip(size_t operand, OperandType type, con
// - 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
+ // - TRANSPOSE_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]) {
@@ -356,6 +357,7 @@ static bool mutateOperationOperandTypeSkip(size_t operand, OperandType type, con
return true;
}
} break;
+ case OperationType::TRANSPOSE_CONV_2D:
case OperationType::GROUPED_CONV_2D:
case OperationType::DEPTHWISE_CONV_2D:
case OperationType::CONV_2D: {