[PyTorch][easy] Use sizes()[x] instead of size(x) in addr (#55247)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/55247
When x is known to be in-bounds, sizes() is faster.
ghstack-source-id: 126170048
Test Plan: CI
Reviewed By: hlu1
Differential Revision: D27523681
fbshipit-source-id: 021c82a8a6b770802f4cd51cf6ff77046d71c938