XLA: IWYU on op definitions

Adding missing headers that are used in these files

PiperOrigin-RevId: 263229994
This commit is contained in:
A. Unique TensorFlower 2019-08-13 15:35:12 -07:00 committed by TensorFlower Gardener
parent 638af77f8a
commit 64b01d5574
2 changed files with 26 additions and 1 deletions

View File

@ -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;

View File

@ -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 {