Add a map from TPU core id to name to TfOpStats.

PiperOrigin-RevId: 189620850

diff --git a/tensorflow/contrib/tpu/profiler/tf_op_stats.proto b/tensorflow/contrib/tpu/profiler/tf_op_stats.proto
index e5c798a..20ed741 100644 (file)
--- a/tensorflow/contrib/tpu/profiler/tf_op_stats.proto
+++ b/tensorflow/contrib/tpu/profiler/tf_op_stats.proto
@@ -223,4 +223,6 @@ message TfOpStats {
   optional RunEnvironmentResult run_environment = 7;
   // The result for the host operations.
   optional HostOpsResult host_ops = 8;
+  // A map from core ID to name.
+  map<uint32, string> core_id_to_name_map = 9;
 }