Add backward graph generation in graphgen

This will help improve the memory optimizations in `training` mode, as for the moment the backward graph in `optimizeMemory` is very simplistic and has several redundant/unwanted links.
