TT-MLIR
Classes | Enumerations | Functions
tt::target::ttnn Namespace Reference

Classes

struct  TTNNBinaryBinarySchema
 
struct  FLATBUFFERS_FINAL_CLASS
 
struct  TTNNBinaryBuilder
 
struct  OpTypeTraits
 
struct  OpTypeTraits< tt::target::ttnn::GetDeviceOp >
 
struct  OpTypeTraits< tt::target::ttnn::ToMemoryConfigOp >
 
struct  OpTypeTraits< tt::target::ttnn::ToLayoutOp >
 
struct  OpTypeTraits< tt::target::ttnn::ToDeviceOp >
 
struct  OpTypeTraits< tt::target::ttnn::FromDeviceOp >
 
struct  OpTypeTraits< tt::target::ttnn::EmptyOp >
 
struct  OpTypeTraits< tt::target::ttnn::FullOp >
 
struct  OpTypeTraits< tt::target::ttnn::EltwiseOp >
 
struct  OpTypeTraits< tt::target::ttnn::MatmulOp >
 
struct  OpTypeTraits< tt::target::ttnn::ReductionOp >
 
struct  OpTypeTraits< tt::target::ttnn::EmbeddingOp >
 
struct  OpTypeTraits< tt::target::ttnn::SoftmaxOp >
 
struct  OpTypeTraits< tt::target::ttnn::TransposeOp >
 
struct  OpTypeTraits< tt::target::ttnn::Conv2dOp >
 
struct  OpTypeTraits< tt::target::ttnn::ConcatOp >
 
struct  OpTypeTraits< tt::target::ttnn::ReshapeOp >
 
struct  OpTypeTraits< tt::target::ttnn::MaxPool2dOp >
 
struct  OpTypeTraits< tt::target::ttnn::DeallocOp >
 
struct  GetDeviceOpBuilder
 
struct  ToMemoryConfigOpBuilder
 
struct  ToLayoutOpBuilder
 
struct  ToDeviceOpBuilder
 
struct  FromDeviceOpBuilder
 
struct  EmptyOpBuilder
 
struct  FullOpBuilder
 
struct  EltwiseOpBuilder
 
struct  ReductionOpBuilder
 
struct  EmbeddingOpBuilder
 
struct  SoftmaxOpBuilder
 
struct  TransposeOpBuilder
 
struct  ConcatOpBuilder
 
struct  ReshapeOpBuilder
 
struct  MatmulOpBuilder
 
struct  Conv2dOpBuilder
 
struct  MaxPool2dOpBuilder
 
struct  DeallocOpBuilder
 
struct  OperationBuilder
 
struct  ProgramBuilder
 

Enumerations

enum class  EltwiseOpType : uint32_t {
  Add = 0 , Multiply = 1 , Subtract = 2 , Relu = 3 ,
  GreaterEqual = 4 , Sqrt = 5 , Div = 6 , Sigmoid = 7 ,
  Reciprocal = 8 , Exp = 9 , Maximum = 10 , Abs = 11 ,
  Neg = 12 , MIN = Add , MAX = Neg
}
 
enum class  ReductionOpType : uint32_t {
  Sum = 0 , Mean = 1 , Max = 2 , MIN = Sum ,
  MAX = Max
}
 
enum class  OpType : uint8_t {
  NONE = 0 , GetDeviceOp = 1 , ToMemoryConfigOp = 2 , ToLayoutOp = 3 ,
  ToDeviceOp = 4 , FromDeviceOp = 5 , EmptyOp = 6 , FullOp = 7 ,
  EltwiseOp = 8 , MatmulOp = 9 , ReductionOp = 10 , EmbeddingOp = 11 ,
  SoftmaxOp = 12 , TransposeOp = 13 , Conv2dOp = 14 , ConcatOp = 15 ,
  ReshapeOp = 16 , MaxPool2dOp = 17 , DeallocOp = 18 , MIN = NONE ,
  MAX = DeallocOp
}
 

Functions

inline ::flatbuffers::Offset< TTNNBinary > CreateTTNNBinary (::flatbuffers::FlatBufferBuilder &_fbb, const tt::target::Version *version=nullptr, ::flatbuffers::Offset<::flatbuffers::String > ttmlir_git_hash=0, ::flatbuffers::Offset< tt::target::SystemDesc > system_desc=0, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::Program >>> programs=0)
 
inline ::flatbuffers::Offset< TTNNBinary > CreateTTNNBinaryDirect (::flatbuffers::FlatBufferBuilder &_fbb, const tt::target::Version *version=nullptr, const char *ttmlir_git_hash=nullptr, ::flatbuffers::Offset< tt::target::SystemDesc > system_desc=0, const std::vector<::flatbuffers::Offset< tt::target::ttnn::Program >> *programs=nullptr)
 
const tt::target::ttnn::TTNNBinary * GetTTNNBinary (const void *buf)
 
const tt::target::ttnn::TTNNBinary * GetSizePrefixedTTNNBinary (const void *buf)
 
