Make sure work button is collapsed when keyboard is up upon going to app drawer

Call shrink so that work button is collapsed when keyboard is up.

bug:361589193
Test - manual: video before:https://drive.google.com/file/d/1OChJkEJEWrsuUtEaac2jtc15uGkOeGAm/view?usp=sharing
after: https://drive.google.com/file/d/1DXGia25u4JzWRLTiwqlKQu8TYjyL9SEI/view?usp=sharing
Flag: com.android.launcher3.work_scheduler_in_work_profile

Change-Id: Ifafb76b61f0d2a5e9c6b60519adc300c911b20dc