Write out timeline and graph files on step -2 instead of -1.
Write out timeline and graph files on step -2 instead of -1.

That way, the images/sec reported isn't affected by the time to write out the timeline.

PiperOrigin-RevId: 191834054

annarev/benchmarks,tensorflow/benchmarks,tensorflow/benchmarks,annarev/benchmarks