[pruner] Add bias support (#60970)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/60970
Support adding bias in eager mode
ghstack-source-id: 132695883
Test Plan:
`buck test mode/dev-nosan //caffe2/test:ao -- TestBasePruner`
https://pxl.cl/1L3K3
Reviewed By: z-a-f
Differential Revision: D29441499
fbshipit-source-id: 47e0fff5b3014612bd021e145160ea54e2645e24
