Using tfgnn with tflite without graphtensor input

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

Using tfgnn with tflite without graphtensor input #799

Closed kohlisimranjit closed 6 months ago

kohlisimranjit commented 7 months ago

Hi, I am trying to have a minimum footprint deployment of a tensorflow gnn model. Seems that the tf-GNN models take in graphTensor as input which would require both tensorflow and tfgnn libraries to be present. Is there a way to have tf-GNN lite model to run inference without needing to import the tfgnn library? Thanks!

arnoegw commented 6 months ago

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