|
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) |
|