tt::target::ttnn::PrepareConv2dWeightsOpBuilder Struct Reference

#include <conv_generated.h>

Public Types
typedef PrepareConv2dWeightsOp	Table

Public Member Functions
void	add_weight_tensor (::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight_tensor)
void	add_out (::flatbuffers::Offset< tt::target::ttnn::TensorRef > out)
void	add_input_memory_config (::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > input_memory_config)
void	add_input_tensor_layout (tt::target::TensorLayout input_tensor_layout)
void	add_weights_format (::flatbuffers::Offset<::flatbuffers::String > weights_format)
void	add_in_channels (uint32_t in_channels)
void	add_out_channels (uint32_t out_channels)
void	add_batch_size (uint32_t batch_size)
void	add_input_height (uint32_t input_height)
void	add_input_width (uint32_t input_width)
void	add_kernel_size (::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> kernel_size)
void	add_stride (::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> stride)
void	add_padding (::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> padding)
void	add_dilation (::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> dilation)
void	add_has_bias (bool has_bias)
void	add_groups (uint32_t groups)
void	add_device (::flatbuffers::Offset< tt::target::DeviceRef > device)
void	add_conv2d_config (::flatbuffers::Offset< tt::target::ttnn::Conv2dConfig > conv2d_config)
	PrepareConv2dWeightsOpBuilder (::flatbuffers::FlatBufferBuilder &_fbb)
::flatbuffers::Offset< PrepareConv2dWeightsOp >	Finish ()

Public Attributes
::flatbuffers::FlatBufferBuilder &	fbb_
::flatbuffers::uoffset_t	start_

Member Typedef Documentation

Table

typedef PrepareConv2dWeightsOp tt::target::ttnn::PrepareConv2dWeightsOpBuilder::Table

Constructor & Destructor Documentation

PrepareConv2dWeightsOpBuilder()

tt::target::ttnn::PrepareConv2dWeightsOpBuilder::PrepareConv2dWeightsOpBuilder ( ::flatbuffers::FlatBufferBuilder &  _fbb )

inlineexplicit

Member Function Documentation

add_batch_size()

void tt::target::ttnn::PrepareConv2dWeightsOpBuilder::add_batch_size ( uint32_t  batch_size )

inline

add_conv2d_config()

void tt::target::ttnn::PrepareConv2dWeightsOpBuilder::add_conv2d_config ( ::flatbuffers::Offset< tt::target::ttnn::Conv2dConfig >  conv2d_config )

inline

add_device()

void tt::target::ttnn::PrepareConv2dWeightsOpBuilder::add_device ( ::flatbuffers::Offset< tt::target::DeviceRef >  device )

inline

add_dilation()

void tt::target::ttnn::PrepareConv2dWeightsOpBuilder::add_dilation ( ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >>  dilation )

inline

add_groups()

void tt::target::ttnn::PrepareConv2dWeightsOpBuilder::add_groups ( uint32_t  groups )

inline

add_has_bias()

void tt::target::ttnn::PrepareConv2dWeightsOpBuilder::add_has_bias ( bool  has_bias )

inline

add_in_channels()

void tt::target::ttnn::PrepareConv2dWeightsOpBuilder::add_in_channels ( uint32_t  in_channels )

inline

add_input_height()

void tt::target::ttnn::PrepareConv2dWeightsOpBuilder::add_input_height ( uint32_t  input_height )

inline

add_input_memory_config()

void tt::target::ttnn::PrepareConv2dWeightsOpBuilder::add_input_memory_config ( ::flatbuffers::Offset< tt::target::ttnn::MemoryConfig >  input_memory_config )

inline

add_input_tensor_layout()

void tt::target::ttnn::PrepareConv2dWeightsOpBuilder::add_input_tensor_layout ( tt::target::TensorLayout  input_tensor_layout )

inline

add_input_width()

void tt::target::ttnn::PrepareConv2dWeightsOpBuilder::add_input_width ( uint32_t  input_width )

inline

add_kernel_size()

void tt::target::ttnn::PrepareConv2dWeightsOpBuilder::add_kernel_size ( ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >>  kernel_size )

inline

add_out()

void tt::target::ttnn::PrepareConv2dWeightsOpBuilder::add_out ( ::flatbuffers::Offset< tt::target::ttnn::TensorRef >  out )

inline

add_out_channels()

void tt::target::ttnn::PrepareConv2dWeightsOpBuilder::add_out_channels ( uint32_t  out_channels )

inline

add_padding()

void tt::target::ttnn::PrepareConv2dWeightsOpBuilder::add_padding ( ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >>  padding )

inline

add_stride()

void tt::target::ttnn::PrepareConv2dWeightsOpBuilder::add_stride ( ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >>  stride )

inline

add_weight_tensor()

void tt::target::ttnn::PrepareConv2dWeightsOpBuilder::add_weight_tensor ( ::flatbuffers::Offset< tt::target::ttnn::TensorRef >  weight_tensor )

inline

add_weights_format()

void tt::target::ttnn::PrepareConv2dWeightsOpBuilder::add_weights_format ( ::flatbuffers::Offset<::flatbuffers::String >  weights_format )

inline

Finish()

::flatbuffers::Offset<PrepareConv2dWeightsOp> tt::target::ttnn::PrepareConv2dWeightsOpBuilder::Finish ( )

inline

Member Data Documentation

fbb_

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

start_

::flatbuffers::uoffset_t tt::target::ttnn::PrepareConv2dWeightsOpBuilder::start_

---

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

• /__w/tt-mlir/tt-mlir/build/include/ttmlir/Target/TTNN/operations/conv_generated.h