Move shape prop inside acc_tracer (#55091)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55091
Test Plan: All tests are updated and passing.
Reviewed By: 842974287
Differential Revision: D27471887
fbshipit-source-id: 98969fb1bfc72f6c57835525d82d4a8b78bb19bb