[torch.compile] Adding torch compile annotations to some models

[torch.compile] Adding torch compile annotations to some models #24919