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


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

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

topi
~~~~
81 82
.. autofunction:: topi.negative
.. autofunction:: topi.identity
83 84 85 86
.. autofunction:: topi.floor
.. autofunction:: topi.ceil
.. autofunction:: topi.trunc
.. autofunction:: topi.round
87 88 89 90
.. autofunction:: topi.exp
.. autofunction:: topi.tanh
.. autofunction:: topi.log
.. autofunction:: topi.sqrt
91
.. autofunction:: topi.sigmoid
92 93
.. autofunction:: topi.clip
.. autofunction:: topi.cast
94
.. autofunction:: topi.transpose
95 96
.. autofunction:: topi.flip
.. autofunction:: topi.strided_slice
97
.. autofunction:: topi.expand_dims
98 99 100 101
.. autofunction:: topi.reshape
.. autofunction:: topi.squeeze
.. autofunction:: topi.concatenate
.. autofunction:: topi.split
102
.. autofunction:: topi.take
103 104
.. autofunction:: topi.full
.. autofunction:: topi.full_like
105 106 107
.. autofunction:: topi.max
.. autofunction:: topi.sum
.. autofunction:: topi.min
108
.. autofunction:: topi.broadcast_to
109 110 111 112 113 114 115 116 117 118
.. 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
119 120 121 122

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

135 136 137 138
topi.image
~~~~~~~~~~
.. autofunction:: topi.image.resize

139

140 141 142
topi.generic
~~~~~~~~~~~~
.. automodule:: topi.generic
143

144 145 146 147 148
.. 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