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

Merge "Exclude unused operands from removeOperandTest"

Browse source
This commit is contained in:
Slava Shklyaev 2020-04-21 14:32:23 +00:00 committed by Gerrit Code Review
commit d67c1ee89a
1 changed files with 10 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -535,13 +535,18 @@ static void removeOperand(Model* model, uint32_t index) {
removeValueAndDecrementGreaterValues(&model->main.outputIndexes, index); removeValueAndDecrementGreaterValues(&model->main.outputIndexes, index);
} }
static bool removeOperandSkip(size_t operand, const Model& model) { static bool removeOperandSkip(size_t operandIndex, const Model& model) {
const Operand& operand = model.main.operands[operandIndex];
if (operand.numberOfConsumers == 0) {
// Removing an unused operand has no effect.
return true;
}
for (const Operation& operation : model.main.operations) { for (const Operation& operation : model.main.operations) {
// Skip removeOperandTest for the following operations. // Skip removeOperandTest for the following operations.
// - SPLIT's outputs are not checked during prepareModel. // - SPLIT's outputs are not checked during prepareModel.
if (operation.type == OperationType::SPLIT) { if (operation.type == OperationType::SPLIT) {
for (const size_t outOprand : operation.outputs) { for (const size_t index : operation.outputs) {
if (operand == outOprand) { if (index == operandIndex) {
return true; return true;
} }
} }
@ -556,8 +561,8 @@ static bool removeOperandSkip(size_t operand, const Model& model) {
operation.type == OperationType::UNIDIRECTIONAL_SEQUENCE_RNN || operation.type == OperationType::UNIDIRECTIONAL_SEQUENCE_RNN ||
operation.type == OperationType::BIDIRECTIONAL_SEQUENCE_LSTM || operation.type == OperationType::BIDIRECTIONAL_SEQUENCE_LSTM ||
operation.type == OperationType::BIDIRECTIONAL_SEQUENCE_RNN) { operation.type == OperationType::BIDIRECTIONAL_SEQUENCE_RNN) {
for (const size_t outOprand : operation.outputs) { for (const size_t index : operation.outputs) {
if (operand == outOprand) { if (index == operandIndex) {
return true; return true;
} }
} }