mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
2ba930d4ce
Fixes https://github.com/pytorch/pytorch/issues/155031 * [profiler.rst](https://github.com/pytorch/pytorch/tree/main/docs/source/profiler.rst) Pull Request resolved: https://github.com/pytorch/pytorch/pull/155559 Approved by: https://github.com/svekars
991 B
991 B
.. currentmodule:: torch.profiler
torch.profiler
Overview
.. automodule:: torch.profiler
API Reference
.. 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
.. autofunction:: torch.profiler.itt.is_available
.. autofunction:: torch.profiler.itt.mark
.. autofunction:: torch.profiler.itt.range_push
.. autofunction:: torch.profiler.itt.range_pop
.. py:module:: torch.profiler.itt
.. py:module:: torch.profiler.profiler
.. py:module:: torch.profiler.python_tracer