[pruning] Additional pruning tests (#60681)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/60681
Adding additional pruning tests for more complex models and more pruned rows
Test Plan:
`buck test mode/dev-nosan //caffe2/test:ao -- TestBasePruner`
https://pxl.cl/1KQ2Z
Reviewed By: z-a-f
Differential Revision: D29347546
fbshipit-source-id: cb65e564dd46d24f4aca1b00dd915ee8d64f8318
