message: Add required header
diff: @@ -32,6 +32,8 @@ limitations under the License.
  #include "tensorflow/compiler/xla/xla_data.pb.h"
  #include "tensorflow/core/lib/core/status_test_util.h"
  +#include "absl/strings/str_replace.h"
  +
  namespace xla {
  namespace {
  