Add runtime statistics

[fvisin]

Add a flag to optionally save runtime statistics every n batches to be visualized in tensorboard.
See more information here:
https://www.tensorflow.org/get_started/graph_viz#runtime_statistics