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

one less loop

Browse Source
This commit is contained in:
George Hotz
2020-10-19 09:34:55 -07:00
parent 5c2ac48c11
commit bee89a4840
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tinygrad/tensor.py
View File

@@ -163,10 +163,9 @@ class Conv2D(Function):
for X in range(ret.shape[3]):
for j in range(H):
for i in range(W):
for c in range(cout):
tx = x[:, :, Y+j, X+i]
tw = w[c, :, j, i]
ret[:, c, Y, X] += tx.dot(tw.reshape(-1, 1)).reshape(-1)
tx = x[:, :, Y+j, X+i]
tw = w[:, :, j, i]
ret[:, :, Y, X] += tx.dot(tw.T)
return ret
@staticmethod