github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

rename max_pool2d to match torch, remove more fast conv crap

Browse Source
This commit is contained in:
George Hotz
2020-10-25 17:16:47 -07:00
parent 5d1373c71b
commit 567707a5f6
4 changed files with 10 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_tensor.py
View File

@@ -90,7 +90,7 @@ class TestOps(unittest.TestCase):
xt = Tensor(x.detach().numpy())
# in tinygrad
ret = xt.maxpool2x2()
ret = xt.max_pool2d()
assert ret.shape == (5,2,10//2,8//2)
ret.mean().backward()