mlir-hlo/tools/mlir-tblgen-builder/Builder/OpImpl.h

#ifndef BUILDER_OPIMPL_
#define BUILDER_OPIMPL_

#include "Builder.h"
#include "llvm/Support/Casting.h"
#include "mlir/IR/Attributes.h"
#include "mlir/IR/Builders.h"
#include "mlir/IR/MLIRContext.h"
#include "mlir/IR/Operation.h"
#include "mlir/IR/Types.h"
#include "mlir/IR/Value.h"

namespace builder {

class Op::Impl {
 public:
  Impl() = default;
  void SetOperation(mlir::Operation *Op) { op_ = Op; }
  mlir::Value GetResult() { return op_->getResult(0); }

 private:
  mlir::Operation *op_;
};

}  // namespace builder

#endif