const char * TTNNBinaryIdentifier ()
 
bool TTNNBinaryBufferHasIdentifier (const void *buf)
 
bool SizePrefixedTTNNBinaryBufferHasIdentifier (const void *buf)
 
bool VerifyTTNNBinaryBuffer (::flatbuffers::Verifier &verifier)
 
bool VerifySizePrefixedTTNNBinaryBuffer (::flatbuffers::Verifier &verifier)
 
const char * TTNNBinaryExtension ()
 
void FinishTTNNBinaryBuffer (::flatbuffers::FlatBufferBuilder &fbb, ::flatbuffers::Offset< tt::target::ttnn::TTNNBinary > root)
 
void FinishSizePrefixedTTNNBinaryBuffer (::flatbuffers::FlatBufferBuilder &fbb, ::flatbuffers::Offset< tt::target::ttnn::TTNNBinary > root)
 
const EltwiseOpType(& EnumValuesEltwiseOpType ())[13]
 
const char *const * EnumNamesEltwiseOpType ()
 
const char * EnumNameEltwiseOpType (EltwiseOpType e)
 
const ReductionOpType(& EnumValuesReductionOpType ())[3]
 
const char *const * EnumNamesReductionOpType ()
 
const char * EnumNameReductionOpType (ReductionOpType e)
 
const OpType(& EnumValuesOpType ())[19]
 
const char *const * EnumNamesOpType ()
 
const char * EnumNameOpType (OpType e)
 
bool VerifyOpType (::flatbuffers::Verifier &verifier, const void *obj, OpType type)
 
bool VerifyOpTypeVector (::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset< void >> *values, const ::flatbuffers::Vector< OpType > *types)
 
inline ::flatbuffers::Offset< GetDeviceOp > CreateGetDeviceOp (::flatbuffers::FlatBufferBuilder &_fbb, const tt::target::Dim2d *mesh=nullptr, ::flatbuffers::Offset<::flatbuffers::Vector< uint32_t >> chip_ids=0, ::flatbuffers::Offset< tt::target::DeviceRef > out=0)
 
inline ::flatbuffers::Offset< GetDeviceOp > CreateGetDeviceOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, const tt::target::Dim2d *mesh=nullptr, const std::vector< uint32_t > *chip_ids=nullptr, ::flatbuffers::Offset< tt::target::DeviceRef > out=0)
 
inline ::flatbuffers::Offset< ToMemoryConfigOp > CreateToMemoryConfigOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in0=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
 
inline ::flatbuffers::Offset< ToLayoutOp > CreateToLayoutOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, tt::target::TensorLayout layout=tt::target::TensorLayout::RowMajor, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
 
inline ::flatbuffers::Offset< ToDeviceOp > CreateToDeviceOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
 
inline ::flatbuffers::Offset< FromDeviceOp > CreateFromDeviceOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
 
inline ::flatbuffers::Offset< EmptyOp > CreateEmptyOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> shape=0, tt::target::DataType dtype=tt::target::DataType::Float32, tt::target::TensorLayout layout=tt::target::TensorLayout::RowMajor, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
 
inline ::flatbuffers::Offset< EmptyOp > CreateEmptyOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, const std::vector< int64_t > *shape=nullptr, tt::target::DataType dtype=tt::target::DataType::Float32, tt::target::TensorLayout layout=tt::target::TensorLayout::RowMajor, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
 
inline ::flatbuffers::Offset< FullOp > CreateFullOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, float fill_value=0.0f, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
 
inline ::flatbuffers::Offset< EltwiseOp > CreateEltwiseOp (::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::EltwiseOpType type=tt::target::ttnn::EltwiseOpType::Add, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> ins=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
 
inline ::flatbuffers::Offset< EltwiseOp > CreateEltwiseOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::EltwiseOpType type=tt::target::ttnn::EltwiseOpType::Add, const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> *ins=nullptr, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
 
inline ::flatbuffers::Offset< ReductionOp > CreateReductionOp (::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::ReductionOpType type=tt::target::ttnn::ReductionOpType::Sum, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> dim_arg=0, bool keep_dim=false)
 
inline ::flatbuffers::Offset< ReductionOp > CreateReductionOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::ReductionOpType type=tt::target::ttnn::ReductionOpType::Sum, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, const std::vector< int32_t > *dim_arg=nullptr, bool keep_dim=false)
 
inline ::flatbuffers::Offset< EmbeddingOp > CreateEmbeddingOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > input=0, ::flatbuffers::Offset< tt::target::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::TensorRef > output=0)
 
inline ::flatbuffers::Offset< SoftmaxOp > CreateSoftmaxOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, int32_t dimension=0)
 
inline ::flatbuffers::Offset< TransposeOp > CreateTransposeOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, int32_t dim0=0, int32_t dim1=0)
 
inline ::flatbuffers::Offset< ConcatOp > CreateConcatOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> inputs=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, int32_t dim=0)
 
