github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

readme and .ane()

Browse Source
This commit is contained in:
George Hotz
2020-12-12 16:15:38 -08:00
parent da873cd556
commit bc5df477de
3 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/use_ane.py
View File

@@ -2,7 +2,7 @@
import numpy as np
from tinygrad.tensor import Tensor
a = Tensor([-2,-1,0,1,2]).ane_()
a = Tensor([-2,-1,0,1,2]).ane()
print(a.cpu())
b = a.relu()
print(b.cpu())