diff options
| author |  Yuefeng Zhou <yuefengz@google.com> | 2018-01-05 18:43:41 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-05 18:47:20 -0800 |
| commit | cfbcbc66a8756d087fd895ff6ec3f5cea32e5157 (patch) | |
| tree | 16f19b932c7c4cee8d492b27b674c2312aa91243 /tensorflow/core/framework/allocator_test.cc | |
| parent | 93f2998add91f9c7a53b1fcd13ab6d43e4397297 (diff) | |
Measure memory for restore subgraph in SaveRestoreMeasuringCoseEstimator.
PiperOrigin-RevId: 181003320
Diffstat (limited to 'tensorflow/core/framework/allocator_test.cc')
| -rw-r--r-- | tensorflow/core/framework/allocator_test.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/framework/allocator_test.cc b/tensorflow/core/framework/allocator_test.cc index 032aeec161..a409cb2de7 100644 --- a/tensorflow/core/framework/allocator_test.cc +++ b/tensorflow/core/framework/allocator_test.cc @@ -110,6 +110,8 @@ TEST(CPUAllocatorTest, Simple) { CheckStats(a, 1025, 0, 1048576 * sizeof(double) + 1024 * sizeof(float), 1048576 * sizeof(double)); + a->ClearStats(); + CheckStats(a, 0, 0, 0, 0); EnableCPUAllocatorStats(false); } |