37 lines
977 B
C++
37 lines
977 B
C++
#include "Builder.h"
|
|
|
|
#include "BuilderImpl.h"
|
|
#include "llvm/Support/Casting.h"
|
|
// #include "mlir/Dialect/StandardOps/Ops.h"
|
|
// #include "mlir/IR/Attributes.h"
|
|
// #include "mlir/IR/Operation.h"
|
|
// #include "mlir/IR/StandardTypes.h"
|
|
// #include "mlir/IR/Types.h"
|
|
// #include "mlir/IR/Value.h"
|
|
#include "Attribute.h"
|
|
#include "AttributeImpl.h"
|
|
#include "Op.h"
|
|
#include "OpImpl.h"
|
|
#include "mlir-hlo/Dialect/mhlo/IR/hlo_ops.h"
|
|
#include "mlir-hlo/Dialect/mhlo/transforms/PassDetail.h"
|
|
#include "mlir-hlo/Dialect/mhlo/transforms/passes.h"
|
|
|
|
namespace builder {
|
|
|
|
Builder::Builder() : impl_(std::make_shared<Impl>()) {}
|
|
|
|
builder::Op Builder::CreateInput(const builder::Type& type) {
|
|
return impl_->CreateInput(type);
|
|
}
|
|
|
|
void Builder::SetOutput(const std::vector<builder::Op>& outputs) {
|
|
impl_->SetOutput(outputs);
|
|
}
|
|
|
|
void Builder::DumpModule() { impl_->DumpModule(); }
|
|
|
|
} // namespace builder
|
|
|
|
#define GET_OP_CLASSES
|
|
#include "include/mlir-hlo/Dialect/mhlo/IR/hlo_builder.cc.inc"
|