What I want to add is an

evidence, that my colleague came back on AP and later got his H1 extention successfully. Our international office gave the same explanation as the above. If you are holding H1, surely you can remain your H1 status valid after your return on AP. But if you are J1 or F1, you have to work under EAD.