conv stride support

2026-01-09 15:08:02 -05:00 · 2020-10-26 08:54:43 -07:00
parent 2a55d7402b
commit 1654008c1f
2 changed files with 12 additions and 6 deletions
--- a/test/test_ops.py
+++ b/test/test_ops.py
@@ -40,7 +40,6 @@ class TestOps(unittest.TestCase):
              lambda x,w: torch.nn.functional.conv2d(x,w).relu(),
              lambda x,w: Tensor.conv2d(x,w).relu(), atol=2e-5, grad_atol=2e-6)

-  @unittest.skip("please write stride support")
  def test_strided_conv2d(self):
    bs = 4
    cin = 3