TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.contrib.memory_stats.MaxBytesInUse

View source on GitHub

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

tf.contrib.memory_stats.MaxBytesInUse()