Skip to content

T
tic
Commit 7384cd0f by Tatsuya Nishiyama Committed by Tianqi Chen
Options

Fix tutorial to follow the conv2d change (#1390)

parent e1283ff5
Showing with 1 additions and 1 deletions
tutorials/topi/intro_topi.py
...@@ -107,7 +107,7 @@ with tvm.target.create("cuda"): ...@@ -107,7 +107,7 @@ with tvm.target.create("cuda"):
# #
data = tvm.placeholder((1, 3, 224, 224)) data = tvm.placeholder((1, 3, 224, 224))
kernel = tvm.placeholder((10, 3, 5, 5)) kernel = tvm.placeholder((10, 3, 5, 5))
conv = topi.nn.conv2d(data, kernel, stride=1, padding=2) conv = topi.nn.conv2d(data, kernel, strides=1, padding=2)
out = topi.nn.relu(conv) out = topi.nn.relu(conv)
with tvm.target.create("cuda"): with tvm.target.create("cuda"):
sconv = topi.generic.nn.schedule_conv2d_nchw(out) sconv = topi.generic.nn.schedule_conv2d_nchw(out)
......
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