23 lines
706 B
C++
23 lines
706 B
C++
#include "tools/mlir-tblgen-builder/Builder/Builder.h"
|
|
|
|
int main() {
|
|
builder::Builder builder;
|
|
|
|
builder::Shape shape({10, 10});
|
|
auto pType = builder::PrimitiveType::F32();
|
|
builder::Type OutType(shape, pType);
|
|
|
|
std::vector<float> data(100);
|
|
builder::Tensor tensor(shape, data);
|
|
|
|
auto in1 = builder.CreateInput(OutType);
|
|
auto op1 = builder::mhlo::ConstOp::build(builder, OutType, tensor);
|
|
auto op2 = builder::mhlo::ConstOp::build(builder, OutType, tensor);
|
|
auto op3 = builder::mhlo::AddOp::build(builder, OutType, op1, op2);
|
|
auto op4 = builder::mhlo::MulOp::build(builder, OutType, op3, in1);
|
|
|
|
builder.SetOutput(std::vector<builder::Op>({op4}));
|
|
builder.DumpModule();
|
|
return 0;
|
|
}
|