Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit c02a133f authored by TreeHugger Robot's avatar TreeHugger Robot Committed by Android (Google) Code Review
Browse files

Merge "mutateOperationOperandTypeSkip for new TRANSPOSE_CONV_2D variant"

parents f52680bd 1509a266

neuralnetworks/1.2/vts/functional/ValidateModel.cpp

+2 −0
Original line number Diff line number Diff line
@@ -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: {