github-featured/apache--tvm
2
0
Fork 0
Code Issues Pull Requests Deployments Models Agent Notes Packages Projects Releases Wiki Activity

[TOPI][RELAY][PYTORCH]Conv3d_transpose op support added (#5737)

Browse Source 
* [TOPI][RELAY][PYTORCH]Conv3d_transpose op support added

* Test cases in topi/relay

* conv3d_transpose_ncdhw_python added

* Review comments fixed
This commit is contained in:
Samuel
2020-06-11 14:42:35 +05:30
committed by GitHub
parent c7274fd3b0
commit e2fb5039a2
25 changed files with 1077 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/api/python/topi.rst
+4
View File
@@ -214,6 +214,8 @@ topi.nn
.. autofunction:: topi.nn.conv2d_hwcn
.. autofunction:: topi.nn.depthwise_conv2d_nchw
.. autofunction:: topi.nn.depthwise_conv2d_nhwc
.. autofunction:: topi.nn.conv3d_ncdhw
.. autofunction:: topi.nn.conv3d_transpose_ncdhw
.. autofunction:: topi.nn.fifo_buffer
topi.image
@@ -233,6 +235,8 @@ topi.generic
.. autofunction:: topi.generic.schedule_conv2d_nchw
.. autofunction:: topi.generic.schedule_depthwise_conv2d_nchw
.. autofunction:: topi.generic.schedule_conv3d_ncdhw
.. autofunction:: topi.generic.schedule_conv3d_transpose_ncdhw
.. autofunction:: topi.generic.schedule_reduce
.. autofunction:: topi.generic.schedule_broadcast
.. autofunction:: topi.generic.schedule_injective