Add compute_time and memory_time to CostGraphDef::Node.

Change: 147871989

1 files changed, 8 insertions, 0 deletions

diff --git a/tensorflow/core/framework/cost_graph.proto b/tensorflow/core/framework/cost_graph.proto
index dfd4e5b327..a6535853a2 100644
--- a/tensorflow/core/framework/cost_graph.proto
+++ b/tensorflow/core/framework/cost_graph.proto
@@ -53,6 +53,14 @@ message CostGraphDef {
     // Estimate of the computational cost of this node, in microseconds.
     int64 compute_cost = 9;
 
+    // Analytical estimate of the computational cost of this node, in
+    // microseconds.
+    int64 compute_time = 14;
+
+    // Analytical estimate of the memory access cost of this node, in
+    // microseconds.
+    int64 memory_time = 15;
+
     // If true, the output is permanent: it can't be discarded, because this
     // node is part of the "final output". Nodes may depend on final nodes.
     bool is_final = 7;