tt::target::ttnn::BatchNormOpBuilder Struct Reference

#include <normalization_generated.h>

Public Types
typedef BatchNormOp	Table

Public Member Functions
void	add_input (::flatbuffers::Offset< tt::target::ttnn::TensorRef > input)
void	add_running_mean (::flatbuffers::Offset< tt::target::ttnn::TensorRef > running_mean)
void	add_running_var (::flatbuffers::Offset< tt::target::ttnn::TensorRef > running_var)
void	add_training (bool training)
void	add_epsilon (float epsilon)
void	add_momentum (float momentum)
void	add_weight (::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight)
void	add_bias (::flatbuffers::Offset< tt::target::ttnn::TensorRef > bias)
void	add_memory_config (::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memory_config)
void	add_out (::flatbuffers::Offset< tt::target::ttnn::TensorRef > out)
	BatchNormOpBuilder (::flatbuffers::FlatBufferBuilder &_fbb)
::flatbuffers::Offset< BatchNormOp >	Finish ()

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

Member Typedef Documentation

Table

typedef BatchNormOp tt::target::ttnn::BatchNormOpBuilder::Table

Constructor & Destructor Documentation

BatchNormOpBuilder()

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

inlineexplicit

Member Function Documentation

add_bias()

void tt::target::ttnn::BatchNormOpBuilder::add_bias ( ::flatbuffers::Offset< tt::target::ttnn::TensorRef >  bias )

inline

add_epsilon()

void tt::target::ttnn::BatchNormOpBuilder::add_epsilon ( float  epsilon )

inline

add_input()

void tt::target::ttnn::BatchNormOpBuilder::add_input ( ::flatbuffers::Offset< tt::target::ttnn::TensorRef >  input )

inline

add_memory_config()

void tt::target::ttnn::BatchNormOpBuilder::add_memory_config ( ::flatbuffers::Offset< tt::target::ttnn::MemoryConfig >  memory_config )

inline

add_momentum()

void tt::target::ttnn::BatchNormOpBuilder::add_momentum ( float  momentum )

inline

add_out()

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

inline

add_running_mean()

void tt::target::ttnn::BatchNormOpBuilder::add_running_mean ( ::flatbuffers::Offset< tt::target::ttnn::TensorRef >  running_mean )

inline

add_running_var()

void tt::target::ttnn::BatchNormOpBuilder::add_running_var ( ::flatbuffers::Offset< tt::target::ttnn::TensorRef >  running_var )

inline

add_training()

void tt::target::ttnn::BatchNormOpBuilder::add_training ( bool  training )

inline

add_weight()

void tt::target::ttnn::BatchNormOpBuilder::add_weight ( ::flatbuffers::Offset< tt::target::ttnn::TensorRef >  weight )

inline

Finish()

::flatbuffers::Offset<BatchNormOp> tt::target::ttnn::BatchNormOpBuilder::Finish ( )

inline

Member Data Documentation

fbb_

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

start_

::flatbuffers::uoffset_t tt::target::ttnn::BatchNormOpBuilder::start_

---

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

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