MLIR  17.0.0git
Public Member Functions | List of all members
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...
BuildersetShape (ArrayRef< int64_t > newShape)
BuildersetElementType (Type newElementType)
BuildersetEncoding (Attribute newEncoding)
BuilderdropDim (unsigned pos)
 Erase a dim from shape @pos. More...
BuilderinsertDim (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)

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 

Build from scratch.

Definition at line 222 of file BuiltinTypes.h.

Member Function Documentation

◆ dropDim()

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

◆ insertDim()

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

Insert a val into shape @pos.

Definition at line 251 of file BuiltinTypes.h.

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

◆ operator RankedTensorType()

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

Definition at line 260 of file BuiltinTypes.h.

◆ setElementType()

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

Definition at line 230 of file BuiltinTypes.h.

◆ setEncoding()

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

Definition at line 235 of file BuiltinTypes.h.

◆ setShape()

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

Definition at line 225 of file BuiltinTypes.h.

Referenced by computeTransposedType(), and permuteShape().

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