TT-MLIR
Classes | Namespaces | Enumerations | Functions | Variables
program_generated.h File Reference
#include "flatbuffers/flatbuffers.h"
#include "ttmlir/Target/Common/debug_info_generated.h"
#include "ttmlir/Target/Common/types_generated.h"
#include "ttmlir/Target/TTNN/operations/ccl_generated.h"
#include "ttmlir/Target/TTNN/operations/conv_generated.h"
#include "ttmlir/Target/TTNN/operations/cpu_generated.h"
#include "ttmlir/Target/TTNN/operations/creation_generated.h"
#include "ttmlir/Target/TTNN/operations/data_movement_generated.h"
#include "ttmlir/Target/TTNN/operations/deallocate_generated.h"
#include "ttmlir/Target/TTNN/operations/eltwise_generated.h"
#include "ttmlir/Target/TTNN/operations/embedding_generated.h"
#include "ttmlir/Target/TTNN/operations/get_device_generated.h"
#include "ttmlir/Target/TTNN/operations/kv_cache_generated.h"
#include "ttmlir/Target/TTNN/operations/layout_generated.h"
#include "ttmlir/Target/TTNN/operations/matmul_generated.h"
#include "ttmlir/Target/TTNN/operations/moreh_cumsum_generated.h"
#include "ttmlir/Target/TTNN/operations/pool_generated.h"
#include "ttmlir/Target/TTNN/operations/reduction_generated.h"
#include "ttmlir/Target/TTNN/operations/softmax_generated.h"
#include "ttmlir/Target/TTNN/types_generated.h"

Go to the source code of this file.

Classes

struct  tt::target::ttnn::OpTypeTraits< T >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::AllGatherOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::CollectivePermuteOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::MeshShardOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::ReduceScatterOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::GetDeviceOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::PrepareConv2dWeightsOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::Conv2dOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::ConvTranspose2dOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::CpuOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::ArangeOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::ConstantOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::ConstructTensorOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::EmptyOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::FullOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::NamedFullOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::ConcatOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::PadOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::PermuteOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::RepeatInterleaveOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::RepeatOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::ReshapeOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::SliceOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::TransposeOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::DeallocateOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::EltwiseBinaryOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::EltwiseBinaryCompositeOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::EltwiseTernaryWhereOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::EltwiseQuantizationOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::EltwiseUnaryOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::EltwiseUnaryCompositeOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::EmbeddingBackwardOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::EmbeddingOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::FillCacheOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::UpdateCacheOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::FromDeviceOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::ToDeviceOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::ToDTypeOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::ToLayoutOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::ToMemoryConfigOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::TypecastOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::LinearOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::MatmulOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::MorehCumSumOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::SoftmaxOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::MaxPool2dOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::UpsampleOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::ReductionArgMaxOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::ReductionOp >
 
struct  tt::target::ttnn::OpTypeTraits< tt::target::ttnn::ReductionProdOp >
 
struct  tt::target::ttnn::FLATBUFFERS_FINAL_CLASS
 
struct  tt::target::ttnn::OperationBuilder
 
struct  tt::target::ttnn::Operation::Traits
 
struct  tt::target::ttnn::ProgramBuilder
 
struct  tt::target::ttnn::Program::Traits
 

Namespaces

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

Enumerations

