TT-MLIR
Classes | Namespaces | Enumerations | Functions | Variables
types_generated.h File Reference
#include "flatbuffers/flatbuffers.h"
#include "types_generated.h"

Go to the source code of this file.

Classes

struct  tt::target::ttnn::DistributedTensorConfigTraits< T >
 
struct  tt::target::ttnn::DistributedTensorConfigTraits< tt::target::ttnn::ReplicateTensor >
 
struct  tt::target::ttnn::DistributedTensorConfigTraits< tt::target::ttnn::ShardTensor >
 
struct  tt::target::ttnn::DistributedTensorConfigTraits< tt::target::ttnn::ShardTensor2D >
 
struct  tt::target::ttnn::DistributedTensorConfigTraits< tt::target::ttnn::AllGatherTensor >
 
struct  tt::target::ttnn::FLATBUFFERS_FINAL_CLASS
 
struct  tt::target::ttnn::ReplicateTensorBuilder
 
struct  tt::target::ttnn::ReplicateTensor::Traits
 
struct  tt::target::ttnn::ShardTensorBuilder
 
struct  tt::target::ttnn::ShardTensor::Traits
 
struct  tt::target::ttnn::ShardTensor2DBuilder
 
struct  tt::target::ttnn::ShardTensor2D::Traits
 
struct  tt::target::ttnn::AllGatherTensorBuilder
 
struct  tt::target::ttnn::AllGatherTensor::Traits
 
struct  tt::target::ttnn::DistributionStrategyBuilder
 
struct  tt::target::ttnn::DistributionStrategy::Traits
 
struct  tt::target::ttnn::ShardSpecBuilder
 
struct  tt::target::ttnn::ShardSpec::Traits
 
struct  tt::target::ttnn::MemoryConfigBuilder
 
struct  tt::target::ttnn::MemoryConfig::Traits
 
struct  tt::target::ttnn::MemoryDescBuilder
 
struct  tt::target::ttnn::MemoryDesc::Traits
 
struct  tt::target::ttnn::LayoutDescBuilder
 
struct  tt::target::ttnn::LayoutDesc::Traits
 
struct  tt::target::ttnn::TensorDescBuilder
 
struct  tt::target::ttnn::TensorDesc::Traits
 
struct  tt::target::ttnn::TensorRefBuilder
 
struct  tt::target::ttnn::TensorRef::Traits
 

Namespaces

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

Enumerations

enum class  tt::target::ttnn::TensorMemoryLayout : uint16_t {
  tt::target::ttnn::Interleaved = 0 , tt::target::ttnn::SingleBank = 1 , tt::target::ttnn::HeightSharded = 2 , tt::target::ttnn::WidthSharded = 3 ,
  tt::target::ttnn::BlockSharded = 4 , tt::target::ttnn::MIN = Interleaved , tt::target::ttnn::MAX = BlockSharded
}
 
enum class  tt::target::ttnn::StorageType : uint16_t {
  tt::target::ttnn::Owned = 0 , tt::target::ttnn::Device = 1 , tt::target::ttnn::Borrowed = 2 , tt::target::ttnn::MultiDevice = 3 ,
  tt::target::ttnn::MultiDeviceHost = 4 , tt::target::ttnn::MIN = Owned , tt::target::ttnn::MAX = MultiDeviceHost
}
 
enum class  tt::target::ttnn::MeshShardDirection : uint32_t { tt::target::ttnn::FullToShardShape = 0 , tt::target::ttnn::ShardToFullShape = 1 , tt::target::ttnn::MIN = FullToShardShape , tt::target::ttnn::MAX = ShardToFullShape }
 
enum class  tt::target::ttnn::MeshShardType : uint32_t {
  tt::target::ttnn::Manual = 0 , tt::target::ttnn::Replicate = 1 , tt::target::ttnn::Maximal = 2 , tt::target::ttnn::Devices = 3 ,
  tt::target::ttnn::MIN = Manual , tt::target::ttnn::MAX = Devices
}
 
