[Profiler][Minor] Extend Python bindings (#83622)
Adding some fields which are needed for memory profiling.
Differential Revision: [D38528382](https://our.internmc.facebook.com/intern/diff/D38528382/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/83622
Approved by: https://github.com/Gamrix
