Fix a test bug

[yzheng124]

Changes

I double checked the test and found that there is a bug in the original filter pruning test. Maybe this case can let Nikolay want to merge our PR. Thanks!

Reason for changes

The old function cannot deal with the following test case MobilenetV3BlockSEReshape:
Here is part of the original graph of MobilenetV3BlockSEReshape:

For test case MobilenetV3BlockSEReshape, here is the result of tmp_out_channels and tmp_in_channels using the function in public NNCF: (tmp_out_channels and tmp_in_channels are generated here):

tmp_out_channels = 
{
'MobilenetV3BlockSEReshape/NNCFConv2d[first_conv]/conv2d_0': 1,
'MobilenetV3BlockSEReshape/InvertedResidual[inverted_residual]/Sequential[conv]/NNCFConv2d[0]/conv2d_0': 1,
'MobilenetV3BlockSEReshape/InvertedResidual[inverted_residual]/Sequential[conv]/SELayerWithReshapeAndLinear[3]/Sequential[fc]/NNCFLinear[0]/linear_0': 1,
'MobilenetV3BlockSEReshape/InvertedResidual[inverted_residual]/Sequential[conv]/SELayerWithReshapeAndLinear[3]/Sequential[fc]/NNCFLinear[3]/linear_0': 1,
'MobilenetV3BlockSEReshape/InvertedResidual[inverted_residual]/Sequential[conv]/NNCFConv2d[4]/conv2d_0': 1,
'MobilenetV3BlockSEReshape/NNCFConv2d[last_conv]/conv2d_0': 1
}


tmp_in_channels = 
{
'MobilenetV3BlockSEReshape/NNCFConv2d[first_conv]/conv2d_0': 1,
'MobilenetV3BlockSEReshape/InvertedResidual[inverted_residual]/Sequential[conv]/NNCFConv2d[0]/conv2d_0': 1,
'MobilenetV3BlockSEReshape/InvertedResidual[inverted_residual]/Sequential[conv]/SELayerWithReshapeAndLinear[3]/Sequential[fc]/NNCFLinear[0]/linear_0': 1,
'MobilenetV3BlockSEReshape/InvertedResidual[inverted_residual]/Sequential[conv]/SELayerWithReshapeAndLinear[3]/Sequential[fc]/NNCFLinear[3]/linear_0': 1,
'MobilenetV3BlockSEReshape/InvertedResidual[inverted_residual]/Sequential[conv]/NNCFConv2d[4]/conv2d_0': 6,
'MobilenetV3BlockSEReshape/NNCFConv2d[last_conv]/conv2d_0': 1
}

We can see node MobilenetV3BlockSEReshape/InvertedResidual[inverted_residual]/Sequential[conv]/NNCFConv2d[4]/conv2d_0: 6 in tmp_in_channels, which is incorrect.

This is because node MobilenetV3BlockSEReshape/InvertedResidual[inverted_residual]/Sequential[conv]/NNCFConv2d[4]/conv2d_0 is in clusters(not in next_nodes) and it is not a depthwise conv, so its input channel doesn't minus pruned elements.

Our PR can deal with this test case. If use our PR,
MobilenetV3BlockSEReshape/InvertedResidual[inverted_residual]/Sequential[conv]/NNCFConv2d[4]/conv2d_0: 1 in tmp_in_channels

Related tickets

N/A

Tests

N/A