Skip to content
Projects
Groups
Snippets
Help
This project
Loading...
Sign in / Register
Toggle navigation
T
tic
Overview
Overview
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
wenyuanbo
tic
Commits
b805c69c
Commit
b805c69c
authored
Feb 01, 2019
by
Yong Wu
Committed by
Tianqi Chen
Feb 01, 2019
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
[DOCS][RELAY] Sync up ops with code base (#2532)
parent
32067196
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
22 additions
and
15 deletions
+22
-15
docs/langref/relay_op.rst
+22
-15
No files found.
docs/langref/relay_op.rst
View file @
b805c69c
...
@@ -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
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment