TT-MLIR
Classes | Namespaces | Functions | Variables
conv_generated.h File Reference
#include "flatbuffers/flatbuffers.h"
#include "ttmlir/Target/Common/types_generated.h"
#include "ttmlir/Target/TTNN/types_generated.h"

Go to the source code of this file.

Classes

struct  tt::target::ttnn::FLATBUFFERS_FINAL_CLASS
 
struct  tt::target::ttnn::Conv2dConfigBuilder
 
struct  tt::target::ttnn::Conv2dConfig::Traits
 
struct  tt::target::ttnn::PrepareConv2dWeightsOpBuilder
 
struct  tt::target::ttnn::PrepareConv2dWeightsOp::Traits
 
struct  tt::target::ttnn::Conv2dOpBuilder
 
struct  tt::target::ttnn::Conv2dOp::Traits
 
struct  tt::target::ttnn::ConvTranspose2dOpBuilder
 
struct  tt::target::ttnn::ConvTranspose2dOp::Traits
 

Namespaces

 tt
 
 tt::target
 
 tt::target::ttnn
 

Functions

inline ::flatbuffers::Offset< Conv2dConfig > tt::target::ttnn::CreateConv2dConfig (::flatbuffers::FlatBufferBuilder &_fbb, tt::target::DataType dtype=tt::target::DataType::Float32, tt::target::DataType weights_dtype=tt::target::DataType::Float32, ::flatbuffers::Offset<::flatbuffers::String > activation=0, uint32_t input_channels_alignment=0, bool deallocate_activation=false, bool reallocate_halo_output=false, uint32_t act_block_h_override=0, uint32_t act_block_w_div=0, bool reshard_if_not_optimal=false, bool override_sharding_config=false, ::flatbuffers::Optional< tt::target::ttnn::TensorMemoryLayout > shard_layout=::flatbuffers::nullopt, ::flatbuffers::Offset< tt::target::ttnn::CoreRangeSet > core_grid=0, bool transpose_shards=false, tt::target::TensorLayout output_layout=tt::target::TensorLayout::RowMajor, bool preprocess_weights_on_device=false, bool always_preprocess_weights=false, bool enable_act_double_buffer=false, bool enable_weights_double_buffer=false, bool enable_split_reader=false, bool enable_subblock_padding=false)
 
inline ::flatbuffers::Offset< Conv2dConfig > tt::target::ttnn::CreateConv2dConfigDirect (::flatbuffers::FlatBufferBuilder &_fbb, tt::target::DataType dtype=tt::target::DataType::Float32, tt::target::DataType weights_dtype=tt::target::DataType::Float32, const char *activation=nullptr, uint32_t input_channels_alignment=0, bool deallocate_activation=false, bool reallocate_halo_output=false, uint32_t act_block_h_override=0, uint32_t act_block_w_div=0, bool reshard_if_not_optimal=false, bool override_sharding_config=false, ::flatbuffers::Optional< tt::target::ttnn::TensorMemoryLayout > shard_layout=::flatbuffers::nullopt, ::flatbuffers::Offset< tt::target::ttnn::CoreRangeSet > core_grid=0, bool transpose_shards=false, tt::target::TensorLayout output_layout=tt::target::TensorLayout::RowMajor, bool preprocess_weights_on_device=false, bool always_preprocess_weights=false, bool enable_act_double_buffer=false, bool enable_weights_double_buffer=false, bool enable_split_reader=false, bool enable_subblock_padding=false)
 
inline ::flatbuffers::Offset< PrepareConv2dWeightsOp > tt::target::ttnn::CreatePrepareConv2dWeightsOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight_tensor=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0, ::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > input_memory_config=0, tt::target::TensorLayout input_tensor_layout=tt::target::TensorLayout::RowMajor, ::flatbuffers::Offset<::flatbuffers::String > weights_format=0, uint32_t in_channels=0, uint32_t out_channels=0, uint32_t batch_size=0, uint32_t input_height=0, uint32_t input_width=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> kernel_size=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> stride=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> padding=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> dilation=0, bool has_bias=false, uint32_t groups=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::ttnn::Conv2dConfig > conv2d_config=0)
 
inline ::flatbuffers::Offset< PrepareConv2dWeightsOp > tt::target::ttnn::CreatePrepareConv2dWeightsOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight_tensor=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0, ::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > input_memory_config=0, tt::target::TensorLayout input_tensor_layout=tt::target::TensorLayout::RowMajor, const char *weights_format=nullptr, uint32_t in_channels=0, uint32_t out_channels=0, uint32_t batch_size=0, uint32_t input_height=0, uint32_t input_width=0, const std::vector< int32_t > *kernel_size=nullptr, const std::vector< int32_t > *stride=nullptr, const std::vector< int32_t > *padding=nullptr, const std::vector< int32_t > *dilation=nullptr, bool has_bias=false, uint32_t groups=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::ttnn::Conv2dConfig > conv2d_config=0)
 
inline ::flatbuffers::Offset< Conv2dOp > tt::target::ttnn::CreateConv2dOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > input=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > bias=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, uint32_t in_channels=0, uint32_t out_channels=0, uint32_t batch_size=0, uint32_t input_height=0, uint32_t input_width=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> kernel_size=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> stride=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> padding=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> dilation=0, uint32_t groups=0, ::flatbuffers::Offset< tt::target::ttnn::Conv2dConfig > conv2d_config=0)
 
inline ::flatbuffers::Offset< Conv2dOp > tt::target::ttnn::CreateConv2dOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > input=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > bias=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, uint32_t in_channels=0, uint32_t out_channels=0, uint32_t batch_size=0, uint32_t input_height=0, uint32_t input_width=0, const std::vector< int32_t > *kernel_size=nullptr, const std::vector< int32_t > *stride=nullptr, const std::vector< int32_t > *padding=nullptr, const std::vector< int32_t > *dilation=nullptr, uint32_t groups=0, ::flatbuffers::Offset< tt::target::ttnn::Conv2dConfig > conv2d_config=0)
 
inline ::flatbuffers::Offset< ConvTranspose2dOp > tt::target::ttnn::CreateConvTranspose2dOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > input=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > bias=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, uint32_t in_channels=0, uint32_t out_channels=0, uint32_t batch_size=0, uint32_t input_height=0, uint32_t input_width=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> kernel_size=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> stride=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> padding=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> output_padding=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> dilation=0, uint32_t groups=0)
 
inline ::flatbuffers::Offset< ConvTranspose2dOp > tt::target::ttnn::CreateConvTranspose2dOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > input=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > bias=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, uint32_t in_channels=0, uint32_t out_channels=0, uint32_t batch_size=0, uint32_t input_height=0, uint32_t input_width=0, const std::vector< int32_t > *kernel_size=nullptr, const std::vector< int32_t > *stride=nullptr, const std::vector< int32_t > *padding=nullptr, const std::vector< int32_t > *output_padding=nullptr, const std::vector< int32_t > *dilation=nullptr, uint32_t groups=0)
 

Variables

 VT_DTYPE = 4
 
 VT_WEIGHTS_DTYPE = 6
 
 VT_ACTIVATION = 8
 
 VT_INPUT_CHANNELS_ALIGNMENT = 10
 
 VT_DEALLOCATE_ACTIVATION = 12
 
 VT_REALLOCATE_HALO_OUTPUT = 14
 
 VT_ACT_BLOCK_H_OVERRIDE = 16
 
 VT_ACT_BLOCK_W_DIV = 18
 
 VT_RESHARD_IF_NOT_OPTIMAL = 20
 
 VT_OVERRIDE_SHARDING_CONFIG = 22
 
 VT_SHARD_LAYOUT = 24
 
 VT_CORE_GRID = 26
 
 VT_TRANSPOSE_SHARDS = 28
 
 VT_OUTPUT_LAYOUT = 30
 
 VT_PREPROCESS_WEIGHTS_ON_DEVICE = 32
 
 VT_ALWAYS_PREPROCESS_WEIGHTS = 34
 
 VT_ENABLE_ACT_DOUBLE_BUFFER = 36
 
 VT_ENABLE_WEIGHTS_DOUBLE_BUFFER = 38
 
 VT_ENABLE_SPLIT_READER = 40
 
 VT_WEIGHT_TENSOR = 4
 
 VT_OUT = 6
 
 VT_INPUT_MEMORY_CONFIG = 8
 
 VT_INPUT_TENSOR_LAYOUT = 10
 
 VT_WEIGHTS_FORMAT = 12
 
 VT_IN_CHANNELS = 14
 
 VT_OUT_CHANNELS = 16
 
 VT_BATCH_SIZE = 18
 
 VT_INPUT_HEIGHT = 20
 
 VT_INPUT_WIDTH = 22
 
 VT_KERNEL_SIZE = 24
 
 VT_STRIDE = 26
 
 VT_PADDING = 28
 
 VT_DILATION = 30
 
 VT_HAS_BIAS = 32
 
 VT_GROUPS = 34
 
 VT_DEVICE = 36
 
 VT_INPUT = 4
 
 VT_WEIGHT = 6
 
 VT_BIAS = 8
 
 VT_OUTPUT_PADDING = 30
 

Variable Documentation

◆ VT_ACT_BLOCK_H_OVERRIDE

VT_ACT_BLOCK_H_OVERRIDE = 16

◆ VT_ACT_BLOCK_W_DIV

VT_ACT_BLOCK_W_DIV = 18

◆ VT_ACTIVATION

VT_ACTIVATION = 8

◆ VT_ALWAYS_PREPROCESS_WEIGHTS

VT_ALWAYS_PREPROCESS_WEIGHTS = 34

◆ VT_BATCH_SIZE

VT_BATCH_SIZE = 18

◆ VT_BIAS

VT_BIAS = 8

◆ VT_CORE_GRID

VT_CORE_GRID = 26

◆ VT_DEALLOCATE_ACTIVATION

VT_DEALLOCATE_ACTIVATION = 12

◆ VT_DEVICE

VT_DEVICE = 36

◆ VT_DILATION

VT_DILATION = 30

◆ VT_DTYPE

VT_DTYPE = 4

◆ VT_ENABLE_ACT_DOUBLE_BUFFER

VT_ENABLE_ACT_DOUBLE_BUFFER = 36

◆ VT_ENABLE_SPLIT_READER

VT_ENABLE_SPLIT_READER = 40

◆ VT_ENABLE_WEIGHTS_DOUBLE_BUFFER

VT_ENABLE_WEIGHTS_DOUBLE_BUFFER = 38

◆ VT_GROUPS

VT_GROUPS = 34

◆ VT_HAS_BIAS

VT_HAS_BIAS = 32

◆ VT_IN_CHANNELS

VT_IN_CHANNELS = 14

◆ VT_INPUT

VT_INPUT = 4

◆ VT_INPUT_CHANNELS_ALIGNMENT

VT_INPUT_CHANNELS_ALIGNMENT = 10

◆ VT_INPUT_HEIGHT

VT_INPUT_HEIGHT = 20

◆ VT_INPUT_MEMORY_CONFIG

VT_INPUT_MEMORY_CONFIG = 8

◆ VT_INPUT_TENSOR_LAYOUT

VT_INPUT_TENSOR_LAYOUT = 10

◆ VT_INPUT_WIDTH

VT_INPUT_WIDTH = 22

◆ VT_KERNEL_SIZE

VT_KERNEL_SIZE = 24

◆ VT_OUT

VT_OUT = 6

◆ VT_OUT_CHANNELS

VT_OUT_CHANNELS = 16

◆ VT_OUTPUT_LAYOUT

VT_OUTPUT_LAYOUT = 30

◆ VT_OUTPUT_PADDING

VT_OUTPUT_PADDING = 30

◆ VT_OVERRIDE_SHARDING_CONFIG

VT_OVERRIDE_SHARDING_CONFIG = 22

◆ VT_PADDING

VT_PADDING = 28

◆ VT_PREPROCESS_WEIGHTS_ON_DEVICE

VT_PREPROCESS_WEIGHTS_ON_DEVICE = 32

◆ VT_REALLOCATE_HALO_OUTPUT

VT_REALLOCATE_HALO_OUTPUT = 14

◆ VT_RESHARD_IF_NOT_OPTIMAL

VT_RESHARD_IF_NOT_OPTIMAL = 20

◆ VT_SHARD_LAYOUT

VT_SHARD_LAYOUT = 24

◆ VT_STRIDE

VT_STRIDE = 26

◆ VT_TRANSPOSE_SHARDS

VT_TRANSPOSE_SHARDS = 28

◆ VT_WEIGHT

VT_WEIGHT = 6

◆ VT_WEIGHT_TENSOR

VT_WEIGHT_TENSOR = 4

◆ VT_WEIGHTS_DTYPE

VT_WEIGHTS_DTYPE = 6

◆ VT_WEIGHTS_FORMAT

VT_WEIGHTS_FORMAT = 12