search

xmartlabs / Bender

Easily craft fast Neural Networks on iOS! Use TensorFlow models. Metal under the hood.
https://xmartlabs.github.io/Bender/
MIT License
1.8k stars 90 forks source link

Make the metal kernels independents of the type #41

Open bryant1410 opened 7 years ago

bryant1410 commented 7 years ago

Maybe define a macro in a header with the value float, so it could be easily changed to another one.