https://devhubby.com/thread/how-to-remove-nodes-from-tensorflow-graph