mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
45 lines
No EOL
976 B
ReStructuredText
45 lines
No EOL
976 B
ReStructuredText
.. 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
|
|
|
|
.. This module needs to be documented. Adding here in the meantime
|
|
.. for tracking purposes
|
|
.. py:module:: torch.profiler.itt
|
|
.. py:module:: torch.profiler.profiler
|
|
.. py:module:: torch.profiler.python_tracer |