| r''' |
| PyTorch Profiler is a tool that allows the collecton of the performance metrics during the training and inference. |
| Profiler's context manager API can be used to better understand what model operators are the most expensive, |
| examine their input shapes and stack traces, study device kernel activity and visualize the execution trace. |
| |
| .. note:: |
| An earlier version of the API in :mod:`torch.autograd` module is considered legacy and will be deprecated. |
| |
| ''' |
| |
| from .profiler import profile, schedule, supported_activities, tensorboard_trace_handler, ProfilerAction, ProfilerActivity |
| from torch.autograd import kineto_available, _supported_activities, DeviceType |
| from torch.autograd.profiler import record_function |
