[torch.compile] Adding torch compile annotations to some models (#9641)

[torch.compile] Adding torch compile annotations to some models (#9641) #16170