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

Public Member Functions

void print (::mlir::AsmPrinter &odsPrinter) const
 
CoreRangeSetAttr getCoreRangeSet () const
 
ShapeAttr getShape () const
 
ShardOrientationAttr getShardOrientation () const
 
ShardModeAttr getShardMode () const
 
ShapeAttr getPhysicalShardShape () const
 

Static Public Member Functions

static CoreRangeSetAttr getCoreRangeSet (mlir::MLIRContext *context, GridAttr shardGrid, GridAttr deviceGrid)
 
static ShardSpecAttr get (::mlir::MLIRContext *context, CoreRangeSetAttr coreRangeSet, ShapeAttr shape, ShardOrientationAttr shardOrientation, ShardModeAttr shardMode, ShapeAttr physical_shard_shape)
 
static ShardSpecAttr get (::mlir::MLIRContext *context, ShapeAttr shape, GridAttr shardGrid, GridAttr deviceGrid)
 
static ShardSpecAttr get (::mlir::MLIRContext *context, TTNNLayoutAttr layout, GridAttr deviceGrid)
 
static constexpr ::llvm::StringLiteral getMnemonic ()
 
::mlir::Attribute parse (::mlir::AsmParser &odsParser, ::mlir::Type odsType)
 

Static Public Attributes

static constexpr ::llvm::StringLiteral name = "ttnn.shard_spec"
 
static constexpr ::llvm::StringLiteral dialectName = "ttnn"
 

Member Function Documentation

◆ get() [1/3]

ShardSpecAttr mlir::tt::ttnn::ShardSpecAttr::get ( ::mlir::MLIRContext *  context,
CoreRangeSetAttr  coreRangeSet,
ShapeAttr  shape,
ShardOrientationAttr  shardOrientation,
ShardModeAttr  shardMode,
ShapeAttr  physical_shard_shape 
)
static

◆ get() [2/3]

ShardSpecAttr mlir::tt::ttnn::ShardSpecAttr::get ( ::mlir::MLIRContext *  context,
ShapeAttr  shape,
GridAttr  shardGrid,
GridAttr  deviceGrid 
)
static

◆ get() [3/3]

ShardSpecAttr mlir::tt::ttnn::ShardSpecAttr::get ( ::mlir::MLIRContext *  context,
TTNNLayoutAttr  layout,
GridAttr  deviceGrid 
)
static

◆ getCoreRangeSet() [1/2]

CoreRangeSetAttr mlir::tt::ttnn::ShardSpecAttr::getCoreRangeSet ( ) const

◆ getCoreRangeSet() [2/2]

static CoreRangeSetAttr mlir::tt::ttnn::ShardSpecAttr::getCoreRangeSet ( mlir::MLIRContext *  context,
GridAttr  shardGrid,
GridAttr  deviceGrid 
)
static

◆ getMnemonic()

static constexpr ::llvm::StringLiteral mlir::tt::ttnn::ShardSpecAttr::getMnemonic ( )
inlinestatic

◆ getPhysicalShardShape()

ShapeAttr mlir::tt::ttnn::ShardSpecAttr::getPhysicalShardShape ( ) const

◆ getShape()

ShapeAttr mlir::tt::ttnn::ShardSpecAttr::getShape ( ) const

◆ getShardMode()

ShardModeAttr mlir::tt::ttnn::ShardSpecAttr::getShardMode ( ) const

◆ getShardOrientation()

ShardOrientationAttr mlir::tt::ttnn::ShardSpecAttr::getShardOrientation ( ) const

◆ parse()

mlir::Attribute mlir::tt::ttnn::ShardSpecAttr::parse ( ::mlir::AsmParser &  odsParser,
::mlir::Type  odsType 
)
static

◆ print()

void mlir::tt::ttnn::ShardSpecAttr::print ( ::mlir::AsmPrinter &  odsPrinter) const

Member Data Documentation

◆ dialectName

constexpr ::llvm::StringLiteral mlir::tt::ttnn::ShardSpecAttr::dialectName = "ttnn"
static

◆ name

constexpr ::llvm::StringLiteral mlir::tt::ttnn::ShardSpecAttr::name = "ttnn.shard_spec"
static

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