mlir::tt::ttnn::detail::Conv2dConfigAttrStorage Struct Reference

Inheritance diagram for mlir::tt::ttnn::detail::Conv2dConfigAttrStorage:

Public Types
using	KeyTy = std::tuple< DataType, DataType, StringAttr, uint32_t, bool, bool, uint32_t, uint32_t, bool, bool, TensorMemoryLayoutAttr, CoreRangeSetAttr, bool, Layout, bool, bool, bool, bool, bool, bool >

Public Member Functions
	Conv2dConfigAttrStorage (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)
KeyTy	getAsKey () const
bool	operator== (const KeyTy &tblgenKey) const

Static Public Member Functions
::llvm::hash_code	hashKey (const KeyTy &tblgenKey)
static Conv2dConfigAttrStorage *	construct (::mlir::AttributeStorageAllocator &allocator, KeyTy &&tblgenKey)

Public Attributes
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

Member Typedef Documentation

KeyTy

using mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::KeyTy = std::tuple<DataType, DataType, StringAttr, uint32_t, bool, bool, uint32_t, uint32_t, bool, bool, TensorMemoryLayoutAttr, CoreRangeSetAttr, bool, Layout, bool, bool, bool, bool, bool, bool>

Constructor & Destructor Documentation

Conv2dConfigAttrStorage()

mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::Conv2dConfigAttrStorage ( 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  )

inline

Member Function Documentation

construct()

static Conv2dConfigAttrStorage* mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::construct ( ::mlir::AttributeStorageAllocator &  allocator, KeyTy &&  tblgenKey  )

inlinestatic

getAsKey()

KeyTy mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::getAsKey ( ) const

inline

hashKey()

::llvm::hash_code mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::hashKey ( const KeyTy &  tblgenKey )

inlinestatic

operator==()

bool mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::operator== ( const KeyTy &  tblgenKey ) const

inline

Member Data Documentation

act_block_h_override

uint32_t mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::act_block_h_override

act_block_w_div

uint32_t mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::act_block_w_div

activation

StringAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::activation

always_preprocess_weights

bool mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::always_preprocess_weights

core_grid

CoreRangeSetAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::core_grid

deallocate_activation

bool mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::deallocate_activation

dtype

DataType mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::dtype

enable_act_double_buffer

bool mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::enable_act_double_buffer

enable_split_reader

bool mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::enable_split_reader

enable_subblock_padding

bool mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::enable_subblock_padding

enable_weights_double_buffer

bool mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::enable_weights_double_buffer

input_channels_alignment

uint32_t mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::input_channels_alignment

output_layout

Layout mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::output_layout

override_sharding_config

bool mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::override_sharding_config

preprocess_weights_on_device

bool mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::preprocess_weights_on_device

reallocate_halo_output

bool mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::reallocate_halo_output

reshard_if_not_optimal

bool mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::reshard_if_not_optimal

shard_layout

TensorMemoryLayoutAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::shard_layout

transpose_shards

bool mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::transpose_shards

weights_dtype

DataType mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::weights_dtype

---

The documentation for this struct was generated from the following file:

• /__w/tt-mlir/tt-mlir/build/include/ttmlir/Dialect/TTNN/IR/TTNNOpsAttrDefs.cpp.inc