Save view_fn for inplace update on view tensors (#36073)
Summary:
This PR enables inplace updates on view Tensors for tensor types(XLA) that doesn't support as_strided.
(See Notes inside PR)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/36073
Reviewed By: yf225
Differential Revision: D20994282
Pulled By: ailzhang
fbshipit-source-id: 83eeccb297b242f9822f08ad110a7045d7055639
