pytorch

5630c5ac - record_function: update to use custom_class API

record_function: update to use custom_class API Merge after forward-compatibility period is over. Pull Request resolved: https://github.com/pytorch/pytorch/pull/72302 Approved by: https://github.com/albanD