inline ::flatbuffers::Offset< ConcatOp > CreateConcatOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> *inputs=nullptr, ::flatbuffers::Offset< tt::target::TensorRef > out=0, int32_t dim=0)
 
inline ::flatbuffers::Offset< ReshapeOp > CreateReshapeOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> shape=0)
 
inline ::flatbuffers::Offset< ReshapeOp > CreateReshapeOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, const std::vector< int32_t > *shape=nullptr)
 
inline ::flatbuffers::Offset< MatmulOp > CreateMatmulOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in0=0, ::flatbuffers::Offset< tt::target::TensorRef > in1=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
 
inline ::flatbuffers::Offset< Conv2dOp > CreateConv2dOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > input=0, ::flatbuffers::Offset< tt::target::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::TensorRef > bias=0, ::flatbuffers::Offset< tt::target::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, uint32_t kernel_height=0, uint32_t kernel_width=0, uint32_t stride_height=0, uint32_t stride_width=0, uint32_t padding_height=0, uint32_t padding_width=0, uint32_t dilation_height=0, uint32_t dilation_width=0, uint32_t groups=0)
 
inline ::flatbuffers::Offset< MaxPool2dOp > CreateMaxPool2dOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, uint32_t batch_size=0, uint32_t input_height=0, uint32_t input_width=0, uint32_t channels=0, uint32_t kernel_height=0, uint32_t kernel_width=0, uint32_t stride_height=0, uint32_t stride_width=0, uint32_t dilation_height=0, uint32_t dilation_width=0, bool ceil_mode=false, uint32_t padding_height=0, uint32_t padding_width=0)
 
inline ::flatbuffers::Offset< DeallocOp > CreateDeallocOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0)
 
template<>
const tt::target::ttnn::GetDeviceOp * Operation::type_as< tt::target::ttnn::GetDeviceOp > () const
 
template<>
const tt::target::ttnn::ToMemoryConfigOp * Operation::type_as< tt::target::ttnn::ToMemoryConfigOp > () const
 
template<>
const tt::target::ttnn::ToLayoutOp * Operation::type_as< tt::target::ttnn::ToLayoutOp > () const
 
template<>
const tt::target::ttnn::ToDeviceOp * Operation::type_as< tt::target::ttnn::ToDeviceOp > () const
 
template<>
const tt::target::ttnn::FromDeviceOp * Operation::type_as< tt::target::ttnn::FromDeviceOp > () const
 
template<>
const tt::target::ttnn::EmptyOp * Operation::type_as< tt::target::ttnn::EmptyOp > () const
 
template<>
const tt::target::ttnn::FullOp * Operation::type_as< tt::target::ttnn::FullOp > () const
 
template<>
const tt::target::ttnn::EltwiseOp * Operation::type_as< tt::target::ttnn::EltwiseOp > () const
 
template<>
const tt::target::ttnn::MatmulOp * Operation::type_as< tt::target::ttnn::MatmulOp > () const
 
template<>
const tt::target::ttnn::ReductionOp * Operation::type_as< tt::target::ttnn::ReductionOp > () const
 
template<>
const tt::target::ttnn::EmbeddingOp * Operation::type_as< tt::target::ttnn::EmbeddingOp > () const
 
template<>
const tt::target::ttnn::SoftmaxOp * Operation::type_as< tt::target::ttnn::SoftmaxOp > () const
 
template<>
const tt::target::ttnn::TransposeOp * Operation::type_as< tt::target::ttnn::TransposeOp > () const
 
template<>
const tt::target::ttnn::Conv2dOp * Operation::type_as< tt::target::ttnn::Conv2dOp > () const
 
template<>
const tt::target::ttnn::ConcatOp * Operation::type_as< tt::target::ttnn::ConcatOp > () const
 
template<>
const tt::target::ttnn::ReshapeOp * Operation::type_as< tt::target::ttnn::ReshapeOp > () const
 
template<>
const tt::target::ttnn::MaxPool2dOp * Operation::type_as< tt::target::ttnn::MaxPool2dOp > () const
 
template<>
const tt::target::ttnn::DeallocOp * Operation::type_as< tt::target::ttnn::DeallocOp > () const
 
inline ::flatbuffers::Offset< Operation > CreateOperation (::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::OpType type_type=tt::target::ttnn::OpType::NONE, ::flatbuffers::Offset< void > type=0, ::flatbuffers::Offset<::flatbuffers::String > debug_info=0)
 
inline ::flatbuffers::Offset< Operation > CreateOperationDirect (::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::OpType type_type=tt::target::ttnn::OpType::NONE, ::flatbuffers::Offset< void > type=0, const char *debug_info=nullptr)
 
inline ::flatbuffers::Offset< Program > CreateProgram (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset<::flatbuffers::String > name=0, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> inputs=0, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> outputs=0, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::Operation >>> operations=0, ::flatbuffers::Offset< tt::target::DebugInfo > debug_info=0)
 
