Fixed concat axis parsing from tensor flow models

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

Fixed concat axis parsing from tensor flow models #46

Closed dernster closed 7 years ago

dernster commented 7 years ago

Fixes #40

Concat nodes in tensorflow have an input node which contains the axis of the operation. Also, that special input node is named as <concat node name>/axis