TensorFlow 2.0 Beta is available Learn more

Module: tf.contrib.memory_stats

Ops for memory statistics.

View source on GitHub

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.