enum class  tt::target::ttnn::OpType : uint8_t {
  tt::target::ttnn::NONE = 0 , tt::target::ttnn::AllGatherOp = 1 , tt::target::ttnn::CollectivePermuteOp = 2 , tt::target::ttnn::MeshShardOp = 3 ,
  tt::target::ttnn::ReduceScatterOp = 4 , tt::target::ttnn::GetDeviceOp = 5 , tt::target::ttnn::PrepareConv2dWeightsOp = 6 , tt::target::ttnn::Conv2dOp = 7 ,
  tt::target::ttnn::ConvTranspose2dOp = 8 , tt::target::ttnn::CpuOp = 9 , tt::target::ttnn::ArangeOp = 10 , tt::target::ttnn::ConstantOp = 11 ,
  tt::target::ttnn::ConstructTensorOp = 12 , tt::target::ttnn::EmptyOp = 13 , tt::target::ttnn::FullOp = 14 , tt::target::ttnn::NamedFullOp = 15 ,
  tt::target::ttnn::ConcatOp = 16 , tt::target::ttnn::PadOp = 17 , tt::target::ttnn::PermuteOp = 18 , tt::target::ttnn::RepeatInterleaveOp = 19 ,
  tt::target::ttnn::RepeatOp = 20 , tt::target::ttnn::ReshapeOp = 21 , tt::target::ttnn::SliceOp = 22 , tt::target::ttnn::TransposeOp = 23 ,
  tt::target::ttnn::DeallocateOp = 24 , tt::target::ttnn::EltwiseBinaryOp = 25 , tt::target::ttnn::EltwiseBinaryCompositeOp = 26 , tt::target::ttnn::EltwiseTernaryWhereOp = 27 ,
  tt::target::ttnn::EltwiseQuantizationOp = 28 , tt::target::ttnn::EltwiseUnaryOp = 29 , tt::target::ttnn::EltwiseUnaryCompositeOp = 30 , tt::target::ttnn::EmbeddingBackwardOp = 31 ,
  tt::target::ttnn::EmbeddingOp = 32 , tt::target::ttnn::FillCacheOp = 33 , tt::target::ttnn::UpdateCacheOp = 34 , tt::target::ttnn::FromDeviceOp = 35 ,
  tt::target::ttnn::ToDeviceOp = 36 , tt::target::ttnn::ToDTypeOp = 37 , tt::target::ttnn::ToLayoutOp = 38 , tt::target::ttnn::ToMemoryConfigOp = 39 ,
  tt::target::ttnn::TypecastOp = 40 , tt::target::ttnn::LinearOp = 41 , tt::target::ttnn::MatmulOp = 42 , tt::target::ttnn::MorehCumSumOp = 43 ,
  tt::target::ttnn::SoftmaxOp = 44 , tt::target::ttnn::MaxPool2dOp = 45 , tt::target::ttnn::UpsampleOp = 46 , tt::target::ttnn::ReductionArgMaxOp = 47 ,
  tt::target::ttnn::ReductionOp = 48 , tt::target::ttnn::ReductionProdOp = 49 , tt::target::ttnn::MIN = NONE , tt::target::ttnn::MAX = ReductionProdOp
}
 

Functions

const OpType(& tt::target::ttnn::EnumValuesOpType ())[50]
 
const char *const * tt::target::ttnn::EnumNamesOpType ()
 
const char * tt::target::ttnn::EnumNameOpType (OpType e)
 
bool tt::target::ttnn::VerifyOpType (::flatbuffers::Verifier &verifier, const void *obj, OpType type)
 
bool tt::target::ttnn::VerifyOpTypeVector (::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset< void >> *values, const ::flatbuffers::Vector< OpType > *types)
 
template<>
const tt::target::ttnn::AllGatherOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::AllGatherOp > () const
 
template<>
const tt::target::ttnn::CollectivePermuteOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::CollectivePermuteOp > () const
 
template<>
const tt::target::ttnn::MeshShardOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::MeshShardOp > () const
 
template<>
const tt::target::ttnn::ReduceScatterOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::ReduceScatterOp > () const
 
template<>
const tt::target::ttnn::GetDeviceOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::GetDeviceOp > () const
 
template<>
const tt::target::ttnn::PrepareConv2dWeightsOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::PrepareConv2dWeightsOp > () const
 
template<>
const tt::target::ttnn::Conv2dOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::Conv2dOp > () const
 
template<>
const tt::target::ttnn::ConvTranspose2dOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::ConvTranspose2dOp > () const
 
template<>
const tt::target::ttnn::CpuOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::CpuOp > () const
 
template<>
const tt::target::ttnn::ArangeOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::ArangeOp > () const
 
template<>
const tt::target::ttnn::ConstantOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::ConstantOp > () const
 
template<>
const tt::target::ttnn::ConstructTensorOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::ConstructTensorOp > () const
 
template<>
const tt::target::ttnn::EmptyOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::EmptyOp > () const
 
template<>
const tt::target::ttnn::FullOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::FullOp > () const
 
template<>
const tt::target::ttnn::NamedFullOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::NamedFullOp > () const
 
template<>
const tt::target::ttnn::ConcatOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::ConcatOp > () const
 
template<>
const tt::target::ttnn::PadOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::PadOp > () const
 
template<>
const tt::target::ttnn::PermuteOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::PermuteOp > () const
 
template<>
const tt::target::ttnn::RepeatInterleaveOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::RepeatInterleaveOp > () const
 
template<>
const tt::target::ttnn::RepeatOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::RepeatOp > () const
 
template<>
const tt::target::ttnn::ReshapeOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::ReshapeOp > () const
 
template<>
const tt::target::ttnn::SliceOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::SliceOp > () const
 
template<>
const tt::target::ttnn::TransposeOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::TransposeOp > () const
 
template<>
const tt::target::ttnn::DeallocateOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::DeallocateOp > () const
 
