search

tensorflow / gnn

TensorFlow GNN is a library to build Graph Neural Networks on the TensorFlow platform.
Apache License 2.0
1.36k stars 179 forks source link

Is it possible to port a trained gnn model to TensorFlow Lite (TFLite) for use in Java/Android applications? #808

Closed knn1989 closed 4 months ago

knn1989 commented 5 months ago

Hello,

Is it possible to port a trained gnn model to TensorFlow Lite (TFLite) for use in Java/Android applications?

If so, how are the input and output buffers declared? For instance, with a regular TFLite model, we typically use FloatBuffer or IntBuffer in Java/Android.

Could you provide an example?

Thank you.

arnoegw commented 4 months ago

Yes, this is possible; please see issue https://github.com/tensorflow/gnn/issues/33#issuecomment-2049220142 and the links given there.