inline ::flatbuffers::Offset< Program > CreateProgramDirect (::flatbuffers::FlatBufferBuilder &_fbb, const char *name=nullptr, const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> *inputs=nullptr, const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> *outputs=nullptr, const std::vector<::flatbuffers::Offset< tt::target::ttnn::Operation >> *operations=nullptr, ::flatbuffers::Offset< tt::target::DebugInfo > debug_info=0)
 

Enumeration Type Documentation

◆ EltwiseOpType

enum tt::target::ttnn::EltwiseOpType : uint32_t
strong
Enumerator
Add 
Multiply 
Subtract 
Relu 
GreaterEqual 
Sqrt 
Div 
Sigmoid 
Reciprocal 
Exp 
Maximum 
Abs 
Neg 
MIN 
MAX 

◆ OpType

enum tt::target::ttnn::OpType : uint8_t
strong
Enumerator
NONE 
GetDeviceOp 
ToMemoryConfigOp 
ToLayoutOp 
ToDeviceOp 
FromDeviceOp 
EmptyOp 
FullOp 
EltwiseOp 
MatmulOp 
ReductionOp 
EmbeddingOp 
SoftmaxOp 
TransposeOp 
Conv2dOp 
ConcatOp 
ReshapeOp 
MaxPool2dOp 
DeallocOp 
MIN 
MAX 

◆ ReductionOpType

enum tt::target::ttnn::ReductionOpType : uint32_t
strong
Enumerator
Sum 
Mean 
Max 
MIN 
MAX 

Function Documentation

◆ CreateConcatOp()

inline ::flatbuffers::Offset<ConcatOp> tt::target::ttnn::CreateConcatOp ( ::flatbuffers::FlatBufferBuilder &  _fbb,
::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>>  inputs = 0,
::flatbuffers::Offset< tt::target::TensorRef >  out = 0,
int32_t  dim = 0 
)

◆ CreateConcatOpDirect()

inline ::flatbuffers::Offset<ConcatOp> tt::target::ttnn::CreateConcatOpDirect ( ::flatbuffers::FlatBufferBuilder &  _fbb,
const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> *  inputs = nullptr,
::flatbuffers::Offset< tt::target::TensorRef >  out = 0,
int32_t  dim = 0 
)

◆ CreateConv2dOp()

inline ::flatbuffers::Offset<Conv2dOp> tt::target::ttnn::CreateConv2dOp ( ::flatbuffers::FlatBufferBuilder &  _fbb,
::flatbuffers::Offset< tt::target::TensorRef >  input = 0,
::flatbuffers::Offset< tt::target::TensorRef >  weight = 0,
::flatbuffers::Offset< tt::target::TensorRef >  bias = 0,
::flatbuffers::Offset< tt::target::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,
uint32_t  kernel_height = 0,
uint32_t  kernel_width = 0,
uint32_t  stride_height = 0,
uint32_t  stride_width = 0,
uint32_t  padding_height = 0,
uint32_t  padding_width = 0,
uint32_t  dilation_height = 0,
uint32_t  dilation_width = 0,
uint32_t  groups = 0 
)

◆ CreateDeallocOp()

inline ::flatbuffers::Offset<DeallocOp> tt::target::ttnn::CreateDeallocOp ( ::flatbuffers::FlatBufferBuilder &  _fbb,
::flatbuffers::Offset< tt::target::TensorRef >  in = 0 
)

◆ CreateEltwiseOp()

inline ::flatbuffers::Offset<EltwiseOp> tt::target::ttnn::CreateEltwiseOp ( ::flatbuffers::FlatBufferBuilder &  _fbb,
tt::target::ttnn::EltwiseOpType  type = tt::target::ttnn::EltwiseOpType::Add,
::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>>  ins = 0,
::flatbuffers::Offset< tt::target::TensorRef >  out = 0 
)

◆ CreateEltwiseOpDirect()

inline ::flatbuffers::Offset<EltwiseOp> tt::target::ttnn::CreateEltwiseOpDirect ( ::flatbuffers::FlatBufferBuilder &  _fbb,
tt::target::ttnn::EltwiseOpType  type = tt::target::ttnn::EltwiseOpType::Add,
const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> *  ins = nullptr,
::flatbuffers::Offset< tt::target::TensorRef >  out = 0 
)

◆ CreateEmbeddingOp()

inline ::flatbuffers::Offset<EmbeddingOp> tt::target::ttnn::CreateEmbeddingOp ( ::flatbuffers::FlatBufferBuilder &  _fbb,
::flatbuffers::Offset< tt::target::TensorRef >  input = 0,
::flatbuffers::Offset< tt::target::TensorRef >  weight = 0,
::flatbuffers::Offset< tt::target::TensorRef >  output = 0 
)

◆ CreateEmptyOp()

inline ::flatbuffers::Offset<EmptyOp> tt::target::ttnn::CreateEmptyOp ( ::flatbuffers::FlatBufferBuilder &  _fbb,
::flatbuffers::Offset<::flatbuffers::Vector< int64_t >>  shape = 0,
tt::target::DataType  dtype = tt::target::DataType::Float32,
tt::target::TensorLayout  layout = tt::target::TensorLayout::RowMajor,
::flatbuffers::Offset< tt::target::DeviceRef >  device = 0,
::flatbuffers::Offset< tt::target::MemoryConfigDesc >  memcfg = 0,
::flatbuffers::Offset< tt::target::TensorRef >  out = 0 
)

◆ CreateEmptyOpDirect()

inline ::flatbuffers::Offset<EmptyOp> tt::target::ttnn::CreateEmptyOpDirect ( ::flatbuffers::FlatBufferBuilder &  _fbb,
const std::vector< int64_t > *  shape = nullptr,
tt::target::DataType  dtype = tt::target::DataType::Float32,
tt::target::TensorLayout  layout = tt::target::TensorLayout::RowMajor,
::flatbuffers::Offset< tt::target::DeviceRef >  device = 0,
::flatbuffers::Offset< tt::target::MemoryConfigDesc >  memcfg = 0,
::flatbuffers::Offset< tt::target::TensorRef >  out = 0 
)

◆ CreateFromDeviceOp()

inline ::flatbuffers::Offset<FromDeviceOp> tt::target::ttnn::CreateFromDeviceOp ( ::flatbuffers::FlatBufferBuilder &  _fbb,
::flatbuffers::Offset< tt::target::TensorRef >  in = 0,
::flatbuffers::Offset< tt::target::TensorRef >  out = 0 
)

◆ CreateFullOp()

inline ::flatbuffers::Offset<FullOp> tt::target::ttnn::CreateFullOp ( ::flatbuffers::FlatBufferBuilder &  _fbb,
::flatbuffers::Offset< tt::target::DeviceRef >  device = 0,
float  fill_value = 0.0f,
::flatbuffers::Offset< tt::target::TensorRef >  out = 0 
)

◆ CreateGetDeviceOp()

inline ::flatbuffers::Offset<GetDeviceOp> tt::target::ttnn::CreateGetDeviceOp ( ::flatbuffers::FlatBufferBuilder &  _fbb,
const tt::target::Dim2d *  mesh = nullptr,
::flatbuffers::Offset<::flatbuffers::Vector< uint32_t >>  chip_ids = 0,
::flatbuffers::Offset< tt::target::DeviceRef >  out = 0 
)

◆ CreateGetDeviceOpDirect()

inline ::flatbuffers::Offset<GetDeviceOp> tt::target::ttnn::CreateGetDeviceOpDirect ( ::flatbuffers::FlatBufferBuilder &  _fbb,
const tt::target::Dim2d *  mesh = nullptr,
const std::vector< uint32_t > *  chip_ids = nullptr,
::flatbuffers::Offset< tt::target::DeviceRef >  out = 0 
)

◆ CreateMatmulOp()

inline ::flatbuffers::Offset<MatmulOp> tt::target::ttnn::CreateMatmulOp ( ::flatbuffers::FlatBufferBuilder &  _fbb,
::flatbuffers::Offset< tt::target::TensorRef >  in0 = 0,
::flatbuffers::Offset< tt::target::TensorRef >  in1 = 0,
::flatbuffers::Offset< tt::target::TensorRef >  out = 0 
)

◆ CreateMaxPool2dOp()

inline ::flatbuffers::Offset<MaxPool2dOp> tt::target::ttnn::CreateMaxPool2dOp ( ::flatbuffers::FlatBufferBuilder &  _fbb,
::flatbuffers::Offset< tt::target::TensorRef >  in = 0,
::flatbuffers::Offset< tt::target::TensorRef >  out = 0,
::flatbuffers::Offset< tt::target::DeviceRef >  device = 0,
uint32_t  batch_size = 0,
uint32_t  input_height = 0,
uint32_t  input_width = 0,
uint32_t  channels = 0,
uint32_t  kernel_height = 0,
uint32_t  kernel_width = 0,
uint32_t  stride_height = 0,
uint32_t  stride_width = 0,
uint32_t  dilation_height = 0,
uint32_t  dilation_width = 0,
bool  ceil_mode = false,
uint32_t  padding_height = 0,
uint32_t  padding_width = 0 
)

◆ CreateOperation()

inline ::flatbuffers::Offset<Operation> tt::target::ttnn::CreateOperation ( ::flatbuffers::FlatBufferBuilder &  _fbb,
tt::target::ttnn::OpType  type_type = tt::target::ttnn::OpType::NONE,
::flatbuffers::Offset< void >  type = 0,
::flatbuffers::Offset<::flatbuffers::String >  debug_info = 0 
)

◆ CreateOperationDirect()

inline ::flatbuffers::Offset<Operation> tt::target::ttnn::CreateOperationDirect ( ::flatbuffers::FlatBufferBuilder &  _fbb,
tt::target::ttnn::OpType  type_type = tt::target::ttnn::OpType::NONE,
::flatbuffers::Offset< void >  type = 0,
const char *  debug_info = nullptr 
)

