mlir::tt::ttnn::Conv2dConfigAttr Class Reference

Inheritance diagram for mlir::tt::ttnn::Conv2dConfigAttr:

Public Member Functions
void	print (::mlir::AsmPrinter &odsPrinter) const
DataType	getDtype () const
DataType	getWeightsDtype () const
StringAttr	getActivation () const
uint32_t	getInputChannelsAlignment () const
bool	getDeallocateActivation () const
bool	getReallocateHaloOutput () const
uint32_t	getActBlockHOverride () const
uint32_t	getActBlockWDiv () const
bool	getReshardIfNotOptimal () const
bool	getOverrideShardingConfig () const
TensorMemoryLayoutAttr	getShardLayout () const
CoreRangeSetAttr	getCoreGrid () const
bool	getTransposeShards () const
Layout	getOutputLayout () const
bool	getPreprocessWeightsOnDevice () const
bool	getAlwaysPreprocessWeights () const
bool	getEnableActDoubleBuffer () const
bool	getEnableWeightsDoubleBuffer () const
bool	getEnableSplitReader () const
bool	getEnableSubblockPadding () const

Static Public Member Functions
static Conv2dConfigAttr	get (::mlir::MLIRContext *context, DataType dtype, DataType weights_dtype, StringAttr activation, uint32_t input_channels_alignment, bool deallocate_activation, bool reallocate_halo_output, uint32_t act_block_h_override, uint32_t act_block_w_div, bool reshard_if_not_optimal, bool override_sharding_config, TensorMemoryLayoutAttr shard_layout, CoreRangeSetAttr core_grid, bool transpose_shards, Layout output_layout, bool preprocess_weights_on_device, bool always_preprocess_weights, bool enable_act_double_buffer, bool enable_weights_double_buffer, bool enable_split_reader, bool enable_subblock_padding)
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()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::get ( ::mlir::MLIRContext *  context, DataType  dtype, DataType  weights_dtype, StringAttr  activation, uint32_t  input_channels_alignment, bool  deallocate_activation, bool  reallocate_halo_output, uint32_t  act_block_h_override, uint32_t  act_block_w_div, bool  reshard_if_not_optimal, bool  override_sharding_config, TensorMemoryLayoutAttr  shard_layout, CoreRangeSetAttr  core_grid, bool  transpose_shards, Layout  output_layout, bool  preprocess_weights_on_device, bool  always_preprocess_weights, bool  enable_act_double_buffer, bool  enable_weights_double_buffer, bool  enable_split_reader, bool  enable_subblock_padding  )

static

getActBlockHOverride()

uint32_t mlir::tt::ttnn::Conv2dConfigAttr::getActBlockHOverride

(

)

const

getActBlockWDiv()

uint32_t mlir::tt::ttnn::Conv2dConfigAttr::getActBlockWDiv

(

)

const

getActivation()

StringAttr mlir::tt::ttnn::Conv2dConfigAttr::getActivation

(

)

const

getAlwaysPreprocessWeights()

bool mlir::tt::ttnn::Conv2dConfigAttr::getAlwaysPreprocessWeights

(

)

const

getCoreGrid()

CoreRangeSetAttr mlir::tt::ttnn::Conv2dConfigAttr::getCoreGrid

(

)

const

getDeallocateActivation()

bool mlir::tt::ttnn::Conv2dConfigAttr::getDeallocateActivation

(

)

const

getDtype()

DataType mlir::tt::ttnn::Conv2dConfigAttr::getDtype

(

)

const

getEnableActDoubleBuffer()

bool mlir::tt::ttnn::Conv2dConfigAttr::getEnableActDoubleBuffer

(

)

const

getEnableSplitReader()

bool mlir::tt::ttnn::Conv2dConfigAttr::getEnableSplitReader

(

)

const

getEnableSubblockPadding()

bool mlir::tt::ttnn::Conv2dConfigAttr::getEnableSubblockPadding

(

)

const

getEnableWeightsDoubleBuffer()

bool mlir::tt::ttnn::Conv2dConfigAttr::getEnableWeightsDoubleBuffer

(

)

const

getInputChannelsAlignment()

uint32_t mlir::tt::ttnn::Conv2dConfigAttr::getInputChannelsAlignment

(

)

const

getMnemonic()

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

inlinestatic

getOutputLayout()

Layout mlir::tt::ttnn::Conv2dConfigAttr::getOutputLayout

(

)

const

getOverrideShardingConfig()

bool mlir::tt::ttnn::Conv2dConfigAttr::getOverrideShardingConfig

(

)

const

getPreprocessWeightsOnDevice()

bool mlir::tt::ttnn::Conv2dConfigAttr::getPreprocessWeightsOnDevice

(

)

const

getReallocateHaloOutput()

bool mlir::tt::ttnn::Conv2dConfigAttr::getReallocateHaloOutput

(

)

const

getReshardIfNotOptimal()

bool mlir::tt::ttnn::Conv2dConfigAttr::getReshardIfNotOptimal

(

)

const

getShardLayout()

TensorMemoryLayoutAttr mlir::tt::ttnn::Conv2dConfigAttr::getShardLayout

(

)

const

getTransposeShards()

bool mlir::tt::ttnn::Conv2dConfigAttr::getTransposeShards

(

)

const

getWeightsDtype()

DataType mlir::tt::ttnn::Conv2dConfigAttr::getWeightsDtype

(

)

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

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