topi.rst 5.33 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
..  Licensed to the Apache Software Foundation (ASF) under one
    or more contributor license agreements.  See the NOTICE file
    distributed with this work for additional information
    regarding copyright ownership.  The ASF licenses this file
    to you under the Apache License, Version 2.0 (the
    "License"); you may not use this file except in compliance
    with the License.  You may obtain a copy of the License at

..    http://www.apache.org/licenses/LICENSE-2.0

..  Unless required by applicable law or agreed to in writing,
    software distributed under the License is distributed on an
    "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
    KIND, either express or implied.  See the License for the
    specific language governing permissions and limitations
    under the License.

18 19
TOPI
----
20 21
.. automodule:: topi

22 23
List of operators
~~~~~~~~~~~~~~~~~
24 25 26

.. autosummary::

27 28
   topi.identity
   topi.negative
29 30
   topi.floor
   topi.ceil
31
   topi.sign
32 33
   topi.trunc
   topi.round
34
   topi.abs
35
   topi.isnan
36 37 38 39
   topi.exp
   topi.tanh
   topi.log
   topi.sqrt
40
   topi.rsqrt
41
   topi.sigmoid
42 43
   topi.clip
   topi.cast
44
   topi.reinterpret
45
   topi.transpose
46 47
   topi.flip
   topi.strided_slice
48
   topi.expand_dims
49 50 51 52
   topi.reshape
   topi.squeeze
   topi.concatenate
   topi.split
53
   topi.take
54
   topi.gather_nd
55 56
   topi.full
   topi.full_like
57
   topi.nn.relu
58
   topi.nn.leaky_relu
59
   topi.nn.dilate
60 61
   topi.nn.pool
   topi.nn.global_pool
62
   topi.nn.adaptive_pool
63 64
   topi.nn.upsampling
   topi.nn.softmax
65
   topi.nn.dense
66
   topi.nn.batch_matmul
67
   topi.nn.log_softmax
68 69
   topi.nn.conv2d_nchw
   topi.nn.conv2d_hwcn
70 71
   topi.nn.depthwise_conv2d_nchw
   topi.nn.depthwise_conv2d_nhwc
72 73 74
   topi.max
   topi.sum
   topi.min
75 76
   topi.argmax
   topi.argmin
77
   topi.prod
78
   topi.broadcast_to
79 80 81 82 83 84 85 86 87 88
   topi.add
   topi.subtract
   topi.multiply
   topi.divide
   topi.mod
   topi.maximum
   topi.minimum
   topi.power
   topi.greater
   topi.less
89 90 91 92
   topi.equal
   topi.not_equal
   topi.greater_equal
   topi.less_equal
93
   topi.all
94
   topi.any
95 96 97
   topi.logical_and
   topi.logical_or
   topi.logical_not
98
   topi.arange
99
   topi.stack
100 101
   topi.repeat
   topi.tile
102
   topi.shape
103
   topi.ndarray_size
104
   topi.layout_transform
105
   topi.image.resize
106 107
   topi.argsort
   topi.topk
108
   topi.sequence_mask
109
   topi.one_hot
110 111


112 113
List of schedules
~~~~~~~~~~~~~~~~~
114 115
.. autosummary::

116 117 118 119 120
   topi.generic.schedule_conv2d_nchw
   topi.generic.schedule_depthwise_conv2d_nchw
   topi.generic.schedule_reduce
   topi.generic.schedule_broadcast
   topi.generic.schedule_injective
121 122 123

topi
~~~~
124 125
.. autofunction:: topi.negative
.. autofunction:: topi.identity
126 127
.. autofunction:: topi.floor
.. autofunction:: topi.ceil
128
.. autofunction:: topi.sign
129 130
.. autofunction:: topi.trunc
.. autofunction:: topi.round
131
.. autofunction:: topi.abs
132
.. autofunction:: topi.isnan
133 134 135 136
.. autofunction:: topi.exp
.. autofunction:: topi.tanh
.. autofunction:: topi.log
.. autofunction:: topi.sqrt
137
.. autofunction:: topi.rsqrt
138
.. autofunction:: topi.sigmoid
139 140
.. autofunction:: topi.clip
.. autofunction:: topi.cast
141
.. autofunction:: topi.reinterpret
142
.. autofunction:: topi.transpose
143 144
.. autofunction:: topi.flip
.. autofunction:: topi.strided_slice
145
.. autofunction:: topi.expand_dims
146 147 148 149
.. autofunction:: topi.reshape
.. autofunction:: topi.squeeze
.. autofunction:: topi.concatenate
.. autofunction:: topi.split
150
.. autofunction:: topi.take
151
.. autofunction:: topi.gather_nd
152 153
.. autofunction:: topi.full
.. autofunction:: topi.full_like
154
.. autofunction:: topi.all
155
.. autofunction:: topi.any
156 157 158
.. autofunction:: topi.max
.. autofunction:: topi.sum
.. autofunction:: topi.min
159
.. autofunction:: topi.prod
160
.. autofunction:: topi.broadcast_to
161 162 163 164 165 166 167 168 169 170
.. 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
171
.. autofunction:: topi.arange
172
.. autofunction:: topi.stack
173 174
.. autofunction:: topi.repeat
.. autofunction:: topi.tile
175
.. autofunction:: topi.shape
176
.. autofunction:: topi.ndarray_size
177
.. autofunction:: topi.layout_transform
178 179
.. autofunction:: topi.argsort
.. autofunction:: topi.topk
180
.. autofunction:: topi.sequence_mask
181
.. autofunction:: topi.one_hot
182 183 184
.. autofunction:: topi.logical_and
.. autofunction:: topi.logical_or
.. autofunction:: topi.logical_not
185 186 187 188

topi.nn
~~~~~~~
.. autofunction:: topi.nn.relu
189
.. autofunction:: topi.nn.leaky_relu
190
.. autofunction:: topi.nn.dilate
191 192 193 194
.. autofunction:: topi.nn.pool
.. autofunction:: topi.nn.global_pool
.. autofunction:: topi.nn.upsampling
.. autofunction:: topi.nn.softmax
195
.. autofunction:: topi.nn.dense
196
.. autofunction:: topi.nn.batch_matmul
197
.. autofunction:: topi.nn.log_softmax
198 199
.. autofunction:: topi.nn.conv2d_nchw
.. autofunction:: topi.nn.conv2d_hwcn
200 201 202
.. autofunction:: topi.nn.depthwise_conv2d_nchw
.. autofunction:: topi.nn.depthwise_conv2d_nhwc

203 204 205 206
topi.image
~~~~~~~~~~
.. autofunction:: topi.image.resize

207 208 209 210 211
topi.sparse
~~~~~~~~~~~
.. autofunction:: topi.sparse.csrmv
.. autofunction:: topi.sparse.csrmm
.. autofunction:: topi.sparse.dense
212

213 214 215
topi.generic
~~~~~~~~~~~~
.. automodule:: topi.generic
216

217 218 219 220 221
.. 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