forward  pass pruning