Skip to content

T
tic
Commit b5c6b993 by Yuwei HU Committed by Tianqi Chen
Options

update depthwise convolution api (#344)

parent 8ca12d87
Showing with 9 additions and 4 deletions
docs/api/python/topi.rst
...@@ -23,7 +23,8 @@ Index ...@@ -23,7 +23,8 @@ Index
topi.nn.scale_shift topi.nn.scale_shift
topi.nn.conv2d_nchw topi.nn.conv2d_nchw
topi.nn.conv2d_hwcn topi.nn.conv2d_hwcn
topi.nn.depthwise_conv2d topi.nn.depthwise_conv2d_nchw
topi.nn.depthwise_conv2d_nhwc
**List of schedules** **List of schedules**
...@@ -32,7 +33,8 @@ Index ...@@ -32,7 +33,8 @@ Index
topi.cuda.schedule_conv2d_nchw topi.cuda.schedule_conv2d_nchw
topi.cuda.schedule_conv2d_hwcn topi.cuda.schedule_conv2d_hwcn
topi.cuda.schedule_depthwise_conv2d topi.cuda.schedule_depthwise_conv2d_nchw
topi.cuda.schedule_depthwise_conv2d_nhwc
topi.cuda.schedule_reduce topi.cuda.schedule_reduce
topi.cuda.schedule_broadcast_to topi.cuda.schedule_broadcast_to
...@@ -56,7 +58,9 @@ topi.nn ...@@ -56,7 +58,9 @@ topi.nn
.. autofunction:: topi.nn.scale_shift .. autofunction:: topi.nn.scale_shift
.. autofunction:: topi.nn.conv2d_nchw .. autofunction:: topi.nn.conv2d_nchw
.. autofunction:: topi.nn.conv2d_hwcn .. autofunction:: topi.nn.conv2d_hwcn
.. autofunction:: topi.nn.depthwise_conv2d .. autofunction:: topi.nn.depthwise_conv2d_nchw
.. autofunction:: topi.nn.depthwise_conv2d_nhwc
topi.cuda topi.cuda
~~~~~~~~~ ~~~~~~~~~
...@@ -64,6 +68,7 @@ topi.cuda ...@@ -64,6 +68,7 @@ topi.cuda
.. autofunction:: topi.cuda.schedule_conv2d_nchw .. autofunction:: topi.cuda.schedule_conv2d_nchw
.. autofunction:: topi.cuda.schedule_conv2d_hwcn .. autofunction:: topi.cuda.schedule_conv2d_hwcn
.. autofunction:: topi.cuda.schedule_depthwise_conv2d .. autofunction:: topi.cuda.schedule_depthwise_conv2d_nchw
.. autofunction:: topi.cuda.schedule_depthwise_conv2d_nhwc
.. autofunction:: topi.cuda.schedule_reduce .. autofunction:: topi.cuda.schedule_reduce
.. autofunction:: topi.cuda.schedule_broadcast_to .. autofunction:: topi.cuda.schedule_broadcast_to
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