◆ CreateProgram()

inline ::flatbuffers::Offset<Program> tt::target::ttnn::CreateProgram ( ::flatbuffers::FlatBufferBuilder &  _fbb,
::flatbuffers::Offset<::flatbuffers::String >  name = 0,
::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>>  inputs = 0,
::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>>  outputs = 0,
::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::Operation >>>  operations = 0,
::flatbuffers::Offset< tt::target::DebugInfo >  debug_info = 0 
)

◆ CreateProgramDirect()

inline ::flatbuffers::Offset<Program> tt::target::ttnn::CreateProgramDirect ( ::flatbuffers::FlatBufferBuilder &  _fbb,
const char *  name = nullptr,
const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> *  inputs = nullptr,
const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> *  outputs = nullptr,
const std::vector<::flatbuffers::Offset< tt::target::ttnn::Operation >> *  operations = nullptr,
::flatbuffers::Offset< tt::target::DebugInfo >  debug_info = 0 
)

◆ CreateReductionOp()

inline ::flatbuffers::Offset<ReductionOp> tt::target::ttnn::CreateReductionOp ( ::flatbuffers::FlatBufferBuilder &  _fbb,
tt::target::ttnn::ReductionOpType  type = tt::target::ttnn::ReductionOpType::Sum,
::flatbuffers::Offset< tt::target::TensorRef >  in = 0,
::flatbuffers::Offset< tt::target::TensorRef >  out = 0,
::flatbuffers::Offset<::flatbuffers::Vector< int32_t >>  dim_arg = 0,
bool  keep_dim = false 
)

◆ CreateReductionOpDirect()

inline ::flatbuffers::Offset<ReductionOp> tt::target::ttnn::CreateReductionOpDirect ( ::flatbuffers::FlatBufferBuilder &  _fbb,
tt::target::ttnn::ReductionOpType  type = tt::target::ttnn::ReductionOpType::Sum,
::flatbuffers::Offset< tt::target::TensorRef >  in = 0,
::flatbuffers::Offset< tt::target::TensorRef >  out = 0,
const std::vector< int32_t > *  dim_arg = nullptr,
bool  keep_dim = false 
)

◆ CreateReshapeOp()

inline ::flatbuffers::Offset<ReshapeOp> tt::target::ttnn::CreateReshapeOp ( ::flatbuffers::FlatBufferBuilder &  _fbb,
::flatbuffers::Offset< tt::target::TensorRef >  in = 0,
::flatbuffers::Offset< tt::target::TensorRef >  out = 0,
::flatbuffers::Offset<::flatbuffers::Vector< int32_t >>  shape = 0 
)

◆ CreateReshapeOpDirect()

inline ::flatbuffers::Offset<ReshapeOp> tt::target::ttnn::CreateReshapeOpDirect ( ::flatbuffers::FlatBufferBuilder &  _fbb,
::flatbuffers::Offset< tt::target::TensorRef >  in = 0,
::flatbuffers::Offset< tt::target::TensorRef >  out = 0,
const std::vector< int32_t > *  shape = nullptr 
)

◆ CreateSoftmaxOp()

inline ::flatbuffers::Offset<SoftmaxOp> tt::target::ttnn::CreateSoftmaxOp ( ::flatbuffers::FlatBufferBuilder &  _fbb,
::flatbuffers::Offset< tt::target::TensorRef >  in = 0,
::flatbuffers::Offset< tt::target::TensorRef >  out = 0,
int32_t  dimension = 0 
)

◆ CreateToDeviceOp()

inline ::flatbuffers::Offset<ToDeviceOp> tt::target::ttnn::CreateToDeviceOp ( ::flatbuffers::FlatBufferBuilder &  _fbb,
::flatbuffers::Offset< tt::target::TensorRef >  in = 0,
::flatbuffers::Offset< tt::target::DeviceRef >  device = 0,
::flatbuffers::Offset< tt::target::MemoryConfigDesc >  memcfg = 0,
::flatbuffers::Offset< tt::target::TensorRef >  out = 0 
)

◆ CreateToLayoutOp()

inline ::flatbuffers::Offset<ToLayoutOp> tt::target::ttnn::CreateToLayoutOp ( ::flatbuffers::FlatBufferBuilder &  _fbb,
::flatbuffers::Offset< tt::target::TensorRef >  in = 0,
tt::target::TensorLayout  layout = tt::target::TensorLayout::RowMajor,
::flatbuffers::Offset< tt::target::DeviceRef >  device = 0,
::flatbuffers::Offset< tt::target::TensorRef >  out = 0 
)

◆ CreateToMemoryConfigOp()

inline ::flatbuffers::Offset<ToMemoryConfigOp> tt::target::ttnn::CreateToMemoryConfigOp ( ::flatbuffers::FlatBufferBuilder &  _fbb,
::flatbuffers::Offset< tt::target::TensorRef >  in0 = 0,
::flatbuffers::Offset< tt::target::DeviceRef >  device = 0,
::flatbuffers::Offset< tt::target::TensorRef >  out = 0 
)

