tree c172074e83a1a6a8145c562017d93c7543fa7c2b
parent 7a22f1ea068370d5c9b84787c24c3d1102bedeb8
author Jun Xu <xjun@google.com> 1621275763 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1621276569 -0700

Add function tf.config.experimental.reset_memory_stats.

It allows the tracked peak memory to be reset in the middle of a program, which is useful for test and debug.

PiperOrigin-RevId: 374239680
Change-Id: I1db1255f2f6f0acea5c3832df530fc4db7f56f08