From 64b01d55747c1d7ccdaca2585b4b9330cb4179e2 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Tue, 13 Aug 2019 15:35:12 -0700
Subject: [PATCH] XLA: IWYU on op definitions

Adding missing headers that are used in these files

PiperOrigin-RevId: 263229994
---
 tensorflow/compiler/mlir/xla/ir/xla_ops.cc | 20 ++++++++++++++++++++
 tensorflow/compiler/mlir/xla/ir/xla_ops.h  |  7 ++++++-
 2 files changed, 26 insertions(+), 1 deletion(-)

diff --git a/tensorflow/compiler/mlir/xla/ir/xla_ops.cc b/tensorflow/compiler/mlir/xla/ir/xla_ops.cc
index c1f3d010c46..a5266778415 100644
--- a/tensorflow/compiler/mlir/xla/ir/xla_ops.cc
+++ b/tensorflow/compiler/mlir/xla/ir/xla_ops.cc
@@ -17,12 +17,32 @@ limitations under the License.
 
 #include "tensorflow/compiler/mlir/xla/ir/xla_ops.h"
 
+#include <assert.h>
+#include <stddef.h>
+#include <stdint.h>
+
+#include "llvm/ADT/APFloat.h"
+#include "llvm/ADT/APInt.h"
+#include "llvm/ADT/ArrayRef.h"
+#include "llvm/ADT/STLExtras.h"
+#include "llvm/ADT/SmallVector.h"
+#include "llvm/ADT/StringRef.h"
 #include "llvm/Support/FormatVariadic.h"
 #include "mlir/IR/Attributes.h"  // TF:local_config_mlir
 #include "mlir/IR/Builders.h"  // TF:local_config_mlir
+#include "mlir/IR/Dialect.h"  // TF:local_config_mlir
+#include "mlir/IR/Location.h"  // TF:local_config_mlir
+#include "mlir/IR/MLIRContext.h"  // TF:local_config_mlir
+#include "mlir/IR/OpDefinition.h"  // TF:local_config_mlir
 #include "mlir/IR/OpImplementation.h"  // TF:local_config_mlir
+#include "mlir/IR/Operation.h"  // TF:local_config_mlir
+#include "mlir/IR/OperationSupport.h"  // TF:local_config_mlir
 #include "mlir/IR/PatternMatch.h"  // TF:local_config_mlir
+#include "mlir/IR/StandardTypes.h"  // TF:local_config_mlir
 #include "mlir/IR/TypeUtilities.h"  // TF:local_config_mlir
+#include "mlir/IR/Types.h"  // TF:local_config_mlir
+#include "mlir/IR/Value.h"  // TF:local_config_mlir
+#include "tensorflow/compiler/mlir/xla/ir/xla_ops.h.inc"
 
 using namespace mlir;
 using namespace mlir::XLA;
diff --git a/tensorflow/compiler/mlir/xla/ir/xla_ops.h b/tensorflow/compiler/mlir/xla/ir/xla_ops.h
index 9f82392cef5..d993558dc2d 100644
--- a/tensorflow/compiler/mlir/xla/ir/xla_ops.h
+++ b/tensorflow/compiler/mlir/xla/ir/xla_ops.h
@@ -18,14 +18,19 @@ limitations under the License.
 #ifndef TENSORFLOW_COMPILER_MLIR_XLA_IR_XLA_OPS_H_
 #define TENSORFLOW_COMPILER_MLIR_XLA_IR_XLA_OPS_H_
 
+#include "llvm/ADT/StringRef.h"
 #include "mlir/IR/Attributes.h"  // TF:local_config_mlir
 #include "mlir/IR/Dialect.h"  // TF:local_config_mlir
+#include "mlir/IR/Location.h"  // TF:local_config_mlir
+#include "mlir/IR/MLIRContext.h"  // TF:local_config_mlir
 #include "mlir/IR/OpDefinition.h"  // TF:local_config_mlir
+#include "mlir/IR/Operation.h"  // TF:local_config_mlir
 #include "mlir/IR/StandardTypes.h"  // TF:local_config_mlir
+#include "mlir/IR/Types.h"  // TF:local_config_mlir
 #include "mlir/Support/Functional.h"  // TF:local_config_mlir
 
 namespace mlir {
-class Builder;
+class OpBuilder;
 
 namespace XLA {