Added torch.autograd.profiler.record_function() as context manager. (#23428)
Summary:
Added torch.autograd.profiler.record_function() as context manager to annotate block of Python code during profiling.
Fixes https://github.com/pytorch/pytorch/issues/19422 .
Pull Request resolved: https://github.com/pytorch/pytorch/pull/23428
Differential Revision: D16560771
Pulled By: soumith
fbshipit-source-id: 3923130f7647a36a84dbbe28cc59d216d395d3f9