search

doc-ai / tensorio-ios

Tensor/IO for iOS, with support for on-device inference and training with TensorFlow and TensorFlow Lite
Other
31 stars 10 forks source link

Add support for PyTorch backend #184

Open phildow opened 4 years ago

phildow commented 4 years ago

Support for inference should be straightforward:

https://pytorch.org/mobile/ios/

If interested please contact me and I can point you to the classes you would need to add and modify. You can also refer to the subspecs in the TensorIO.podspec file to see how we create separate modules for the TFLite and TensorFlow backends.

phildow commented 3 years ago

We have inference working with Android using the on-device build of libtorch and would like to have inference working with the iOS build as well.