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