pytorch
be3d2758 - Added torch.autograd.profiler.record_function() as context manager. (#23428)

Commit
5 years ago
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
Author
Parents
Loading