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:

• /__w/tt-mlir/tt-mlir/build/include/ttmlir/Dialect/TTNN/IR/TTNNOpsAttrDefs.h.inc
• /__w/tt-mlir/tt-mlir/build/include/ttmlir/Dialect/TTNN/IR/TTNNOpsAttrDefs.cpp.inc