TT-MLIR
Public Types | Public Member Functions | Public Attributes | List of all members
tt::target::ttnn::Conv2dConfigBuilder Struct Reference

#include <conv_generated.h>

Public Types

typedef Conv2dConfig Table
 

Public Member Functions

void add_dtype (tt::target::DataType dtype)
 
void add_weights_dtype (tt::target::DataType weights_dtype)
 
void add_activation (::flatbuffers::Offset<::flatbuffers::String > activation)
 
void add_input_channels_alignment (uint32_t input_channels_alignment)
 
void add_deallocate_activation (bool deallocate_activation)
 
void add_reallocate_halo_output (bool reallocate_halo_output)
 
void add_act_block_h_override (uint32_t act_block_h_override)
 
void add_act_block_w_div (uint32_t act_block_w_div)
 
void add_reshard_if_not_optimal (bool reshard_if_not_optimal)
 
void add_override_sharding_config (bool override_sharding_config)
 
void add_shard_layout (tt::target::ttnn::TensorMemoryLayout shard_layout)
 
void add_core_grid (::flatbuffers::Offset< tt::target::ttnn::CoreRangeSet > core_grid)
 
void add_transpose_shards (bool transpose_shards)
 
void add_output_layout (tt::target::TensorLayout output_layout)
 
void add_preprocess_weights_on_device (bool preprocess_weights_on_device)
 
void add_always_preprocess_weights (bool always_preprocess_weights)
 
void add_enable_act_double_buffer (bool enable_act_double_buffer)
 
void add_enable_weights_double_buffer (bool enable_weights_double_buffer)
 
void add_enable_split_reader (bool enable_split_reader)
 
void add_enable_subblock_padding (bool enable_subblock_padding)
 
 Conv2dConfigBuilder (::flatbuffers::FlatBufferBuilder &_fbb)
 
::flatbuffers::Offset< Conv2dConfig > Finish ()
 

Public Attributes

::flatbuffers::FlatBufferBuilder & fbb_
 
::flatbuffers::uoffset_t start_
 

Member Typedef Documentation

◆ Table

Constructor & Destructor Documentation

◆ Conv2dConfigBuilder()

tt::target::ttnn::Conv2dConfigBuilder::Conv2dConfigBuilder ( ::flatbuffers::FlatBufferBuilder &  _fbb)
inlineexplicit

Member Function Documentation

◆ add_act_block_h_override()

void tt::target::ttnn::Conv2dConfigBuilder::add_act_block_h_override ( uint32_t  act_block_h_override)
inline

◆ add_act_block_w_div()

void tt::target::ttnn::Conv2dConfigBuilder::add_act_block_w_div ( uint32_t  act_block_w_div)
inline

◆ add_activation()

void tt::target::ttnn::Conv2dConfigBuilder::add_activation ( ::flatbuffers::Offset<::flatbuffers::String >  activation)
inline

◆ add_always_preprocess_weights()

void tt::target::ttnn::Conv2dConfigBuilder::add_always_preprocess_weights ( bool  always_preprocess_weights)
inline

◆ add_core_grid()

void tt::target::ttnn::Conv2dConfigBuilder::add_core_grid ( ::flatbuffers::Offset< tt::target::ttnn::CoreRangeSet >  core_grid)
inline

◆ add_deallocate_activation()

void tt::target::ttnn::Conv2dConfigBuilder::add_deallocate_activation ( bool  deallocate_activation)
inline

◆ add_dtype()

void tt::target::ttnn::Conv2dConfigBuilder::add_dtype ( tt::target::DataType  dtype)
inline

◆ add_enable_act_double_buffer()

void tt::target::ttnn::Conv2dConfigBuilder::add_enable_act_double_buffer ( bool  enable_act_double_buffer)
inline

◆ add_enable_split_reader()

void tt::target::ttnn::Conv2dConfigBuilder::add_enable_split_reader ( bool  enable_split_reader)
inline

◆ add_enable_subblock_padding()

void tt::target::ttnn::Conv2dConfigBuilder::add_enable_subblock_padding ( bool  enable_subblock_padding)
inline

◆ add_enable_weights_double_buffer()

void tt::target::ttnn::Conv2dConfigBuilder::add_enable_weights_double_buffer ( bool  enable_weights_double_buffer)
inline

◆ add_input_channels_alignment()

void tt::target::ttnn::Conv2dConfigBuilder::add_input_channels_alignment ( uint32_t  input_channels_alignment)
inline

◆ add_output_layout()

void tt::target::ttnn::Conv2dConfigBuilder::add_output_layout ( tt::target::TensorLayout  output_layout)
inline

◆ add_override_sharding_config()

void tt::target::ttnn::Conv2dConfigBuilder::add_override_sharding_config ( bool  override_sharding_config)
inline

◆ add_preprocess_weights_on_device()

void tt::target::ttnn::Conv2dConfigBuilder::add_preprocess_weights_on_device ( bool  preprocess_weights_on_device)
inline

◆ add_reallocate_halo_output()

void tt::target::ttnn::Conv2dConfigBuilder::add_reallocate_halo_output ( bool  reallocate_halo_output)
inline

◆ add_reshard_if_not_optimal()

void tt::target::ttnn::Conv2dConfigBuilder::add_reshard_if_not_optimal ( bool  reshard_if_not_optimal)
inline

◆ add_shard_layout()

void tt::target::ttnn::Conv2dConfigBuilder::add_shard_layout ( tt::target::ttnn::TensorMemoryLayout  shard_layout)
inline

◆ add_transpose_shards()

void tt::target::ttnn::Conv2dConfigBuilder::add_transpose_shards ( bool  transpose_shards)
inline

◆ add_weights_dtype()

void tt::target::ttnn::Conv2dConfigBuilder::add_weights_dtype ( tt::target::DataType  weights_dtype)
inline

◆ Finish()

::flatbuffers::Offset<Conv2dConfig> tt::target::ttnn::Conv2dConfigBuilder::Finish ( )
inline

Member Data Documentation

◆ fbb_

::flatbuffers::FlatBufferBuilder& tt::target::ttnn::Conv2dConfigBuilder::fbb_

◆ start_

::flatbuffers::uoffset_t tt::target::ttnn::Conv2dConfigBuilder::start_

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