Watch talks from the 2019 TensorFlow Dev Summit Watch now

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.