topi.rst 3.25 KB
Newer Older
1 2
TOPI
----
3 4
.. automodule:: topi

5 6
List of operators
~~~~~~~~~~~~~~~~~
7 8 9

.. autosummary::

10 11
   topi.identity
   topi.negative
12 13 14 15
   topi.floor
   topi.ceil
   topi.trunc
   topi.round
16
   topi.abs
17 18 19 20
   topi.exp
   topi.tanh
   topi.log
   topi.sqrt
21
   topi.sigmoid
22 23
   topi.clip
   topi.cast
24
   topi.transpose
25 26
   topi.flip
   topi.strided_slice
27
   topi.expand_dims
28 29 30 31
   topi.reshape
   topi.squeeze
   topi.concatenate
   topi.split
32
   topi.take
33 34
   topi.full
   topi.full_like
35
   topi.nn.relu
36
   topi.nn.leaky_relu
37
   topi.nn.dilate
38 39 40 41 42
   topi.nn.pool
   topi.nn.global_pool
   topi.nn.upsampling
   topi.nn.softmax
   topi.nn.log_softmax
43 44
   topi.nn.conv2d_nchw
   topi.nn.conv2d_hwcn
45 46
   topi.nn.depthwise_conv2d_nchw
   topi.nn.depthwise_conv2d_nhwc
47 48 49
   topi.max
   topi.sum
   topi.min
50 51
   topi.argmax
   topi.argmin
52
   topi.broadcast_to
53 54 55 56 57 58 59 60 61 62
   topi.add
   topi.subtract
   topi.multiply
   topi.divide
   topi.mod
   topi.maximum
   topi.minimum
   topi.power
   topi.greater
   topi.less
63 64 65 66
   topi.equal
   topi.not_equal
   topi.greater_equal
   topi.less_equal
67
   topi.image.resize
68 69


70 71
List of schedules
~~~~~~~~~~~~~~~~~
72 73
.. autosummary::

74 75 76 77 78
   topi.generic.schedule_conv2d_nchw
   topi.generic.schedule_depthwise_conv2d_nchw
   topi.generic.schedule_reduce
   topi.generic.schedule_broadcast
   topi.generic.schedule_injective
79 80 81

topi
~~~~
82 83
.. autofunction:: topi.negative
.. autofunction:: topi.identity
84 85 86 87
.. autofunction:: topi.floor
.. autofunction:: topi.ceil
.. autofunction:: topi.trunc
.. autofunction:: topi.round
88
.. autofunction:: topi.abs
89 90 91 92
.. autofunction:: topi.exp
.. autofunction:: topi.tanh
.. autofunction:: topi.log
.. autofunction:: topi.sqrt
93
.. autofunction:: topi.sigmoid
94 95
.. autofunction:: topi.clip
.. autofunction:: topi.cast
96
.. autofunction:: topi.transpose
97 98
.. autofunction:: topi.flip
.. autofunction:: topi.strided_slice
99
.. autofunction:: topi.expand_dims
100 101 102 103
.. autofunction:: topi.reshape
.. autofunction:: topi.squeeze
.. autofunction:: topi.concatenate
.. autofunction:: topi.split
104
.. autofunction:: topi.take
105 106
.. autofunction:: topi.full
.. autofunction:: topi.full_like
107 108 109
.. autofunction:: topi.max
.. autofunction:: topi.sum
.. autofunction:: topi.min
110
.. autofunction:: topi.broadcast_to
111 112 113 114 115 116 117 118 119 120
.. autofunction:: topi.add
.. autofunction:: topi.subtract
.. autofunction:: topi.multiply
.. autofunction:: topi.divide
.. autofunction:: topi.mod
.. autofunction:: topi.maximum
.. autofunction:: topi.minimum
.. autofunction:: topi.power
.. autofunction:: topi.greater
.. autofunction:: topi.less
121 122 123 124

topi.nn
~~~~~~~
.. autofunction:: topi.nn.relu
125
.. autofunction:: topi.nn.leaky_relu
126
.. autofunction:: topi.nn.dilate
127 128 129 130 131
.. autofunction:: topi.nn.pool
.. autofunction:: topi.nn.global_pool
.. autofunction:: topi.nn.upsampling
.. autofunction:: topi.nn.softmax
.. autofunction:: topi.nn.log_softmax
132 133
.. autofunction:: topi.nn.conv2d_nchw
.. autofunction:: topi.nn.conv2d_hwcn
134 135 136
.. autofunction:: topi.nn.depthwise_conv2d_nchw
.. autofunction:: topi.nn.depthwise_conv2d_nhwc

137 138 139 140
topi.image
~~~~~~~~~~
.. autofunction:: topi.image.resize

141

142 143 144
topi.generic
~~~~~~~~~~~~
.. automodule:: topi.generic
145

146 147 148 149 150
.. autofunction:: topi.generic.schedule_conv2d_nchw
.. autofunction:: topi.generic.schedule_depthwise_conv2d_nchw
.. autofunction:: topi.generic.schedule_reduce
.. autofunction:: topi.generic.schedule_broadcast
.. autofunction:: topi.generic.schedule_injective