topi.rst 1.82 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
TVM Operator Inventory
----------------------
.. automodule:: topi

Index
~~~~~

**List of operators**

.. autosummary::

   topi.exp
   topi.tanh
   topi.log
   topi.sqrt
16
   topi.sigmoid
17 18
   topi.transpose
   topi.expand_dims
19
   topi.nn.relu
20
   topi.nn.leaky_relu
21 22 23
   topi.nn.dilate
   topi.nn.conv2d_nchw
   topi.nn.conv2d_hwcn
24 25
   topi.nn.depthwise_conv2d_nchw
   topi.nn.depthwise_conv2d_nhwc
26 27 28 29 30 31 32 33
   topi.max
   topi.sum
   topi.min
   topi.broadcast_to
   topi.broadcast_add
   topi.broadcast_sub
   topi.broadcast_mul
   topi.broadcast_div
34 35 36 37 38 39


**List of schedules**

.. autosummary::

40 41 42 43 44
   topi.generic.schedule_conv2d_nchw
   topi.generic.schedule_depthwise_conv2d_nchw
   topi.generic.schedule_reduce
   topi.generic.schedule_broadcast
   topi.generic.schedule_injective
45 46 47 48 49 50 51

topi
~~~~
.. autofunction:: topi.exp
.. autofunction:: topi.tanh
.. autofunction:: topi.log
.. autofunction:: topi.sqrt
52
.. autofunction:: topi.sigmoid
53 54
.. autofunction:: topi.transpose
.. autofunction:: topi.expand_dims
55 56 57
.. autofunction:: topi.max
.. autofunction:: topi.sum
.. autofunction:: topi.min
58 59 60 61 62 63
.. autofunction:: topi.broadcast_to
.. autofunction:: topi.broadcast_add
.. autofunction:: topi.broadcast_sub
.. autofunction:: topi.broadcast_mul
.. autofunction:: topi.broadcast_div

64 65 66 67

topi.nn
~~~~~~~
.. autofunction:: topi.nn.relu
68
.. autofunction:: topi.nn.leaky_relu
69 70 71
.. autofunction:: topi.nn.dilate
.. autofunction:: topi.nn.conv2d_nchw
.. autofunction:: topi.nn.conv2d_hwcn
72 73 74
.. autofunction:: topi.nn.depthwise_conv2d_nchw
.. autofunction:: topi.nn.depthwise_conv2d_nhwc

75

76 77 78
topi.generic
~~~~~~~~~~~~
.. automodule:: topi.generic
79

80 81 82 83 84
.. 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