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

Single ReLU in ANE (#188)

Browse Source 
* aneworks

* cleanup
This commit is contained in:
George Hotz
2020-12-12 16:11:34 -08:00
committed by GitHub
parent 07ece2105e
commit da873cd556
6 changed files with 87 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
examples/use_ane.py Executable file
View File

@@ -0,0 +1,9 @@
#!/usr/bin/env python3
import numpy as np
from tinygrad.tensor import Tensor
a = Tensor([-2,-1,0,1,2]).ane_()
print(a.cpu())
b = a.relu()
print(b.cpu())