TT-MLIR
Public Types | Public Member Functions | Static Public Member Functions | Public Attributes | List of all members
mlir::tt::ttnn::detail::Conv2dConfigAttrStorage Struct Reference
Inheritance diagram for mlir::tt::ttnn::detail::Conv2dConfigAttrStorage:

Public Types

using KeyTy = std::tuple< std::optional< DataType >, std::optional< DataType >, StringAttr, BoolAttr, BoolAttr, std::optional< uint32_t >, std::optional< uint32_t >, BoolAttr, BoolAttr, std::optional< TensorMemoryLayout >, CoreRangeSetAttr, BoolAttr, std::optional< Layout >, BoolAttr, BoolAttr, BoolAttr, BoolAttr, BoolAttr, BoolAttr >
 

Public Member Functions

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

Static Public Member Functions

::llvm::hash_code hashKey (const KeyTy &tblgenKey)
 
static Conv2dConfigAttrStorageconstruct (::mlir::AttributeStorageAllocator &allocator, KeyTy &&tblgenKey)
 

Public Attributes

std::optional< DataTypedtype
 
std::optional< DataTypeweights_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< TensorMemoryLayoutshard_layout
 
CoreRangeSetAttr core_grid
 
BoolAttr transpose_shards
 
std::optional< Layoutoutput_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
 

Member Typedef Documentation

◆ KeyTy

using mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::KeyTy = std::tuple<std::optional<DataType>, std::optional<DataType>, StringAttr, BoolAttr, BoolAttr, std::optional<uint32_t>, std::optional<uint32_t>, BoolAttr, BoolAttr, std::optional<TensorMemoryLayout>, CoreRangeSetAttr, BoolAttr, std::optional<Layout>, BoolAttr, BoolAttr, BoolAttr, BoolAttr, BoolAttr, BoolAttr>

Constructor & Destructor Documentation

◆ Conv2dConfigAttrStorage()

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

std::optional<uint32_t> mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::act_block_h_override

◆ act_block_w_div

std::optional<uint32_t> mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::act_block_w_div

◆ activation

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

◆ always_preprocess_weights

BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::always_preprocess_weights

◆ core_grid

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

◆ deallocate_activation

BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::deallocate_activation

◆ dtype

std::optional<DataType> mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::dtype

◆ enable_act_double_buffer

BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::enable_act_double_buffer

◆ enable_split_reader

BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::enable_split_reader

◆ enable_subblock_padding

BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::enable_subblock_padding

◆ enable_weights_double_buffer

BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::enable_weights_double_buffer

◆ output_layout

std::optional<Layout> mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::output_layout

◆ override_sharding_config

BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::override_sharding_config

◆ preprocess_weights_on_device

BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::preprocess_weights_on_device

◆ reallocate_halo_output

BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::reallocate_halo_output

◆ reshard_if_not_optimal

BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::reshard_if_not_optimal

◆ shard_layout

std::optional<TensorMemoryLayout> mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::shard_layout

◆ transpose_shards

BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::transpose_shards

◆ weights_dtype

std::optional<DataType> mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::weights_dtype

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