◆ CreateTransposeOp()

inline ::flatbuffers::Offset<TransposeOp> tt::target::ttnn::CreateTransposeOp ( ::flatbuffers::FlatBufferBuilder &  _fbb,
::flatbuffers::Offset< tt::target::TensorRef >  in = 0,
::flatbuffers::Offset< tt::target::TensorRef >  out = 0,
int32_t  dim0 = 0,
int32_t  dim1 = 0 
)

◆ CreateTTNNBinary()

inline ::flatbuffers::Offset<TTNNBinary> tt::target::ttnn::CreateTTNNBinary ( ::flatbuffers::FlatBufferBuilder &  _fbb,
const tt::target::Version *  version = nullptr,
::flatbuffers::Offset<::flatbuffers::String >  ttmlir_git_hash = 0,
::flatbuffers::Offset< tt::target::SystemDesc >  system_desc = 0,
::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::Program >>>  programs = 0 
)

◆ CreateTTNNBinaryDirect()

inline ::flatbuffers::Offset<TTNNBinary> tt::target::ttnn::CreateTTNNBinaryDirect ( ::flatbuffers::FlatBufferBuilder &  _fbb,
const tt::target::Version *  version = nullptr,
const char *  ttmlir_git_hash = nullptr,
::flatbuffers::Offset< tt::target::SystemDesc >  system_desc = 0,
const std::vector<::flatbuffers::Offset< tt::target::ttnn::Program >> *  programs = nullptr 
)

◆ EnumNameEltwiseOpType()

const char* tt::target::ttnn::EnumNameEltwiseOpType ( EltwiseOpType  e)
inline

◆ EnumNameOpType()

const char* tt::target::ttnn::EnumNameOpType ( OpType  e)
inline

◆ EnumNameReductionOpType()

const char* tt::target::ttnn::EnumNameReductionOpType ( ReductionOpType  e)
inline

◆ EnumNamesEltwiseOpType()

const char* const* tt::target::ttnn::EnumNamesEltwiseOpType ( )
inline

◆ EnumNamesOpType()

const char* const* tt::target::ttnn::EnumNamesOpType ( )
inline

◆ EnumNamesReductionOpType()

const char* const* tt::target::ttnn::EnumNamesReductionOpType ( )
inline

◆ EnumValuesEltwiseOpType()

const EltwiseOpType(& tt::target::ttnn::EnumValuesEltwiseOpType ( ) )[13]
inline

◆ EnumValuesOpType()

const OpType(& tt::target::ttnn::EnumValuesOpType ( ) )[19]
inline

◆ EnumValuesReductionOpType()

const ReductionOpType(& tt::target::ttnn::EnumValuesReductionOpType ( ) )[3]
inline

◆ FinishSizePrefixedTTNNBinaryBuffer()

void tt::target::ttnn::FinishSizePrefixedTTNNBinaryBuffer ( ::flatbuffers::FlatBufferBuilder &  fbb,
::flatbuffers::Offset< tt::target::ttnn::TTNNBinary >  root 
)
inline

◆ FinishTTNNBinaryBuffer()

void tt::target::ttnn::FinishTTNNBinaryBuffer ( ::flatbuffers::FlatBufferBuilder &  fbb,
::flatbuffers::Offset< tt::target::ttnn::TTNNBinary >  root 
)
inline

◆ GetSizePrefixedTTNNBinary()

const tt::target::ttnn::TTNNBinary* tt::target::ttnn::GetSizePrefixedTTNNBinary ( const void *  buf)
inline

◆ GetTTNNBinary()

const tt::target::ttnn::TTNNBinary* tt::target::ttnn::GetTTNNBinary ( const void *  buf)
inline

◆ Operation::type_as< tt::target::ttnn::ConcatOp >()

template<>
const tt::target::ttnn::ConcatOp* tt::target::ttnn::Operation::type_as< tt::target::ttnn::ConcatOp > ( ) const
inline

◆ Operation::type_as< tt::target::ttnn::Conv2dOp >()

template<>
const tt::target::ttnn::Conv2dOp* tt::target::ttnn::Operation::type_as< tt::target::ttnn::Conv2dOp > ( ) const
inline

◆ Operation::type_as< tt::target::ttnn::DeallocOp >()

template<>
const tt::target::ttnn::DeallocOp* tt::target::ttnn::Operation::type_as< tt::target::ttnn::DeallocOp > ( ) const
inline

◆ Operation::type_as< tt::target::ttnn::EltwiseOp >()

template<>
const tt::target::ttnn::EltwiseOp* tt::target::ttnn::Operation::type_as< tt::target::ttnn::EltwiseOp > ( ) const
inline

◆ Operation::type_as< tt::target::ttnn::EmbeddingOp >()

