Missed TensorFlow World? Check out the recap. Learn more

tf.contrib.memory_stats.MaxBytesInUse

tf.contrib.memory_stats.MaxBytesInUse()

Defined in tensorflow/contrib/memory_stats/python/ops/memory_stats_ops.py.

Generates an op that computes the peak memory of a device.