diff options
author | Benoit Steiner <bsteiner@google.com> | 2017-10-12 18:33:29 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-12 18:37:30 -0700 |
commit | 33fc95f46257e07deed852acf65806055672ce25 (patch) | |
tree | 9a536c1a98189b353657c9bc8937a78293161117 /tensorflow/core/grappler/costs/cost_estimator.h | |
parent | d4d8b81209138332a9b4c16ae106d1f01e9e412d (diff) |
Determine peak memory usage from the result of a simulation.
PiperOrigin-RevId: 172043591
Diffstat (limited to 'tensorflow/core/grappler/costs/cost_estimator.h')
-rw-r--r-- | tensorflow/core/grappler/costs/cost_estimator.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/costs/cost_estimator.h b/tensorflow/core/grappler/costs/cost_estimator.h index 868c4a9733..cf9fa4fdaf 100644 --- a/tensorflow/core/grappler/costs/cost_estimator.h +++ b/tensorflow/core/grappler/costs/cost_estimator.h @@ -121,6 +121,8 @@ Costs::Costs() { Costs Costs::ZeroCosts() { Costs costs; costs.execution_time = Duration::zero(); + costs.compute_time = Duration::zero(); + costs.memory_time = Duration::zero(); costs.max_memory = kZeroMemory; costs.max_per_op_buffers = kZeroMemory; costs.max_per_op_streaming = kZeroMemory; |