Skip to content

Commit a339c17

Browse files
anijain2305Wei Chen
authored andcommitted
Passing dilation argument to account for API change. (apache#3510)
1 parent ad6e83c commit a339c17

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

topi/recipe/conv/test_conv_int8_intel.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -112,8 +112,8 @@ def run_inference(data_dtype, kernel_dtype, out_dtype, im_height, im_width, in_f
112112

113113
with tvm.target.create(TARGET_NAME):
114114
conv = topi.nn.conv2d_NCHWc(data, kernel, stride=hstride,
115-
padding=hpad, layout='NCHWc',
116-
out_layout='NCHWc', out_dtype=out_dtype)
115+
padding=hpad, dilation=(1, 1),
116+
layout='NCHWc', out_layout='NCHWc', out_dtype=out_dtype)
117117
out = topi.nn.relu(conv)
118118
sch = tvm.create_schedule(out.op)
119119
func = tvm.build(sch, [data, kernel, out], target=TARGET_NAME, name='out')

0 commit comments

Comments
 (0)