autotvm.rst 1.34 KB
Newer Older
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 50 51 52 53 54 55 56 57
tvm.autotvm
-----------
.. automodule:: tvm.autotvm

tvm.autotvm.measure
~~~~~~~~~~~~~~~~~~~
.. automodule:: tvm.autotvm.measure.measure

.. autoclass:: tvm.autotvm.measure.MeasureInput
    :members:

.. autoclass:: tvm.autotvm.measure.MeasureResult
    :members:

.. autofunction:: tvm.autotvm.measure.measure_option

.. autofunction:: tvm.autotvm.measure.create_measure_batch


tvm.autotvm.tuner
~~~~~~~~~~~~~~~~~
.. automodule:: tvm.autotvm.tuner
    :members:

.. autoclass:: tvm.autotvm.tuner.Tuner
    :members:

.. autoclass:: tvm.autotvm.tuner.RandomTuner
    :members:
    :inherited-members:

.. autoclass:: tvm.autotvm.tuner.GridSearchTuner
    :members:
    :inherited-members:

.. autoclass:: tvm.autotvm.tuner.GATuner
    :members:
    :inherited-members:

.. autoclass:: tvm.autotvm.tuner.XGBTuner
    :members:
    :inherited-members:

.. automodule:: tvm.autotvm.tuner.callback
    :members:

tvm.autotvm.task
~~~~~~~~~~~~~~~~
.. automodule:: tvm.autotvm.task
    :members:

.. automodule:: tvm.autotvm.task.task
    :members:

.. automodule:: tvm.autotvm.task.space
    :members:

58 59 60 61 62 63 64 65 66
.. automodule:: tvm.autotvm.task.dispatcher
    :members:

.. automodule:: tvm.autotvm.task.topi_integration
    :members:

.. automodule:: tvm.autotvm.task.nnvm_integration
    :members:

67 68 69 70
tvm.autotvm.record
~~~~~~~~~~~~~~~~~~
.. automodule:: tvm.autotvm.record
    :members: