Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

Module: tf.contrib.memory_stats

View source on GitHub

Ops for memory statistics.

Functions

BytesInUse(...): Generates an op that computes the current memory of a device.

BytesLimit(...): Generates an op that measures the total memory (in bytes) of a device.

MaxBytesInUse(...): Generates an op that computes the peak memory of a device.