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

Public Member Functions

Conv2dConfigAttr withActivation (StringRef activation) const
 
Conv2dConfigAttr withDtype (DataType dtype) const
 
Conv2dConfigAttr withWeightsDtype (DataType dtype) const
 
Conv2dConfigAttr withDeallocateActivation (bool value) const
 
Conv2dConfigAttr withReallocateHaloOutput (bool value) const
 
Conv2dConfigAttr withActBlockHOverride (uint32_t value) const
 
Conv2dConfigAttr withActBlockWDiv (uint32_t value) const
 
Conv2dConfigAttr withReshardIfNotOptimal (bool value) const
 
Conv2dConfigAttr withOverrideShardingConfig (bool value) const
 
Conv2dConfigAttr withShardLayout (TensorMemoryLayout layout) const
 
Conv2dConfigAttr withCoreGrid (CoreRangeSetAttr grid) const
 
Conv2dConfigAttr withTransposeShards (bool value) const
 
Conv2dConfigAttr withOutputLayout (Layout layout) const
 
Conv2dConfigAttr withPreprocessWeightsOnDevice (bool value) const
 
Conv2dConfigAttr withAlwaysPreprocessWeights (bool value) const
 
Conv2dConfigAttr withEnableActDoubleBuffer (bool value) const
 
Conv2dConfigAttr withEnableWeightsDoubleBuffer (bool value) const
 
Conv2dConfigAttr withEnableSplitReader (bool value) const
 
Conv2dConfigAttr withEnableSubblockPadding (bool value) const
 
bool hasActivation () const
 
void print (::mlir::AsmPrinter &odsPrinter) const
 
std::optional< DataTypegetDtype () const
 
std::optional< DataTypegetWeightsDtype () const
 
StringAttr getActivation () const
 
BoolAttr getDeallocateActivation () const
 
BoolAttr getReallocateHaloOutput () const
 
std::optional< uint32_t > getActBlockHOverride () const
 
std::optional< uint32_t > getActBlockWDiv () const
 
BoolAttr getReshardIfNotOptimal () const
 
BoolAttr getOverrideShardingConfig () const
 
std::optional< TensorMemoryLayoutgetShardLayout () const
 
CoreRangeSetAttr getCoreGrid () const
 
BoolAttr getTransposeShards () const
 
std::optional< LayoutgetOutputLayout () const
 
BoolAttr getPreprocessWeightsOnDevice () const
 
BoolAttr getAlwaysPreprocessWeights () const
 
BoolAttr getEnableActDoubleBuffer () const
 
BoolAttr getEnableWeightsDoubleBuffer () const
 
BoolAttr getEnableSplitReader () const
 
BoolAttr getEnableSubblockPadding () const
 

Static Public Member Functions

static Conv2dConfigAttr get (::mlir::MLIRContext *context, std::optional< DataType > dtype, std::optional< DataType > weights_dtype, StringAttr activation, BoolAttr deallocate_activation, BoolAttr reallocate_halo_output, std::optional< uint32_t > act_block_h_override, std::optional< uint32_t > act_block_w_div, BoolAttr reshard_if_not_optimal, BoolAttr override_sharding_config, std::optional< TensorMemoryLayout > shard_layout, CoreRangeSetAttr core_grid, BoolAttr transpose_shards, std::optional< Layout > output_layout, BoolAttr preprocess_weights_on_device, BoolAttr always_preprocess_weights, BoolAttr enable_act_double_buffer, BoolAttr enable_weights_double_buffer, BoolAttr enable_split_reader, BoolAttr enable_subblock_padding)
 
static Conv2dConfigAttr get (::mlir::MLIRContext *context)
 
static constexpr ::llvm::StringLiteral getMnemonic ()
 
::mlir::Attribute parse (::mlir::AsmParser &odsParser, ::mlir::Type odsType)
 

Static Public Attributes

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

Member Function Documentation

◆ get() [1/2]

static Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::get ( ::mlir::MLIRContext *  context)
static

◆ get() [2/2]

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::get ( ::mlir::MLIRContext *  context,
std::optional< DataType dtype,
std::optional< DataType weights_dtype,
StringAttr  activation,
BoolAttr  deallocate_activation,
BoolAttr  reallocate_halo_output,
std::optional< uint32_t >  act_block_h_override,
std::optional< uint32_t >  act_block_w_div,
BoolAttr  reshard_if_not_optimal,
BoolAttr  override_sharding_config,
std::optional< TensorMemoryLayout shard_layout,
CoreRangeSetAttr  core_grid,
BoolAttr  transpose_shards,
std::optional< Layout output_layout,
BoolAttr  preprocess_weights_on_device,
BoolAttr  always_preprocess_weights,
BoolAttr  enable_act_double_buffer,
BoolAttr  enable_weights_double_buffer,
BoolAttr  enable_split_reader,
BoolAttr  enable_subblock_padding 
)
static

◆ getActBlockHOverride()

