pytorch/docs/source/benchmark_utils.rst
chronos_secgrp_pytorch_oss_ci_oncall 7d3f48cab3 2022-03-11 nightly release (794f813522)
2022-03-11 00:02:29 -08:00

27 lines
681 B
ReStructuredText

.. role:: hidden
:class: hidden-section
Benchmark Utils - torch.utils.benchmark
==================================================
.. automodule:: torch.utils.benchmark
.. currentmodule:: torch.utils.benchmark
.. autoclass:: Timer
:members:
.. autoclass:: Measurement
:members:
.. autoclass:: CallgrindStats
:members:
.. autoclass:: FunctionCounts
:members:
.. These are missing documentation. Adding them here until a better place
.. is made in this file.
.. py:module:: torch.utils.benchmark.examples
.. py:module:: torch.utils.benchmark.op_fuzzers
.. py:module:: torch.utils.benchmark.utils
.. py:module:: torch.utils.benchmark.utils.valgrind_wrapper