how can I do graph traversal on a tensorflow model? Basically I\'d like to search for blocks and replace them automatically, it should work with model loaded from .pb or sav