Add prune channels per layer determination logic

[soulsharp]

This PR:

• Adds the logic to determine the number of channels to prune from each layer given the pruning budget
• Adds utilities to make it happen
• Adds heuristics to determine allocation quantities