std::optional< uint32_t > mlir::tt::ttnn::Conv2dConfigAttr::getActBlockHOverride ( ) const

◆ getActBlockWDiv()

std::optional< uint32_t > mlir::tt::ttnn::Conv2dConfigAttr::getActBlockWDiv ( ) const

◆ getActivation()

StringAttr mlir::tt::ttnn::Conv2dConfigAttr::getActivation ( ) const

◆ getAlwaysPreprocessWeights()

BoolAttr mlir::tt::ttnn::Conv2dConfigAttr::getAlwaysPreprocessWeights ( ) const

◆ getCoreGrid()

CoreRangeSetAttr mlir::tt::ttnn::Conv2dConfigAttr::getCoreGrid ( ) const

◆ getDeallocateActivation()

BoolAttr mlir::tt::ttnn::Conv2dConfigAttr::getDeallocateActivation ( ) const

◆ getDtype()

std::optional< DataType > mlir::tt::ttnn::Conv2dConfigAttr::getDtype ( ) const

◆ getEnableActDoubleBuffer()

BoolAttr mlir::tt::ttnn::Conv2dConfigAttr::getEnableActDoubleBuffer ( ) const

◆ getEnableSplitReader()

BoolAttr mlir::tt::ttnn::Conv2dConfigAttr::getEnableSplitReader ( ) const

◆ getEnableSubblockPadding()

BoolAttr mlir::tt::ttnn::Conv2dConfigAttr::getEnableSubblockPadding ( ) const

◆ getEnableWeightsDoubleBuffer()

BoolAttr mlir::tt::ttnn::Conv2dConfigAttr::getEnableWeightsDoubleBuffer ( ) const

◆ getMnemonic()

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

◆ getOutputLayout()

std::optional< Layout > mlir::tt::ttnn::Conv2dConfigAttr::getOutputLayout ( ) const

◆ getOverrideShardingConfig()

BoolAttr mlir::tt::ttnn::Conv2dConfigAttr::getOverrideShardingConfig ( ) const

◆ getPreprocessWeightsOnDevice()

BoolAttr mlir::tt::ttnn::Conv2dConfigAttr::getPreprocessWeightsOnDevice ( ) const

◆ getReallocateHaloOutput()

BoolAttr mlir::tt::ttnn::Conv2dConfigAttr::getReallocateHaloOutput ( ) const

◆ getReshardIfNotOptimal()

BoolAttr mlir::tt::ttnn::Conv2dConfigAttr::getReshardIfNotOptimal ( ) const

◆ getShardLayout()

std::optional< TensorMemoryLayout > mlir::tt::ttnn::Conv2dConfigAttr::getShardLayout ( ) const

◆ getTransposeShards()

BoolAttr mlir::tt::ttnn::Conv2dConfigAttr::getTransposeShards ( ) const

◆ getWeightsDtype()

std::optional< DataType > mlir::tt::ttnn::Conv2dConfigAttr::getWeightsDtype ( ) const

◆ hasActivation()

bool mlir::tt::ttnn::Conv2dConfigAttr::hasActivation ( ) const

◆ parse()

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

◆ print()

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

◆ withActBlockHOverride()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withActBlockHOverride ( uint32_t  value) const

◆ withActBlockWDiv()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withActBlockWDiv ( uint32_t  value) const

◆ withActivation()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withActivation ( StringRef  activation) const

◆ withAlwaysPreprocessWeights()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withAlwaysPreprocessWeights ( bool  value) const

◆ withCoreGrid()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withCoreGrid ( CoreRangeSetAttr  grid) const

◆ withDeallocateActivation()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withDeallocateActivation ( bool  value) const

◆ withDtype()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withDtype ( DataType  dtype) const

◆ withEnableActDoubleBuffer()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withEnableActDoubleBuffer ( bool  value) const

◆ withEnableSplitReader()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withEnableSplitReader ( bool  value) const

◆ withEnableSubblockPadding()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withEnableSubblockPadding ( bool  value) const

◆ withEnableWeightsDoubleBuffer()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withEnableWeightsDoubleBuffer ( bool  value) const

◆ withOutputLayout()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withOutputLayout ( Layout  layout) const

◆ withOverrideShardingConfig()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withOverrideShardingConfig ( bool  value) const

◆ withPreprocessWeightsOnDevice()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withPreprocessWeightsOnDevice ( bool  value) const

◆ withReallocateHaloOutput()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withReallocateHaloOutput ( bool  value) const

◆ withReshardIfNotOptimal()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withReshardIfNotOptimal ( bool  value) const

◆ withShardLayout()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withShardLayout ( TensorMemoryLayout  layout) const

◆ withTransposeShards()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withTransposeShards ( bool  value) const

◆ withWeightsDtype()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withWeightsDtype ( DataType  dtype) const

Member Data Documentation

◆ dialectName

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

◆ name

constexpr ::llvm::StringLiteral mlir::tt::ttnn::Conv2dConfigAttr::name = "ttnn.conv2d_config"
static

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