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.79k stars 90 forks source link

Temporary images #102

Closed mats-claassen closed 6 years ago

mats-claassen commented 6 years ago

This PR has to be merged after #100

Change MPSImage for MPSTemporaryImage but leaving open the possibility to have some layers with MPSImage output.

This seems to greatly reduce the memory used by the GPU