tls access helpers (#138061)
Summary:
X-link: https://github.com/pytorch/pytorch/pull/138061
Approved by: https://github.com/yf225
ghstack dependencies: #137953, #137821
Reviewed By: wdvr
Differential Revision: D64796226
Pulled By: xmfan
fbshipit-source-id: 9bf80c1492d7a800a308cb1e99fac63c4752fc52
