github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-22 21:38:10 -05:00
Code Issues Packages Projects Releases Wiki Activity

oops, we should have a linter in a precommit hook

Browse Source
This commit is contained in:
George Hotz
2022-06-10 10:35:50 -07:00
parent 160f651c25
commit c4238187be
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/llops/ops_cpu.py
View File

@@ -100,7 +100,7 @@ def convdw(x,grad_output,dw,stride,groups):
return dw
def convdx(w,grad_output,dx,stride,groups):
C = get_conv_args(x.shape, dw.shape, stride, groups)
C = get_conv_args(dx.shape, w.shape, stride, groups)
ggg = grad_output.reshape(C.bs, C.groups, C.rcout, C.oy, C.ox)
tw = w.reshape(C.groups, C.rcout, C.cin, C.H, C.W)
gdx = dx.reshape((C.bs, C.groups, C.cin, C.iy, C.ix))