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

Public Member Functions

void print (::mlir::AsmPrinter &odsPrinter) const
 
int64_t getShardShape () const
 
::llvm::ArrayRef< int64_t > getAxes () const
 

Static Public Member Functions

static TensorMeshShardingAxisAttr get (::mlir::MLIRContext *context, int64_t shard_shape, ::llvm::ArrayRef< int64_t > axes)
 
static constexpr ::llvm::StringLiteral getMnemonic ()
 
::mlir::Attribute parse (::mlir::AsmParser &odsParser, ::mlir::Type odsType)
 

Static Public Attributes

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

Member Function Documentation

◆ get()

TensorMeshShardingAxisAttr mlir::tt::TensorMeshShardingAxisAttr::get ( ::mlir::MLIRContext *  context,
int64_t  shard_shape,
::llvm::ArrayRef< int64_t >  axes 
)
static

◆ getAxes()

llvm::ArrayRef< int64_t > mlir::tt::TensorMeshShardingAxisAttr::getAxes ( ) const

◆ getMnemonic()

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

◆ getShardShape()

int64_t mlir::tt::TensorMeshShardingAxisAttr::getShardShape ( ) const

◆ parse()

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

◆ print()

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

Member Data Documentation

◆ dialectName

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

◆ name

constexpr ::llvm::StringLiteral mlir::tt::TensorMeshShardingAxisAttr::name = "tt.tensor_sharding"
static

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