get_collection doesn’t work unless I use a checkpoint, what’s the equivalent for a .pb?

Hey guys, so I also made this stackoverflow post if you want to
see it:
https://stackoverflow.com/questions/65662127/what-is-the-equivalent-of-get-collection-for-the-frozen-inference-model

So when testing my predictions are correct on the model,
normally I write get_collection(“logits”).

However, for a .pb/inference model I can’t do this anymore
because the saver.restore only works for checkpoints, not .pb, so
even though I can get tensors, I’m not sure how to get this.

What is the equivalent action for the inference model?