TT-MLIR
Public Member Functions | Static Public Member Functions | Static Public Attributes | List of all members
mlir::tt::TileType Class Reference
Inheritance diagram for mlir::tt::TileType:

Public Member Functions

SmallVector< int64_t > getScalarShape (SmallVector< int64_t > tiledShape) const
 
SmallVector< int64_t > getTiledShape (SmallVector< int64_t > scalarShape) const
 
uint64_t getSizeBytes () const
 
int64_t getHeight () const
 
int64_t getWidth () const
 
Type getElementType () const
 
void print (::mlir::AsmPrinter &odsPrinter) const
 
::llvm::ArrayRef< int64_t > getShape () const
 
DataType getDataType () const
 

Static Public Member Functions

static TileType get (::mlir::MLIRContext *context, Type elementType, ArrayRef< int64_t > shape={32, 32})
 
static TileType get (::mlir::MLIRContext *context, ::llvm::ArrayRef< int64_t > shape, DataType dataType)
 
static TileType getChecked (::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError, ::mlir::MLIRContext *context, ::llvm::ArrayRef< int64_t > shape, DataType dataType)
 
::llvm::LogicalResult verify (::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError, ::llvm::ArrayRef< int64_t > shape, DataType dataType)
 
static constexpr ::llvm::StringLiteral getMnemonic ()
 
::mlir::Type parse (::mlir::AsmParser &odsParser)
 

Static Public Attributes

static constexpr ::llvm::StringLiteral name = "tt.tile"
 
static constexpr ::llvm::StringLiteral dialectName = "tt"
 

Member Function Documentation

◆ get() [1/2]

TileType mlir::tt::TileType::get ( ::mlir::MLIRContext *  context,
::llvm::ArrayRef< int64_t >  shape,
DataType  dataType 
)
static

◆ get() [2/2]

static TileType mlir::tt::TileType::get ( ::mlir::MLIRContext *  context,
Type  elementType,
ArrayRef< int64_t >  shape = {32, 32} 
)
static

◆ getChecked()

TileType mlir::tt::TileType::getChecked ( ::llvm::function_ref<::mlir::InFlightDiagnostic()>  emitError,
::mlir::MLIRContext *  context,
::llvm::ArrayRef< int64_t >  shape,
DataType  dataType 
)
static

◆ getDataType()

DataType mlir::tt::TileType::getDataType ( ) const

◆ getElementType()

Type mlir::tt::TileType::getElementType ( ) const

◆ getHeight()

int64_t mlir::tt::TileType::getHeight ( ) const
inline

◆ getMnemonic()

static constexpr ::llvm::StringLiteral mlir::tt::TileType::getMnemonic ( )
inlinestatic

◆ getScalarShape()

SmallVector<int64_t> mlir::tt::TileType::getScalarShape ( SmallVector< int64_t >  tiledShape) const

◆ getShape()

llvm::ArrayRef< int64_t > mlir::tt::TileType::getShape ( ) const

◆ getSizeBytes()

uint64_t mlir::tt::TileType::getSizeBytes ( ) const

◆ getTiledShape()

SmallVector<int64_t> mlir::tt::TileType::getTiledShape ( SmallVector< int64_t >  scalarShape) const

◆ getWidth()

int64_t mlir::tt::TileType::getWidth ( ) const
inline

◆ parse()

mlir::Type mlir::tt::TileType::parse ( ::mlir::AsmParser &  odsParser)
static

◆ print()

void mlir::tt::TileType::print ( ::mlir::AsmPrinter &  odsPrinter) const

◆ verify()

::llvm::LogicalResult mlir::tt::TileType::verify ( ::llvm::function_ref<::mlir::InFlightDiagnostic()>  emitError,
::llvm::ArrayRef< int64_t >  shape,
DataType  dataType 
)
static

Member Data Documentation

◆ dialectName

constexpr ::llvm::StringLiteral mlir::tt::TileType::dialectName = "tt"
static

◆ name

constexpr ::llvm::StringLiteral mlir::tt::TileType::name = "tt.tile"
static

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