The mentioned TF functions seem to be related to some internal utility functions for RNNs, while your PyTorch code traces a model and stores it.
I’m not sure, how these approaches are related.
If your PyTorch model also implements the TF methods, then you should be probably fine.