enum class  tt::target::ttnn::DistributedTensorConfig : uint8_t {
  tt::target::ttnn::NONE = 0 , tt::target::ttnn::ReplicateTensor = 1 , tt::target::ttnn::ShardTensor = 2 , tt::target::ttnn::ShardTensor2D = 3 ,
  tt::target::ttnn::AllGatherTensor = 4 , tt::target::ttnn::MIN = NONE , tt::target::ttnn::MAX = AllGatherTensor
}
 

Functions

const TensorMemoryLayout(& tt::target::ttnn::EnumValuesTensorMemoryLayout ())[5]
 
const char *const * tt::target::ttnn::EnumNamesTensorMemoryLayout ()
 
const char * tt::target::ttnn::EnumNameTensorMemoryLayout (TensorMemoryLayout e)
 
const StorageType(& tt::target::ttnn::EnumValuesStorageType ())[5]
 
const char *const * tt::target::ttnn::EnumNamesStorageType ()
 
const char * tt::target::ttnn::EnumNameStorageType (StorageType e)
 
const MeshShardDirection(& tt::target::ttnn::EnumValuesMeshShardDirection ())[2]
 
const char *const * tt::target::ttnn::EnumNamesMeshShardDirection ()
 
const char * tt::target::ttnn::EnumNameMeshShardDirection (MeshShardDirection e)
 
const MeshShardType(& tt::target::ttnn::EnumValuesMeshShardType ())[4]
 
const char *const * tt::target::ttnn::EnumNamesMeshShardType ()
 
const char * tt::target::ttnn::EnumNameMeshShardType (MeshShardType e)
 
const DistributedTensorConfig(& tt::target::ttnn::EnumValuesDistributedTensorConfig ())[5]
 
const char *const * tt::target::ttnn::EnumNamesDistributedTensorConfig ()
 
const char * tt::target::ttnn::EnumNameDistributedTensorConfig (DistributedTensorConfig e)
 
bool tt::target::ttnn::VerifyDistributedTensorConfig (::flatbuffers::Verifier &verifier, const void *obj, DistributedTensorConfig type)
 
bool tt::target::ttnn::VerifyDistributedTensorConfigVector (::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset< void >> *values, const ::flatbuffers::Vector< DistributedTensorConfig > *types)
 
inline ::flatbuffers::Offset< ReplicateTensor > tt::target::ttnn::CreateReplicateTensor (::flatbuffers::FlatBufferBuilder &_fbb, uint32_t replication_factor=0)
 
inline ::flatbuffers::Offset< ShardTensor > tt::target::ttnn::CreateShardTensor (::flatbuffers::FlatBufferBuilder &_fbb, uint32_t shard_dim=0)
 
inline ::flatbuffers::Offset< ShardTensor2D > tt::target::ttnn::CreateShardTensor2D (::flatbuffers::FlatBufferBuilder &_fbb, const tt::target::Dim2d *shard_mesh=nullptr)
 
inline ::flatbuffers::Offset< AllGatherTensor > tt::target::ttnn::CreateAllGatherTensor (::flatbuffers::FlatBufferBuilder &_fbb)
 
template<>
const tt::target::ttnn::ReplicateTensor * tt::target::ttnn::DistributionStrategy::strategy_as< tt::target::ttnn::ReplicateTensor > () const
 
template<>
const tt::target::ttnn::ShardTensor * tt::target::ttnn::DistributionStrategy::strategy_as< tt::target::ttnn::ShardTensor > () const
 
template<>
const tt::target::ttnn::ShardTensor2D * tt::target::ttnn::DistributionStrategy::strategy_as< tt::target::ttnn::ShardTensor2D > () const
 
template<>
const tt::target::ttnn::AllGatherTensor * tt::target::ttnn::DistributionStrategy::strategy_as< tt::target::ttnn::AllGatherTensor > () const
 
inline ::flatbuffers::Offset< DistributionStrategy > tt::target::ttnn::CreateDistributionStrategy (::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::DistributedTensorConfig strategy_type=tt::target::ttnn::DistributedTensorConfig::NONE, ::flatbuffers::Offset< void > strategy=0)
 
inline ::flatbuffers::Offset< ShardSpec > tt::target::ttnn::CreateShardSpec (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset<::flatbuffers::Vector< const tt::target::Dim2dRange * >> grid=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> shard_shape=0)
 
