mlir::tt::ttnn::MemoryConfigAttr Class Reference

Inheritance diagram for mlir::tt::ttnn::MemoryConfigAttr:

Public Member Functions
llvm::ArrayRef< int64_t >	getShardShape (bool convertTileToScalar=true) const
MemoryConfigAttr	withBufferType (::mlir::MLIRContext *context, BufferType bufferType)
MemoryConfigAttr	withMemoryLayout (::mlir::MLIRContext *context, TensorMemoryLayout memLayout)
void	print (::mlir::AsmPrinter &odsPrinter) const
BufferTypeAttr	getBufferType () const
ShardSpecAttr	getShardSpec () const
TensorMemoryLayoutAttr	getTensorMemoryLayout () const

Static Public Member Functions
static MemoryConfigAttr	get (::mlir::MLIRContext *context, BufferTypeAttr bufferType, ShardSpecAttr shardSpec, TensorMemoryLayoutAttr tensorMemoryLayout)
static MemoryConfigAttr	getChecked (::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError, ::mlir::MLIRContext *context, BufferTypeAttr bufferType, ShardSpecAttr shardSpec, TensorMemoryLayoutAttr tensorMemoryLayout)
::llvm::LogicalResult	verify (::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError, BufferTypeAttr bufferType, ShardSpecAttr shardSpec, TensorMemoryLayoutAttr tensorMemoryLayout)
::llvm::LogicalResult	verifyInvariants (::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError, BufferTypeAttr bufferType, ShardSpecAttr shardSpec, TensorMemoryLayoutAttr tensorMemoryLayout)
static constexpr ::llvm::StringLiteral	getMnemonic ()
::mlir::Attribute	parse (::mlir::AsmParser &odsParser, ::mlir::Type odsType)

Static Public Attributes
static constexpr ::llvm::StringLiteral	name = "ttnn.memory_config"
static constexpr ::llvm::StringLiteral	dialectName = "ttnn"

Member Function Documentation

get()

MemoryConfigAttr mlir::tt::ttnn::MemoryConfigAttr::get ( ::mlir::MLIRContext *  context, BufferTypeAttr  bufferType, ShardSpecAttr  shardSpec, TensorMemoryLayoutAttr  tensorMemoryLayout  )

static

getBufferType()

BufferTypeAttr mlir::tt::ttnn::MemoryConfigAttr::getBufferType

(

)

const

getChecked()

MemoryConfigAttr mlir::tt::ttnn::MemoryConfigAttr::getChecked ( ::llvm::function_ref<::mlir::InFlightDiagnostic()>  emitError, ::mlir::MLIRContext *  context, BufferTypeAttr  bufferType, ShardSpecAttr  shardSpec, TensorMemoryLayoutAttr  tensorMemoryLayout  )

static

getMnemonic()

static constexpr ::llvm::StringLiteral mlir::tt::ttnn::MemoryConfigAttr::getMnemonic ( )

inlinestatic

getShardShape()

llvm::ArrayRef<int64_t> mlir::tt::ttnn::MemoryConfigAttr::getShardShape

(

bool

convertTileToScalar = true

)

const

getShardSpec()

ShardSpecAttr mlir::tt::ttnn::MemoryConfigAttr::getShardSpec

(

)

const

getTensorMemoryLayout()

TensorMemoryLayoutAttr mlir::tt::ttnn::MemoryConfigAttr::getTensorMemoryLayout

(

)

const

parse()

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

static

print()

void mlir::tt::ttnn::MemoryConfigAttr::print

(

::mlir::AsmPrinter &

odsPrinter

)

const

verify()

::llvm::LogicalResult mlir::tt::ttnn::MemoryConfigAttr::verify ( ::llvm::function_ref<::mlir::InFlightDiagnostic()>  emitError, BufferTypeAttr  bufferType, ShardSpecAttr  shardSpec, TensorMemoryLayoutAttr  tensorMemoryLayout  )

static

verifyInvariants()

llvm::LogicalResult mlir::tt::ttnn::MemoryConfigAttr::verifyInvariants ( ::llvm::function_ref<::mlir::InFlightDiagnostic()>  emitError, BufferTypeAttr  bufferType, ShardSpecAttr  shardSpec, TensorMemoryLayoutAttr  tensorMemoryLayout  )

static

withBufferType()

MemoryConfigAttr mlir::tt::ttnn::MemoryConfigAttr::withBufferType	(	::mlir::MLIRContext *	context,
		BufferType	bufferType
	)

withMemoryLayout()

MemoryConfigAttr mlir::tt::ttnn::MemoryConfigAttr::withMemoryLayout	(	::mlir::MLIRContext *	context,
		TensorMemoryLayout	memLayout
	)

Member Data Documentation

dialectName

constexpr ::llvm::StringLiteral mlir::tt::ttnn::MemoryConfigAttr::dialectName = "ttnn"

static

name

constexpr ::llvm::StringLiteral mlir::tt::ttnn::MemoryConfigAttr::name = "ttnn.memory_config"

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