Skip to content

T
tic
Commit 5f0a517d by Lianmin Zheng Committed by Tianqi Chen
Options

fix default out type of depthwise conv2d (#1134)

parent 6f355762
Showing with 1 additions and 1 deletions
topi/python/topi/nn/depthwise_conv2d.py
...@@ -10,7 +10,7 @@ from ..util import simplify ...@@ -10,7 +10,7 @@ from ..util import simplify
@tvm.target.generic_func @tvm.target.generic_func
def depthwise_conv2d_nchw(Input, Filter, stride, padding, out_dtype='float32'): def depthwise_conv2d_nchw(Input, Filter, stride, padding, out_dtype=None):
"""Depthwise convolution nchw forward operator. """Depthwise convolution nchw forward operator.
Parameters Parameters
......
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