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

ops in readme

Browse Source
This commit is contained in:
George Hotz
2020-12-30 10:48:55 -05:00
parent 2d44bf7f1a
commit e5b2803b5d
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@@ -105,16 +105,18 @@ Warning: do not rely on the ANE port. It segfaults sometimes. So if you were doi
### Adding an accelerator
You need to support 14 basic ops:
You need to support 14 first class ops:
```
Relu, Log, Exp # unary ops
Add, Sub, Mul, Pow # binary ops (with broadcasting)
Sum, Max # reduce ops (with axis argument)
Matmul, Conv2D # matrix multiplication and conv
Reshape, Transpose, Slice # moving things around ops
Matmul, Conv2D # heavy data processing ops
```
While more ops may be added (like Sign), I think these base 14 are stable.
## ImageNet inference
Despite being tiny, tinygrad supports the full EfficientNet. Pass in a picture to discover what it is.