Trace events with the new ops (#167177)
Summary:
X-link: https://github.com/pytorch/pytorch/pull/167177
Approved by: https://github.com/anijain2305
ghstack dependencies: #167175, #167176, #167180, #167195, #167260
Reviewed By: donigian
Differential Revision: D86494579
fbshipit-source-id: 29f7ede329be1b9c84ffb8adc5f9ada37d1054a2