mlir::RankedTensorType::Builder Class Reference

This is a builder type that keeps local references to arguments. More...

#include "mlir/IR/BuiltinTypes.h"

Public Member Functions
	Builder (RankedTensorType other)
	Build from another RankedTensorType. More...
	Builder (ArrayRef< int64_t > shape, Type elementType, Attribute encoding)
	Build from scratch. More...
Builder &	setShape (ArrayRef< int64_t > newShape)
Builder &	setElementType (Type newElementType)
Builder &	setEncoding (Attribute newEncoding)
Builder &	dropDim (unsigned pos)
	Erase a dim from shape @pos. More...
Builder &	insertDim (int64_t val, unsigned pos)
	Insert a val into shape @pos. More...
	operator RankedTensorType ()

Detailed Description

This is a builder type that keeps local references to arguments.

Arguments that are passed into the builder must outlive the builder.

Definition at line 214 of file BuiltinTypes.h.

Constructor & Destructor Documentation

Builder() [1/2]

mlir::RankedTensorType::Builder::Builder ( RankedTensorType  other )

inlineexplicit

Build from another RankedTensorType.

Definition at line 217 of file BuiltinTypes.h.

Builder() [2/2]

mlir::RankedTensorType::Builder::Builder ( ArrayRef< int64_t >  shape, Type  elementType, Attribute  encoding  )

inline

Build from scratch.

Definition at line 222 of file BuiltinTypes.h.

Member Function Documentation

dropDim()

Builder& mlir::RankedTensorType::Builder::dropDim ( unsigned  pos )

inline

Erase a dim from shape @pos.

Definition at line 241 of file BuiltinTypes.h.

References mlir::CopyOnWriteArrayRef< T >::erase(), and mlir::CopyOnWriteArrayRef< T >::size().

Referenced by mlir::linalg::DownscaleSizeOneWindowed2DConvolution< Conv2DOp, Conv1DOp >::returningMatchAndRewrite(), mlir::linalg::DownscaleConv2DOp::returningMatchAndRewrite(), and mlir::linalg::DownscaleDepthwiseConv2DNhwcHwcOp::returningMatchAndRewrite().

insertDim()

Builder& mlir::RankedTensorType::Builder::insertDim ( int64_t  val, unsigned  pos  )

inline

Insert a val into shape @pos.

Definition at line 248 of file BuiltinTypes.h.

References mlir::CopyOnWriteArrayRef< T >::insert(), and mlir::CopyOnWriteArrayRef< T >::size().

Referenced by mlir::linalg::splitReductionByScaling().

operator RankedTensorType()

mlir::RankedTensorType::Builder::operator RankedTensorType ( )

inline

Definition at line 254 of file BuiltinTypes.h.

References mlir::get().

setElementType()

Builder& mlir::RankedTensorType::Builder::setElementType ( Type  newElementType )

inline

Definition at line 230 of file BuiltinTypes.h.

setEncoding()

Builder& mlir::RankedTensorType::Builder::setEncoding ( Attribute  newEncoding )

inline

Definition at line 235 of file BuiltinTypes.h.

setShape()

Builder& mlir::RankedTensorType::Builder::setShape ( ArrayRef< int64_t >  newShape )

inline

Definition at line 225 of file BuiltinTypes.h.

Referenced by mlir::tensor::computeTransposedType(), mlir::linalg::lowerPack(), mlir::linalg::lowerUnPack(), and permuteShape().

---

The documentation for this class was generated from the following file:

• include/mlir/IR/BuiltinTypes.h