message: Internal clean up
diff: @@ -21,6 +21,7 @@ limitations under the License.
  #include "absl/memory/memory.h"
  #include "absl/strings/str_cat.h"
  #include "absl/strings/str_join.h"
  +#include "absl/strings/str_replace.h"
  #include "tensorflow/compiler/xla/layout_util.h"
  #include "tensorflow/compiler/xla/literal.h"
  #include "tensorflow/compiler/xla/service/hlo_casting_utils.h"
  