I was looking for a solution for pruning. If I prune layer_1 whose output say has a connection with layer_3, I wanted to identify this connection and prune layer_3 accordingly. Pruning here is the context of channel/filter pruning.