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

break maxpool2d on GPU

Browse Source
This commit is contained in:
George Hotz
2020-12-29 13:05:57 -05:00
parent 061e37de39
commit 02655c07d5
5 changed files with 27 additions and 108 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@@ -111,9 +111,9 @@ You need to support 15 basic ops:
Add, Sub, Mul, Pow # binary ops (with broadcasting)
Relu, Log, Exp # unary ops
Sum, Max # reduce ops (with axis argument)
Dot # matrix multiplication
Conv2D, MaxPool2D # 2D ops
Pad2D, Reshape, Transpose # moving things around ops
Dot, Conv2D # matrix multiplication and conv
Reshape, Transpose # moving things around ops
Unpad2D, Pad2D # stupid slices
```
## ImageNet inference