reference.rst 1.5 KB
Newer Older
Yuge Zhang's avatar
Yuge Zhang committed
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
Retiarii API Reference
======================

nni.retiarii
------------

..  automodule:: nni.retiarii
    :imported-members:
    :members:

nni.retiarii.codegen
--------------------

..  automodule:: nni.retiarii.codegen
    :imported-members:
    :members:

nni.retiarii.converter
----------------------

..  automodule:: nni.retiarii.converter
    :imported-members:
    :members:

nni.retiarii.evaluator
----------------------

..  automodule:: nni.retiarii.evaluator
    :imported-members:
    :members:

..  automodule:: nni.retiarii.evaluator.pytorch
    :imported-members:
    :members:

nni.retiarii.execution
----------------------

..  automodule:: nni.retiarii.execution
    :imported-members:
    :members:
    :undoc-members:

nni.retiarii.experiment.pytorch
-------------------------------

..  automodule:: nni.retiarii.experiment.pytorch
    :members:

nni.retiarii.nn.pytorch
-----------------------

Please refer to:

* :doc:`construct_space`.
* :doc:`mutator`.
* `torch.nn reference <https://pytorch.org/docs/stable/nn.html>`_.

nni.retiarii.oneshot
--------------------

..  automodule:: nni.retiarii.oneshot
    :imported-members:
    :members:


nni.retiarii.operation_def
--------------------------

..  automodule:: nni.retiarii.operation_def
    :imported-members:
    :members:

nni.retiarii.strategy
---------------------

..  automodule:: nni.retiarii.strategy
    :imported-members:
    :members:

nni.retiarii.utils
------------------

..  automodule:: nni.retiarii.utils
    :members:

.. footbibliography::