template<>
const tt::target::ttnn::EltwiseBinaryOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::EltwiseBinaryOp > () const
 
template<>
const tt::target::ttnn::EltwiseBinaryCompositeOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::EltwiseBinaryCompositeOp > () const
 
template<>
const tt::target::ttnn::EltwiseTernaryWhereOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::EltwiseTernaryWhereOp > () const
 
template<>
const tt::target::ttnn::EltwiseQuantizationOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::EltwiseQuantizationOp > () const
 
template<>
const tt::target::ttnn::EltwiseUnaryOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::EltwiseUnaryOp > () const
 
template<>
const tt::target::ttnn::EltwiseUnaryCompositeOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::EltwiseUnaryCompositeOp > () const
 
template<>
const tt::target::ttnn::EmbeddingBackwardOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::EmbeddingBackwardOp > () const
 
template<>
const tt::target::ttnn::EmbeddingOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::EmbeddingOp > () const
 
template<>
const tt::target::ttnn::FillCacheOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::FillCacheOp > () const
 
template<>
const tt::target::ttnn::UpdateCacheOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::UpdateCacheOp > () const
 
template<>
const tt::target::ttnn::FromDeviceOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::FromDeviceOp > () const
 
template<>
const tt::target::ttnn::ToDeviceOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::ToDeviceOp > () const
 
template<>
const tt::target::ttnn::ToDTypeOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::ToDTypeOp > () const
 
template<>
const tt::target::ttnn::ToLayoutOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::ToLayoutOp > () const
 
template<>
const tt::target::ttnn::ToMemoryConfigOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::ToMemoryConfigOp > () const
 
template<>
const tt::target::ttnn::TypecastOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::TypecastOp > () const
 
template<>
const tt::target::ttnn::LinearOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::LinearOp > () const
 
template<>
const tt::target::ttnn::MatmulOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::MatmulOp > () const
 
template<>
const tt::target::ttnn::MorehCumSumOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::MorehCumSumOp > () const
 
template<>
const tt::target::ttnn::SoftmaxOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::SoftmaxOp > () const
 
template<>
const tt::target::ttnn::MaxPool2dOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::MaxPool2dOp > () const
 
template<>
const tt::target::ttnn::UpsampleOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::UpsampleOp > () const
 
template<>
const tt::target::ttnn::ReductionArgMaxOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::ReductionArgMaxOp > () const
 
template<>
const tt::target::ttnn::ReductionOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::ReductionOp > () const
 
template<>
const tt::target::ttnn::ReductionProdOp * tt::target::ttnn::Operation::type_as< tt::target::ttnn::ReductionProdOp > () const
 
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, ::flatbuffers::Offset<::flatbuffers::String > loc_info=0)
 
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, const char *loc_info=nullptr)
 
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::ttnn::TensorRef >>> inputs=0, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::TensorRef >>> outputs=0, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::Operation >>> operations=0, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::DynamicLib >>> dylibs=0, ::flatbuffers::Offset< tt::target::DebugInfo > debug_info=0)
 
inline ::flatbuffers::Offset< Program > tt::target::ttnn::CreateProgramDirect (::flatbuffers::FlatBufferBuilder &_fbb, const char *name=nullptr, const std::vector<::flatbuffers::Offset< tt::target::ttnn::TensorRef >> *inputs=nullptr, const std::vector<::flatbuffers::Offset< tt::target::ttnn::TensorRef >> *outputs=nullptr, const std::vector<::flatbuffers::Offset< tt::target::ttnn::Operation >> *operations=nullptr, const std::vector<::flatbuffers::Offset< tt::target::DynamicLib >> *dylibs=nullptr, ::flatbuffers::Offset< tt::target::DebugInfo > debug_info=0)
 

Variables

 VT_TYPE_TYPE = 4
 
 VT_TYPE = 6
 
 VT_DEBUG_INFO = 8
 
 VT_NAME = 4
 
 VT_INPUTS = 6
 
 VT_OUTPUTS = 8
 
 VT_OPERATIONS = 10
 
 VT_DYLIBS = 12
 

Variable Documentation

◆ VT_DEBUG_INFO

VT_DEBUG_INFO = 8

◆ VT_DYLIBS

VT_DYLIBS = 12

◆ VT_INPUTS

VT_INPUTS = 6

◆ VT_NAME

VT_NAME = 4

◆ VT_OPERATIONS

VT_OPERATIONS = 10

◆ VT_OUTPUTS

VT_OUTPUTS = 8

◆ VT_TYPE

VT_TYPE = 6

◆ VT_TYPE_TYPE

VT_TYPE_TYPE = 4