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< DataType >	getDtype () const
std::optional< DataType >	getWeightsDtype () 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< TensorMemoryLayout >	getShardLayout () const
CoreRangeSetAttr	getCoreGrid () const
BoolAttr	getTransposeShards () const
std::optional< Layout >	getOutputLayout () 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:

• /__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