Add API to support graph_def based snapshots
Bump extend to avoid deprecation warnings