inline ::flatbuffers::Offset< ShardSpec > tt::target::ttnn::CreateShardSpecDirect (::flatbuffers::FlatBufferBuilder &_fbb, const std::vector< tt::target::Dim2dRange > *grid=nullptr, const std::vector< int32_t > *shard_shape=nullptr)
 
inline ::flatbuffers::Offset< MemoryConfig > tt::target::ttnn::CreateMemoryConfig (::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::TensorMemoryLayout tensor_memory_layout=tt::target::ttnn::TensorMemoryLayout::Interleaved, tt::target::BufferType buffer_type=tt::target::BufferType::DRAM, ::flatbuffers::Offset< tt::target::ttnn::ShardSpec > shard_spec=0)
 
inline ::flatbuffers::Offset< MemoryDesc > tt::target::ttnn::CreateMemoryDesc (::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::StorageType storage_type=tt::target::ttnn::StorageType::Owned, const tt::target::Dim2d *tile_shape=nullptr, tt::target::DataType data_type=tt::target::DataType::Float32, ::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memory_config=0, uint64_t size=0)
 
inline ::flatbuffers::Offset< LayoutDesc > tt::target::ttnn::CreateLayoutDesc (::flatbuffers::FlatBufferBuilder &_fbb, tt::target::OOBVal oob_val=tt::target::OOBVal::Undef, ::flatbuffers::Offset< tt::target::ttnn::MemoryDesc > memory_desc=0, ::flatbuffers::Offset< tt::target::ttnn::DistributionStrategy > strategy=0)
 
inline ::flatbuffers::Offset< TensorDesc > tt::target::ttnn::CreateTensorDesc (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> shape=0, ::flatbuffers::Offset< tt::target::ttnn::LayoutDesc > layout=0)
 
inline ::flatbuffers::Offset< TensorDesc > tt::target::ttnn::CreateTensorDescDirect (::flatbuffers::FlatBufferBuilder &_fbb, const std::vector< int32_t > *shape=nullptr, ::flatbuffers::Offset< tt::target::ttnn::LayoutDesc > layout=0)
 
inline ::flatbuffers::Offset< TensorRef > tt::target::ttnn::CreateTensorRef (::flatbuffers::FlatBufferBuilder &_fbb, uint32_t global_id=0, uint64_t size=0, ::flatbuffers::Offset< tt::target::ttnn::TensorDesc > desc=0)
 

Variables

 VT_STRATEGY_TYPE = 4
 
 VT_GRID = 4
 
 VT_TENSOR_MEMORY_LAYOUT = 4
 
 VT_BUFFER_TYPE = 6
 
 VT_STORAGE_TYPE = 4
 
 VT_TILE_SHAPE = 6
 
 VT_DATA_TYPE = 8
 
 VT_MEMORY_CONFIG = 10
 
 VT_OOB_VAL = 4
 
 VT_MEMORY_DESC = 6
 
 VT_SHAPE = 4
 
 VT_GLOBAL_ID = 4
 
 VT_SIZE = 6
 

Variable Documentation

◆ VT_BUFFER_TYPE

VT_BUFFER_TYPE = 6

◆ VT_DATA_TYPE

VT_DATA_TYPE = 8

◆ VT_GLOBAL_ID

VT_GLOBAL_ID = 4

◆ VT_GRID

VT_GRID = 4

◆ VT_MEMORY_CONFIG

VT_MEMORY_CONFIG = 10

◆ VT_MEMORY_DESC

VT_MEMORY_DESC = 6

◆ VT_OOB_VAL

VT_OOB_VAL = 4

◆ VT_SHAPE

VT_SHAPE = 4

◆ VT_SIZE

VT_SIZE = 6

◆ VT_STORAGE_TYPE

VT_STORAGE_TYPE = 4

◆ VT_STRATEGY_TYPE

VT_STRATEGY_TYPE = 4

◆ VT_TENSOR_MEMORY_LAYOUT

VT_TENSOR_MEMORY_LAYOUT = 4

◆ VT_TILE_SHAPE

VT_TILE_SHAPE = 6