template<>
const tt::target::ttnn::EmbeddingOp* tt::target::ttnn::Operation::type_as< tt::target::ttnn::EmbeddingOp > ( ) const
inline

◆ Operation::type_as< tt::target::ttnn::EmptyOp >()

template<>
const tt::target::ttnn::EmptyOp* tt::target::ttnn::Operation::type_as< tt::target::ttnn::EmptyOp > ( ) const
inline

◆ Operation::type_as< tt::target::ttnn::FromDeviceOp >()

template<>
const tt::target::ttnn::FromDeviceOp* tt::target::ttnn::Operation::type_as< tt::target::ttnn::FromDeviceOp > ( ) const
inline

◆ Operation::type_as< tt::target::ttnn::FullOp >()

template<>
const tt::target::ttnn::FullOp* tt::target::ttnn::Operation::type_as< tt::target::ttnn::FullOp > ( ) const
inline

◆ Operation::type_as< tt::target::ttnn::GetDeviceOp >()

template<>
const tt::target::ttnn::GetDeviceOp* tt::target::ttnn::Operation::type_as< tt::target::ttnn::GetDeviceOp > ( ) const
inline

◆ Operation::type_as< tt::target::ttnn::MatmulOp >()

template<>
const tt::target::ttnn::MatmulOp* tt::target::ttnn::Operation::type_as< tt::target::ttnn::MatmulOp > ( ) const
inline

◆ Operation::type_as< tt::target::ttnn::MaxPool2dOp >()

template<>
const tt::target::ttnn::MaxPool2dOp* tt::target::ttnn::Operation::type_as< tt::target::ttnn::MaxPool2dOp > ( ) const
inline

◆ Operation::type_as< tt::target::ttnn::ReductionOp >()

template<>
const tt::target::ttnn::ReductionOp* tt::target::ttnn::Operation::type_as< tt::target::ttnn::ReductionOp > ( ) const
inline

◆ Operation::type_as< tt::target::ttnn::ReshapeOp >()

template<>
const tt::target::ttnn::ReshapeOp* tt::target::ttnn::Operation::type_as< tt::target::ttnn::ReshapeOp > ( ) const
inline

◆ Operation::type_as< tt::target::ttnn::SoftmaxOp >()

template<>
const tt::target::ttnn::SoftmaxOp* tt::target::ttnn::Operation::type_as< tt::target::ttnn::SoftmaxOp > ( ) const
inline

◆ Operation::type_as< tt::target::ttnn::ToDeviceOp >()

template<>
const tt::target::ttnn::ToDeviceOp* tt::target::ttnn::Operation::type_as< tt::target::ttnn::ToDeviceOp > ( ) const
inline

◆ Operation::type_as< tt::target::ttnn::ToLayoutOp >()

template<>
const tt::target::ttnn::ToLayoutOp* tt::target::ttnn::Operation::type_as< tt::target::ttnn::ToLayoutOp > ( ) const
inline

◆ Operation::type_as< tt::target::ttnn::ToMemoryConfigOp >()

template<>
const tt::target::ttnn::ToMemoryConfigOp* tt::target::ttnn::Operation::type_as< tt::target::ttnn::ToMemoryConfigOp > ( ) const
inline

◆ Operation::type_as< tt::target::ttnn::TransposeOp >()

template<>
const tt::target::ttnn::TransposeOp* tt::target::ttnn::Operation::type_as< tt::target::ttnn::TransposeOp > ( ) const
inline

◆ SizePrefixedTTNNBinaryBufferHasIdentifier()

bool tt::target::ttnn::SizePrefixedTTNNBinaryBufferHasIdentifier ( const void *  buf)
inline

◆ TTNNBinaryBufferHasIdentifier()

bool tt::target::ttnn::TTNNBinaryBufferHasIdentifier ( const void *  buf)
inline

◆ TTNNBinaryExtension()

const char* tt::target::ttnn::TTNNBinaryExtension ( )
inline

◆ TTNNBinaryIdentifier()

const char* tt::target::ttnn::TTNNBinaryIdentifier ( )
inline

◆ VerifyOpType()

bool tt::target::ttnn::VerifyOpType ( ::flatbuffers::Verifier &  verifier,
const void *  obj,
OpType  type 
)
inline

◆ VerifyOpTypeVector()

bool tt::target::ttnn::VerifyOpTypeVector ( ::flatbuffers::Verifier &  verifier,
const ::flatbuffers::Vector<::flatbuffers::Offset< void >> *  values,
const ::flatbuffers::Vector< OpType > *  types 
)
inline

◆ VerifySizePrefixedTTNNBinaryBuffer()

bool tt::target::ttnn::VerifySizePrefixedTTNNBinaryBuffer ( ::flatbuffers::Verifier &  verifier)
inline

◆ VerifyTTNNBinaryBuffer()

bool tt::target::ttnn::VerifyTTNNBinaryBuffer ( ::flatbuffers::Verifier &  verifier)
inline
Generated by doxygen 1.9.1