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

all realize 2 (#4527)

Browse Source 
* all realize 2

* tests fixup

* fix more tests

* fix openpilot

* fix tests

* unneeded
This commit is contained in:
George Hotz
2024-05-10 22:43:09 -07:00
committed by GitHub
parent d2c347fc74
commit 2f970a4fc2
21 changed files with 142 additions and 139 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extra/export_model.py
View File

@@ -1,6 +1,6 @@
from typing import Tuple, Dict, List
from tinygrad.dtype import DType
from tinygrad.device import Program
from tinygrad.renderer import Program
from tinygrad.tensor import Device, Tensor
from tinygrad.engine.jit import TinyJit
from tinygrad.nn.state import get_state_dict