MLIR  16.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 . More...
BuilderinsertDim (int64_t val, unsigned pos)
 Insert a val into shape . 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 212 of file BuiltinTypes.h.

Constructor & Destructor Documentation

◆ Builder() [1/2]

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

Build from another RankedTensorType.

Definition at line 215 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 220 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 .

Definition at line 249 of file BuiltinTypes.h.

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

◆ operator RankedTensorType()

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

Definition at line 258 of file BuiltinTypes.h.

◆ setElementType()

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

Definition at line 228 of file BuiltinTypes.h.

◆ setEncoding()

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

Definition at line 233 of file BuiltinTypes.h.

◆ setShape()

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

Definition at line 223 of file BuiltinTypes.h.

Referenced by computeTransposedType(), and joinShapes().

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