record the datapipe for each pieces of Dataset (#67613)
Summary:
Add record_function for each DataPipe.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/67613
Reviewed By: H-Huang
Differential Revision: D32246672
Pulled By: ejguan
fbshipit-source-id: 02ef7e75748c5b84fdcbb103398532e1f2962fbf
