Handle torch function subclass/mode dispatch on generic tensor methods (#137119)
Summary:
X-link: https://github.com/pytorch/pytorch/pull/137119
Approved by: https://github.com/williamwen42, https://github.com/anijain2305
ghstack dependencies: #137114, #137115, #137116, #137117, #137120, #137227
Reviewed By: jovianjaison
Differential Revision: D64088048
Pulled By: mlazos
fbshipit-source-id: 34fe09f7fa6292d89a438b780852f00e042ec950