Add internal uint b stats to TfOpStats.

PiperOrigin-RevId: 194625155

diff --git a/tensorflow/contrib/tpu/profiler/tf_op_stats.proto b/tensorflow/contrib/tpu/profiler/tf_op_stats.proto
index 63955d1..b9ac1a5 100644 (file)
--- a/tensorflow/contrib/tpu/profiler/tf_op_stats.proto
+++ b/tensorflow/contrib/tpu/profiler/tf_op_stats.proto
@@ -245,4 +245,6 @@ message TfOpStats {
   optional HostOpsResult host_ops = 8;
   // A map from core ID to name.
   map<uint32, string> core_id_to_name_map = 9;
+  // The result for hw unit b stats.
+  optional bytes unit_b_stats = 10;
 }