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

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

Public Member Functions
Conv2dConfigAttr	withActivation (StringRef activation) 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

withActivation()

Conv2dConfigAttr mlir::tt::ttnn::Conv2dConfigAttr::withActivation

(

StringRef

activation

)

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