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

Exclude unused operands from removeOperandTest

Browse source 
Bug: 148208229
Test: TestGenerated/ValidationTest.Test/nnapi_sample*_while*_unused_output*
Change-Id: I39bebefd0e40d370882d614d4beb97ab36c9da8f
Merged-In: I39bebefd0e40d370882d614d4beb97ab36c9da8f
(cherry picked from commit baa4cf02f2)
This commit is contained in:
Slava Shklyaev 2020-04-16 14:58:55 +01:00
parent d05127d16d
commit af1c77d74f
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);
}
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) {
// Skip removeOperandTest for the following operations.
// - SPLIT's outputs are not checked during prepareModel.
if (operation.type == OperationType::SPLIT) {
for (const size_t outOprand : operation.outputs) {
if (operand == outOprand) {
for (const size_t index : operation.outputs) {
if (index == operandIndex) {
return true;
}
}
@ -556,8 +561,8 @@ static bool removeOperandSkip(size_t operand, const Model& model) {
operation.type == OperationType::UNIDIRECTIONAL_SEQUENCE_RNN ||
operation.type == OperationType::BIDIRECTIONAL_SEQUENCE_LSTM ||
operation.type == OperationType::BIDIRECTIONAL_SEQUENCE_RNN) {
for (const size_t outOprand : operation.outputs) {
if (operand == outOprand) {
for (const size_t index : operation.outputs) {
if (index == operandIndex) {
return true;
}
}