Fix typo in event field name.

PiperOrigin-RevId: 194117352

3 files changed, 8 insertions, 8 deletions

diff --git a/tensorflow/contrib/lite/profiling/profile_buffer.h b/tensorflow/contrib/lite/profiling/profile_buffer.h
index 3bfe02571b..b2f565376c 100644
--- a/tensorflow/contrib/lite/profiling/profile_buffer.h
+++ b/tensorflow/contrib/lite/profiling/profile_buffer.h
@@ -37,9 +37,9 @@ struct ProfileEvent {
   // Label of the event. This usually describes the event.
   const char* tag;
   // Timestamp in microseconds when the event began.
-  int64_t begin_timestamp_ms;
+  int64_t begin_timestamp_us;
   // Timestamp in microseconds when the event ended.
-  int64_t end_timestamp_ms;
+  int64_t end_timestamp_us;
   // The field containing the type of event. This must be one of the event types
   // in EventType.
   EventType event_type;
@@ -79,8 +79,8 @@ class ProfileBuffer {
     event_buffer_[index].tag = tag;
     event_buffer_[index].event_type = event_type;
     event_buffer_[index].event_metadata = event_metadata;
-    event_buffer_[index].begin_timestamp_ms = timestamp;
-    event_buffer_[index].end_timestamp_ms = 0;
+    event_buffer_[index].begin_timestamp_us = timestamp;
+    event_buffer_[index].end_timestamp_us = 0;
     current_index_++;
     return index;
   }
@@ -103,7 +103,7 @@ class ProfileBuffer {
     }
 
     int event_index = event_handle % max_size;
-    event_buffer_[event_index].end_timestamp_ms = NowMicros();
+    event_buffer_[event_index].end_timestamp_us = NowMicros();
   }
 
   // Returns the size of the buffer.
diff --git a/tensorflow/contrib/lite/profiling/profile_buffer_test.cc b/tensorflow/contrib/lite/profiling/profile_buffer_test.cc
index 0c5f0cd314..b8784cca45 100644
--- a/tensorflow/contrib/lite/profiling/profile_buffer_test.cc
+++ b/tensorflow/contrib/lite/profiling/profile_buffer_test.cc
@@ -49,13 +49,13 @@ TEST(ProfileBufferTest, AddEvent) {
 
   auto event = GetProfileEvents(buffer)[0];
   EXPECT_EQ(event->tag, "hello");
-  EXPECT_GT(event->begin_timestamp_ms, 0);
+  EXPECT_GT(event->begin_timestamp_us, 0);
   EXPECT_EQ(event->event_type, ProfileEvent::EventType::DEFAULT);
   EXPECT_EQ(event->event_metadata, 42);
 
   buffer.EndEvent(event_handle);
   EXPECT_EQ(1, buffer.Size());
-  EXPECT_GE(event->end_timestamp_ms, event->begin_timestamp_ms);
+  EXPECT_GE(event->end_timestamp_us, event->begin_timestamp_us);
 }
 
 TEST(ProfileBufferTest, OverFlow) {
diff --git a/tensorflow/contrib/lite/profiling/profiler_test.cc b/tensorflow/contrib/lite/profiling/profiler_test.cc
index 994523a8fb..7914f36a31 100644
--- a/tensorflow/contrib/lite/profiling/profiler_test.cc
+++ b/tensorflow/contrib/lite/profiling/profiler_test.cc
@@ -30,7 +30,7 @@ namespace {
 void AssertDurationOfEventAroundMs(const ProfileEvent* event,
                                    double expected_ms, double eps_ms) {
   double duration_ms =
-      (event->end_timestamp_ms - event->begin_timestamp_ms) / 1e3;
+      (event->end_timestamp_us - event->begin_timestamp_us) / 1e3;
   EXPECT_NEAR(expected_ms, duration_ms, eps_ms);
 }