This commit is contained in:
Balint Cristian 2019-07-28 11:05:37 +03:00 коммит произвёл Thierry Moreau
Родитель 9a542e37de
Коммит 026162ad62
1 изменённых файлов: 6 добавлений и 1 удалений

Просмотреть файл

@ -81,7 +81,12 @@ def verify_conv2d_nchw(batch, in_channel, in_size, num_filter, kernel, stride, p
else:
func = tvm.build(s, [A, W, C], device, name="relu_%d_%d_%d_%d_%d_%d_%d_%d" % (batch, in_channel, in_size, num_filter, kernel, stride, padding, dilation))
func(a, w, c)
tvm.testing.assert_allclose(c.asnumpy(), c_np, rtol=1e-5)
rtol = 1e-5
if (kernel > 3):
rtol = 2e-5
tvm.testing.assert_allclose(c.asnumpy(), c_np, rtol=rtol)
for device in devices: