diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-03-30 14:45:21 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-30 14:47:32 -0700 |
| commit | 15c10899c9c0e1717251b380330cc248b2c76c9c (patch) | |
| tree | 850efeb043975596144819a19b968c6db75e8abb /tensorflow/core/grappler/costs/virtual_scheduler.cc | |
| parent | 5bc8b00f3c62549ff6954fa8929afce85c4da46b (diff) | |
show breakdown of total execution time with compute and memory time
PiperOrigin-RevId: 191119550
Diffstat (limited to 'tensorflow/core/grappler/costs/virtual_scheduler.cc')
| -rw-r--r-- | tensorflow/core/grappler/costs/virtual_scheduler.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/costs/virtual_scheduler.cc b/tensorflow/core/grappler/costs/virtual_scheduler.cc index 3ac3ae0f8f..0e5c654acf 100644 --- a/tensorflow/core/grappler/costs/virtual_scheduler.cc +++ b/tensorflow/core/grappler/costs/virtual_scheduler.cc @@ -44,6 +44,8 @@ Costs CombineCosts(const Costs& left, const Costs& right) { Costs result = left; result.execution_time += right.execution_time; + result.compute_time += right.compute_time; + result.memory_time += right.memory_time; if (right.inaccurate) { result.inaccurate = true; } @@ -841,6 +843,8 @@ bool VirtualScheduler::MarkCurrNodeExecuted(const Costs& node_costs) { Costs VirtualScheduler::Summary() const { // Print out basic execution summary. VLOG(1) << "Expected execution time: " << graph_costs_.execution_time.count(); + VLOG(1) << "Expected compute time: " << graph_costs_.compute_time.count(); + VLOG(1) << "Expected memory time: " << graph_costs_.memory_time.count(); VLOG(1) << "Expected max memory: " << graph_costs_.max_memory; VLOG(1) << "Expected max per-op buffers: " << graph_costs_.max_per_op_buffers; VLOG(1) << "Expected max per-op streaming buffers: " |