#include "tensorflow/compiler/mlir/tfrt/ir/mlrt/tf_mlrt_ops.h"

namespace tensorflow {
namespace tf_mlrt_tpu {

TensorflowMlrtTpuDialect::TensorflowMlrtTpuDialect(mlir::MLIRContext *context)
    : mlir::Dialect(/*name=*/"tf_mlrt_tpu", context,
                    mlir::TypeID::get<TensorflowMlrtTpuDialect>()) {
  addOperations<
#define GET_OP_LIST
#include "tensorflow/compiler/mlir/tfrt/ir/mlrt/tf_mlrt_tpu_ops.cpp.inc"
      >();

#include "mlir/Interfaces/SideEffectInterfaces.h"  // from @llvm-project

namespace tensorflow {
namespace tf_mlrt_tpu {

class TensorflowMlrtTpuDialect : public mlir::Dialect {
 public:
  explicit TensorflowMlrtTpuDialect(mlir::MLIRContext *context);
  static llvm::StringRef getDialectNamespace() { return "tf_mlrt_tpu"; }
};

}  // namespace tf_mlrt_tpu
}  // namespace tensorflow