#include "Tensor.h" #include "Shape.h" #include "TensorImpl.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" namespace builder { // Tensor::Tensor() : impl_(std::make_shared()) {} Tensor::Tensor(Shape& shape, PrimitiveType& primitiveType) : impl_(std::make_shared(shape, primitiveType)) {} Shape Tensor::GetShape() { return impl_->GetShape(); } PrimitiveType Tensor::GetType() { return impl_->GetType(); } } // namespace builder