topi.rst 4.99 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 36 37 38
   topi.exp
   topi.tanh
   topi.log
   topi.sqrt
39
   topi.rsqrt
40
   topi.sigmoid
41 42
   topi.clip
   topi.cast
43
   topi.transpose
44 45
   topi.flip
   topi.strided_slice
46
   topi.expand_dims
47 48 49 50
   topi.reshape
   topi.squeeze
   topi.concatenate
   topi.split
51
   topi.take
52
   topi.gather_nd
53 54
   topi.full
   topi.full_like
55
   topi.nn.relu
56
   topi.nn.leaky_relu
57
   topi.nn.dilate
58 59
   topi.nn.pool
   topi.nn.global_pool
60
   topi.nn.adaptive_pool
61 62
   topi.nn.upsampling
   topi.nn.softmax
63
   topi.nn.dense
64
   topi.nn.batch_matmul
65
   topi.nn.log_softmax
66 67
   topi.nn.conv2d_nchw
   topi.nn.conv2d_hwcn
68 69
   topi.nn.depthwise_conv2d_nchw
   topi.nn.depthwise_conv2d_nhwc
70 71 72
   topi.max
   topi.sum
   topi.min
73 74
   topi.argmax
   topi.argmin
75
   topi.prod
76
   topi.broadcast_to
77 78 79 80 81 82 83 84 85 86
   topi.add
   topi.subtract
   topi.multiply
   topi.divide
   topi.mod
   topi.maximum
   topi.minimum
   topi.power
   topi.greater
   topi.less
87 88 89 90
   topi.equal
   topi.not_equal
   topi.greater_equal
   topi.less_equal
91
   topi.all
92 93 94
   topi.logical_and
   topi.logical_or
   topi.logical_not
95
   topi.arange
96
   topi.stack
97 98
   topi.repeat
   topi.tile
99
   topi.shape
100
   topi.layout_transform
101
   topi.image.resize
102 103
   topi.argsort
   topi.topk
104
   topi.sequence_mask
105 106


107 108
List of schedules
~~~~~~~~~~~~~~~~~
109 110
.. autosummary::

111 112 113 114 115
   topi.generic.schedule_conv2d_nchw
   topi.generic.schedule_depthwise_conv2d_nchw
   topi.generic.schedule_reduce
   topi.generic.schedule_broadcast
   topi.generic.schedule_injective
116 117 118

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

topi.nn
~~~~~~~
.. autofunction:: topi.nn.relu
176
.. autofunction:: topi.nn.leaky_relu
177
.. autofunction:: topi.nn.dilate
178 179 180 181
.. autofunction:: topi.nn.pool
.. autofunction:: topi.nn.global_pool
.. autofunction:: topi.nn.upsampling
.. autofunction:: topi.nn.softmax
182
.. autofunction:: topi.nn.dense
183
.. autofunction:: topi.nn.batch_matmul
184
.. autofunction:: topi.nn.log_softmax
185 186
.. autofunction:: topi.nn.conv2d_nchw
.. autofunction:: topi.nn.conv2d_hwcn
187 188 189
.. autofunction:: topi.nn.depthwise_conv2d_nchw
.. autofunction:: topi.nn.depthwise_conv2d_nhwc

190 191 192 193
topi.image
~~~~~~~~~~
.. autofunction:: topi.image.resize

194 195 196 197 198
topi.sparse
~~~~~~~~~~~
.. autofunction:: topi.sparse.csrmv
.. autofunction:: topi.sparse.csrmm
.. autofunction:: topi.sparse.dense
199

200 201 202
topi.generic
~~~~~~~~~~~~
.. automodule:: topi.generic
203

204 205 206 207 208
.. 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