• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1.. currentmodule:: torch.profiler
2
3torch.profiler
4==============
5
6Overview
7--------
8.. automodule:: torch.profiler
9
10
11API Reference
12-------------
13
14.. autoclass:: torch.profiler._KinetoProfile
15  :members:
16
17.. autoclass:: torch.profiler.profile
18  :members:
19
20.. autoclass:: torch.profiler.ProfilerAction
21  :members:
22
23.. autoclass:: torch.profiler.ProfilerActivity
24  :members:
25
26.. autofunction:: torch.profiler.schedule
27
28.. autofunction:: torch.profiler.tensorboard_trace_handler
29
30Intel Instrumentation and Tracing Technology APIs
31-------------------------------------------------
32
33.. autofunction:: torch.profiler.itt.is_available
34
35.. autofunction:: torch.profiler.itt.mark
36
37.. autofunction:: torch.profiler.itt.range_push
38
39.. autofunction:: torch.profiler.itt.range_pop
40
41.. This module needs to be documented. Adding here in the meantime
42.. for tracking purposes
43.. py:module:: torch.profiler.itt
44.. py:module:: torch.profiler.profiler
45.. py:module:: torch.profiler.python_tracer