diff --git a/tensorflow/compiler/xla/service/buffer_value.h b/tensorflow/compiler/xla/service/buffer_value.h
index bd2a09e4aaf..8177eb63389 100644
--- a/tensorflow/compiler/xla/service/buffer_value.h
+++ b/tensorflow/compiler/xla/service/buffer_value.h
@@ -108,18 +108,21 @@ class BufferValue {
 
   // Return the color of the BufferValue. Differently colored buffers can not be
   // parts of the same allocation.
+  ABSL_DEPRECATED("Use Layout::memory_space instead.")
   Color color() const {
     CHECK_NE(color_, kInvalidColor)
         << "Should not query the color of a buffer that was never colored";
     return color_;
   }
 
+  ABSL_DEPRECATED("Use Layout::memory_space instead.")
   void set_color(Color color) {
     CHECK_NE(color, kInvalidColor)
         << "Should not set the color of a buffer to the invalid color";
     color_ = color;
   }
 
+  ABSL_DEPRECATED("Use Layout::memory_space instead.")
   bool has_color() const { return color_ != kInvalidColor; }
 
   // Return the shape of the buffer. This reference points into the shape field