22 lines
650 B
C++
22 lines
650 B
C++
|
#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<Impl>()) {}
|
||
|
|
Tensor::Tensor(Shape& shape, PrimitiveType& primitiveType)
|
||
|
|
: impl_(std::make_shared<Impl>(shape, primitiveType)) {}
|
||
|
|
|
||
|
|
Shape Tensor::GetShape() { return impl_->GetShape(); }
|
||
|
|
PrimitiveType Tensor::GetType() { return impl_->GetType(); }
|
||
|
|
|
||
|
|
} // namespace builder
|