If it was the case of some individual contributors' productivity dropping because of WFH then yes, you can pinpoint that person, return him to the office and put a manager next to him to make him work

But recent years brought me to conclusion that many people have a need for in person communication, and your morale, productivity and even mental health going down little by little every day you interact with others only digitally. You can not fix this, if you bring back to office only selected few. You have to bring back the whole teams — so if a person goes to the office, everyone he works with and interacts with are also there