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

document graph mode

Browse Source
This commit is contained in:
George Hotz
2022-06-05 12:13:05 -07:00
parent 2097d814f6
commit 88de42fb6e
2 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
README.md
View File

@@ -119,7 +119,7 @@ Relu, Log, Exp # unary ops
Sum, Max # reduce ops (with axis argument)
Add, Sub, Mul, Pow # binary ops (with broadcasting)
Reshape, Transpose, Slice # movement ops
Matmul, Conv2D # processing ops
Matmul, Conv2D(NCHW) # processing ops
```
While more ops may be added, I think this base is stable.
@@ -162,6 +162,13 @@ See `examples/yolov3.py`
tinygrad will always be below 1000 lines. If it isn't, we will revert commits until tinygrad becomes smaller.
### Drawing Execution Graph
```bash
GRAPH=1 python3 test/test_mnist.py TestMNIST.test_conv_onestep
dot -Tsvg /tmp/net.dot -o /tmp/net.svg
open /tmp/net.svg
```
### Running tests
```bash