Commit b805c69c by Yong Wu Committed by Tianqi Chen

[DOCS][RELAY] Sync up ops with code base (#2532)

parent 32067196
...@@ -27,22 +27,21 @@ This level enables fully connected multi-layer perceptron. ...@@ -27,22 +27,21 @@ This level enables fully connected multi-layer perceptron.
tvm.relay.exp tvm.relay.exp
tvm.relay.sigmoid tvm.relay.sigmoid
tvm.relay.add tvm.relay.add
tvm.relay.expand_dims
tvm.relay.concatenate
tvm.relay.nn.softmax
tvm.relay.nn.log_softmax
tvm.relay.subtract tvm.relay.subtract
tvm.relay.multiply tvm.relay.multiply
tvm.relay.divide tvm.relay.divide
tvm.relay.mod tvm.relay.mod
tvm.relay.tanh tvm.relay.tanh
tvm.relay.concatenate
tvm.relay.expand_dims
tvm.relay.nn.softmax
tvm.relay.nn.log_softmax
tvm.relay.nn.relu tvm.relay.nn.relu
tvm.relay.nn.dropout tvm.relay.nn.dropout
tvm.relay.nn.batch_norm tvm.relay.nn.batch_norm
tvm.relay.nn.bias_add tvm.relay.nn.bias_add
**Level 2: Convolutions** **Level 2: Convolutions**
This level enables typical convnet models. This level enables typical convnet models.
...@@ -62,6 +61,8 @@ This level enables typical convnet models. ...@@ -62,6 +61,8 @@ This level enables typical convnet models.
tvm.relay.nn.pad tvm.relay.nn.pad
tvm.relay.nn.lrn tvm.relay.nn.lrn
tvm.relay.nn.l2_normalize tvm.relay.nn.l2_normalize
tvm.relay.nn.contrib_conv2d_winograd_without_weight_transform
tvm.relay.nn.contrib_conv2d_winograd_weight_transform
**Level 3: Additional Math And Transform Operators** **Level 3: Additional Math And Transform Operators**
...@@ -71,12 +72,8 @@ This level enables additional math and transform operators. ...@@ -71,12 +72,8 @@ This level enables additional math and transform operators.
.. autosummary:: .. autosummary::
:nosignatures: :nosignatures:
tvm.relay.zeros
tvm.relay.nn.leaky_relu tvm.relay.nn.leaky_relu
tvm.relay.nn.prelu tvm.relay.nn.prelu
tvm.relay.zeros_like
tvm.relay.ones
tvm.relay.ones_like
tvm.relay.reshape tvm.relay.reshape
tvm.relay.reshape_like tvm.relay.reshape_like
tvm.relay.copy tvm.relay.copy
...@@ -85,6 +82,7 @@ This level enables additional math and transform operators. ...@@ -85,6 +82,7 @@ This level enables additional math and transform operators.
tvm.relay.floor tvm.relay.floor
tvm.relay.ceil tvm.relay.ceil
tvm.relay.trunc tvm.relay.trunc
tvm.relay.clip
tvm.relay.round tvm.relay.round
tvm.relay.abs tvm.relay.abs
tvm.relay.negative tvm.relay.negative
...@@ -148,6 +146,7 @@ This level support backpropagation of broadcast operators. It is temporary. ...@@ -148,6 +146,7 @@ This level support backpropagation of broadcast operators. It is temporary.
tvm.relay.broadcast_to_like tvm.relay.broadcast_to_like
tvm.relay.collapse_sum_like tvm.relay.collapse_sum_like
tvm.relay.slice_like tvm.relay.slice_like
tvm.relay.layout_transform
tvm.relay.device_copy tvm.relay.device_copy
tvm.relay.annotation.on_device tvm.relay.annotation.on_device
...@@ -185,25 +184,28 @@ Level 2 Definitions ...@@ -185,25 +184,28 @@ Level 2 Definitions
.. autofunction:: tvm.relay.nn.global_avg_pool2d .. autofunction:: tvm.relay.nn.global_avg_pool2d
.. autofunction:: tvm.relay.nn.upsampling .. autofunction:: tvm.relay.nn.upsampling
.. autofunction:: tvm.relay.nn.batch_flatten .. autofunction:: tvm.relay.nn.batch_flatten
.. autofunction:: tvm.relay.nn.pad
.. autofunction:: tvm.relay.nn.lrn .. autofunction:: tvm.relay.nn.lrn
.. autofunction:: tvm.relay.nn.l2_normalize .. autofunction:: tvm.relay.nn.l2_normalize
.. autofunction:: tvm.relay.nn.contrib_conv2d_winograd_without_weight_transform
.. autofunction:: tvm.relay.nn.contrib_conv2d_winograd_weight_transform
Level 3 Definitions Level 3 Definitions
------------------- -------------------
.. autofunction:: tvm.relay.nn.leaky_relu .. autofunction:: tvm.relay.nn.leaky_relu
.. autofunction:: tvm.relay.nn.prelu .. autofunction:: tvm.relay.nn.prelu
.. autofunction:: tvm.relay.reshape
.. autofunction:: tvm.relay.reshape_like
.. autofunction:: tvm.relay.copy
.. autofunction:: tvm.relay.transpose
.. autofunction:: tvm.relay.squeeze
.. autofunction:: tvm.relay.floor .. autofunction:: tvm.relay.floor
.. autofunction:: tvm.relay.ceil .. autofunction:: tvm.relay.ceil
.. autofunction:: tvm.relay.trunc .. autofunction:: tvm.relay.trunc
.. autofunction:: tvm.relay.clip
.. autofunction:: tvm.relay.round .. autofunction:: tvm.relay.round
.. autofunction:: tvm.relay.abs .. autofunction:: tvm.relay.abs
.. autofunction:: tvm.relay.negative .. autofunction:: tvm.relay.negative
.. autofunction:: tvm.relay.reshape
.. autofunction:: tvm.relay.reshape_like
.. autofunction:: tvm.relay.copy
.. autofunction:: tvm.relay.squeeze
.. autofunction:: tvm.relay.transpose
.. autofunction:: tvm.relay.take .. autofunction:: tvm.relay.take
.. autofunction:: tvm.relay.zeros .. autofunction:: tvm.relay.zeros
.. autofunction:: tvm.relay.zeros_like .. autofunction:: tvm.relay.zeros_like
...@@ -237,6 +239,7 @@ Level 4 Definitions ...@@ -237,6 +239,7 @@ Level 4 Definitions
.. autofunction:: tvm.relay.mean .. autofunction:: tvm.relay.mean
.. autofunction:: tvm.relay.prod .. autofunction:: tvm.relay.prod
.. autofunction:: tvm.relay.strided_slice .. autofunction:: tvm.relay.strided_slice
.. autofunction:: tvm.relay.broadcast_to
Level 5 Definitions Level 5 Definitions
...@@ -252,3 +255,6 @@ Level 10 Definitions ...@@ -252,3 +255,6 @@ Level 10 Definitions
.. autofunction:: tvm.relay.broadcast_to_like .. autofunction:: tvm.relay.broadcast_to_like
.. autofunction:: tvm.relay.collapse_sum_like .. autofunction:: tvm.relay.collapse_sum_like
.. autofunction:: tvm.relay.slice_like .. autofunction:: tvm.relay.slice_like
.. autofunction:: tvm.relay.layout_transform
.. autofunction:: tvm.relay.device_copy
.. autofunction:: tvm.relay.annotation.on_device
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment