4 #ifndef FLATBUFFERS_GENERATED_TYPES_TT_TARGET_TTNN_H_
5 #define FLATBUFFERS_GENERATED_TYPES_TT_TARGET_TTNN_H_
7 #include "flatbuffers/flatbuffers.h"
11 static_assert(FLATBUFFERS_VERSION_MAJOR == 24 &&
12 FLATBUFFERS_VERSION_MINOR == 3 &&
13 FLATBUFFERS_VERSION_REVISION == 25,
14 "Non-compatible flatbuffers version included");
27 struct CoreRangeSetBuilder;
30 struct ReplicateTensorBuilder;
33 struct ShardTensorBuilder;
36 struct ShardTensor2DBuilder;
39 struct AllGatherTensorBuilder;
41 struct DistributionStrategy;
42 struct DistributionStrategyBuilder;
45 struct ShardSpecBuilder;
48 struct MemoryConfigBuilder;
51 struct MemoryDescBuilder;
54 struct LayoutDescBuilder;
57 struct TensorDescBuilder;
60 struct TensorRefBuilder;
82 static const char *
const names[5] = {
94 const size_t index =
static_cast<size_t>(e);
116 static const char *
const names[4] = {
127 const size_t index =
static_cast<size_t>(e);
147 static const char *
const names[3] = {
157 const size_t index =
static_cast<size_t>(e);
181 static const char *
const names[5] = {
193 const size_t index =
static_cast<size_t>(e);
213 static const char *
const names[3] = {
223 const size_t index =
static_cast<size_t>(e);
243 static const char *
const names[3] = {
253 const size_t index =
static_cast<size_t>(e);
279 static const char *
const names[6] = {
292 const size_t index =
static_cast<size_t>(e);
317 bool VerifyDistributedTensorConfigVector(::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset<void>> *values, const ::flatbuffers::Vector<DistributedTensorConfig> *types);
326 static FLATBUFFERS_CONSTEXPR_CPP11
const char *GetFullyQualifiedName() {
327 return "tt.target.ttnn.CoreCoord";
333 CoreCoord(uint64_t _x, uint64_t _y)
334 : x_(::flatbuffers::EndianScalar(_x)),
335 y_(::flatbuffers::EndianScalar(_y)) {
338 return ::flatbuffers::EndianScalar(x_);
341 return ::flatbuffers::EndianScalar(y_);
352 tt::target::ttnn::CoreCoord start_coord_;
353 tt::target::ttnn::CoreCoord end_coord_;
357 static FLATBUFFERS_CONSTEXPR_CPP11
const char *GetFullyQualifiedName() {
358 return "tt.target.ttnn.CoreRange";
364 CoreRange(
const tt::target::ttnn::CoreCoord &_start_coord,
const tt::target::ttnn::CoreCoord &_end_coord)
365 : start_coord_(_start_coord),
366 end_coord_(_end_coord) {
368 const tt::target::ttnn::CoreCoord &start_coord()
const {
371 const tt::target::ttnn::CoreCoord &end_coord()
const {
385 return "tt.target.ttnn.CoreRangeSet";
387 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
390 const ::flatbuffers::Vector<const tt::target::ttnn::CoreRange *> *
core_ranges()
const {
391 return GetPointer<const ::flatbuffers::Vector<const tt::target::ttnn::CoreRange *> *>(VT_CORE_RANGES);
393 bool Verify(::flatbuffers::Verifier &verifier)
const {
394 return VerifyTableStart(verifier) &&
395 VerifyOffset(verifier, VT_CORE_RANGES) &&
396 verifier.VerifyVector(core_ranges()) &&
403 ::flatbuffers::FlatBufferBuilder &
fbb_;
405 void add_core_ranges(::flatbuffers::Offset<::flatbuffers::Vector<const tt::target::ttnn::CoreRange *>> core_ranges) {
406 fbb_.AddOffset(CoreRangeSet::VT_CORE_RANGES, core_ranges);
412 ::flatbuffers::Offset<CoreRangeSet>
Finish() {
414 auto o = ::flatbuffers::Offset<CoreRangeSet>(end);
420 ::flatbuffers::FlatBufferBuilder &_fbb,
421 ::flatbuffers::Offset<::flatbuffers::Vector<const tt::target::ttnn::CoreRange *>> core_ranges = 0) {
433 ::flatbuffers::FlatBufferBuilder &_fbb,
434 const std::vector<tt::target::ttnn::CoreRange> *core_ranges =
nullptr) {
435 auto core_ranges__ = core_ranges ? _fbb.CreateVectorOfStructs<tt::target::ttnn::CoreRange>(*core_ranges) : 0;
445 return "tt.target.ttnn.ReplicateTensor";
447 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
448 VT_REPLICATION_FACTOR = 4
451 return GetField<uint32_t>(VT_REPLICATION_FACTOR, 0);
453 bool Verify(::flatbuffers::Verifier &verifier)
const {
454 return VerifyTableStart(verifier) &&
455 VerifyField<uint32_t>(verifier, VT_REPLICATION_FACTOR, 4) &&
462 ::flatbuffers::FlatBufferBuilder &
fbb_;
465 fbb_.AddElement<uint32_t>(ReplicateTensor::VT_REPLICATION_FACTOR, replication_factor, 0);
471 ::flatbuffers::Offset<ReplicateTensor>
Finish() {
473 auto o = ::flatbuffers::Offset<ReplicateTensor>(end);
479 ::flatbuffers::FlatBufferBuilder &_fbb,
480 uint32_t replication_factor = 0) {
495 return "tt.target.ttnn.ShardTensor";
497 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
501 return GetField<uint32_t>(VT_SHARD_DIM, 0);
503 bool Verify(::flatbuffers::Verifier &verifier)
const {
504 return VerifyTableStart(verifier) &&
505 VerifyField<uint32_t>(verifier, VT_SHARD_DIM, 4) &&
512 ::flatbuffers::FlatBufferBuilder &
fbb_;
515 fbb_.AddElement<uint32_t>(ShardTensor::VT_SHARD_DIM, shard_dim, 0);
521 ::flatbuffers::Offset<ShardTensor>
Finish() {
523 auto o = ::flatbuffers::Offset<ShardTensor>(end);
529 ::flatbuffers::FlatBufferBuilder &_fbb,
530 uint32_t shard_dim = 0) {
545 return "tt.target.ttnn.ShardTensor2D";
547 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
551 return GetStruct<const tt::target::Dim2d *>(VT_SHARD_MESH);
553 bool Verify(::flatbuffers::Verifier &verifier)
const {
554 return VerifyTableStart(verifier) &&
555 VerifyField<tt::target::Dim2d>(verifier, VT_SHARD_MESH, 4) &&
562 ::flatbuffers::FlatBufferBuilder &
fbb_;
565 fbb_.AddStruct(ShardTensor2D::VT_SHARD_MESH, shard_mesh);
571 ::flatbuffers::Offset<ShardTensor2D>
Finish() {
573 auto o = ::flatbuffers::Offset<ShardTensor2D>(end);
579 ::flatbuffers::FlatBufferBuilder &_fbb,
580 const tt::target::Dim2d *shard_mesh =
nullptr) {
595 return "tt.target.ttnn.AllGatherTensor";
597 bool Verify(::flatbuffers::Verifier &verifier)
const {
598 return VerifyTableStart(verifier) &&
605 ::flatbuffers::FlatBufferBuilder &
fbb_;
611 ::flatbuffers::Offset<AllGatherTensor>
Finish() {
613 auto o = ::flatbuffers::Offset<AllGatherTensor>(end);
619 ::flatbuffers::FlatBufferBuilder &_fbb) {
633 return "tt.target.ttnn.DistributionStrategy";
635 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
643 return GetPointer<const void *>(VT_STRATEGY);
658 bool Verify(::flatbuffers::Verifier &verifier)
const {
659 return VerifyTableStart(verifier) &&
661 VerifyOffset(verifier, VT_STRATEGY) &&
667 template<>
inline const tt::target::ttnn::ReplicateTensor *DistributionStrategy::strategy_as<tt::target::ttnn::ReplicateTensor>()
const {
668 return strategy_as_ReplicateTensor();
671 template<>
inline const tt::target::ttnn::ShardTensor *DistributionStrategy::strategy_as<tt::target::ttnn::ShardTensor>()
const {
672 return strategy_as_ShardTensor();
675 template<>
inline const tt::target::ttnn::ShardTensor2D *DistributionStrategy::strategy_as<tt::target::ttnn::ShardTensor2D>()
const {
676 return strategy_as_ShardTensor2D();
679 template<>
inline const tt::target::ttnn::AllGatherTensor *DistributionStrategy::strategy_as<tt::target::ttnn::AllGatherTensor>()
const {
680 return strategy_as_AllGatherTensor();
685 ::flatbuffers::FlatBufferBuilder &
fbb_;
691 fbb_.AddOffset(DistributionStrategy::VT_STRATEGY, strategy);
697 ::flatbuffers::Offset<DistributionStrategy>
Finish() {
699 auto o = ::flatbuffers::Offset<DistributionStrategy>(end);
705 ::flatbuffers::FlatBufferBuilder &_fbb,
707 ::flatbuffers::Offset<void> strategy = 0) {
715 using type = DistributionStrategy;
723 return "tt.target.ttnn.ShardSpec";
725 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
730 VT_PHYSICAL_SHARD_SHAPE = 12
735 const ::flatbuffers::Vector<int32_t> *
shape()
const {
736 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_SHAPE);
745 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(VT_PHYSICAL_SHARD_SHAPE);
747 bool Verify(::flatbuffers::Verifier &verifier)
const {
748 return VerifyTableStart(verifier) &&
750 verifier.VerifyTable(core_range_set()) &&
752 verifier.VerifyVector(shape()) &&
754 VerifyField<uint32_t>(verifier,
VT_MODE, 4) &&
755 VerifyOffset(verifier, VT_PHYSICAL_SHARD_SHAPE) &&
756 verifier.VerifyVector(physical_shard_shape()) &&
763 ::flatbuffers::FlatBufferBuilder &
fbb_;
768 void add_shape(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> shape) {
778 fbb_.AddOffset(ShardSpec::VT_PHYSICAL_SHARD_SHAPE, physical_shard_shape);
784 ::flatbuffers::Offset<ShardSpec>
Finish() {
786 auto o = ::flatbuffers::Offset<ShardSpec>(end);
792 ::flatbuffers::FlatBufferBuilder &_fbb,
793 ::flatbuffers::Offset<tt::target::ttnn::CoreRangeSet> core_range_set = 0,
794 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> shape = 0,
797 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> physical_shard_shape = 0) {
813 ::flatbuffers::FlatBufferBuilder &_fbb,
814 ::flatbuffers::Offset<tt::target::ttnn::CoreRangeSet> core_range_set = 0,
815 const std::vector<int32_t> *shape =
nullptr,
818 const std::vector<int32_t> *physical_shard_shape =
nullptr) {
819 auto shape__ = shape ? _fbb.CreateVector<int32_t>(*shape) : 0;
820 auto physical_shard_shape__ = physical_shard_shape ? _fbb.CreateVector<int32_t>(*physical_shard_shape) : 0;
827 physical_shard_shape__);
834 return "tt.target.ttnn.MemoryConfig";
836 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
848 return GetPointer<const tt::target::ttnn::ShardSpec *>(
VT_SHARD_SPEC);
850 bool Verify(::flatbuffers::Verifier &verifier)
const {
851 return VerifyTableStart(verifier) &&
855 verifier.VerifyTable(shard_spec()) &&
862 ::flatbuffers::FlatBufferBuilder &
fbb_;
870 void add_shard_spec(::flatbuffers::Offset<tt::target::ttnn::ShardSpec> shard_spec) {
877 ::flatbuffers::Offset<MemoryConfig>
Finish() {
879 auto o = ::flatbuffers::Offset<MemoryConfig>(end);
885 ::flatbuffers::FlatBufferBuilder &_fbb,
888 ::flatbuffers::Offset<tt::target::ttnn::ShardSpec> shard_spec = 0) {
905 return "tt.target.ttnn.MemoryDesc";
907 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
924 return GetPointer<const tt::target::ttnn::MemoryConfig *>(
VT_MEMORY_CONFIG);
927 return GetField<uint64_t>(
VT_SIZE, 0);
929 bool Verify(::flatbuffers::Verifier &verifier)
const {
930 return VerifyTableStart(verifier) &&
932 VerifyField<tt::target::Dim2d>(verifier,
VT_TILE_SHAPE, 4) &&
935 verifier.VerifyTable(memory_config()) &&
936 VerifyField<uint64_t>(verifier,
VT_SIZE, 8) &&
943 ::flatbuffers::FlatBufferBuilder &
fbb_;
964 ::flatbuffers::Offset<MemoryDesc>
Finish() {
966 auto o = ::flatbuffers::Offset<MemoryDesc>(end);
972 ::flatbuffers::FlatBufferBuilder &_fbb,
974 const tt::target::Dim2d *tile_shape =
nullptr,
976 ::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> memory_config = 0,
996 return "tt.target.ttnn.LayoutDesc";
998 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1007 return GetPointer<const tt::target::ttnn::MemoryDesc *>(
VT_MEMORY_DESC);
1009 const tt::target::ttnn::DistributionStrategy *
strategy()
const {
1010 return GetPointer<const tt::target::ttnn::DistributionStrategy *>(VT_STRATEGY);
1012 bool Verify(::flatbuffers::Verifier &verifier)
const {
1013 return VerifyTableStart(verifier) &&
1014 VerifyField<uint16_t>(verifier,
VT_OOB_VAL, 2) &&
1016 verifier.VerifyTable(memory_desc()) &&
1017 VerifyOffset(verifier, VT_STRATEGY) &&
1018 verifier.VerifyTable(strategy()) &&
1019 verifier.EndTable();
1025 ::flatbuffers::FlatBufferBuilder &
fbb_;
1033 void add_strategy(::flatbuffers::Offset<tt::target::ttnn::DistributionStrategy> strategy) {
1034 fbb_.AddOffset(LayoutDesc::VT_STRATEGY, strategy);
1042 auto o = ::flatbuffers::Offset<LayoutDesc>(end);
1048 ::flatbuffers::FlatBufferBuilder &_fbb,
1050 ::flatbuffers::Offset<tt::target::ttnn::MemoryDesc> memory_desc = 0,
1051 ::flatbuffers::Offset<tt::target::ttnn::DistributionStrategy> strategy = 0) {
1056 return builder_.
Finish();
1068 return "tt.target.ttnn.TensorDesc";
1070 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1075 const ::flatbuffers::Vector<int32_t> *
shape()
const {
1076 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_SHAPE);
1079 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_MESH_SHAPE);
1081 const tt::target::ttnn::LayoutDesc *
layout()
const {
1082 return GetPointer<const tt::target::ttnn::LayoutDesc *>(
VT_LAYOUT);
1084 bool Verify(::flatbuffers::Verifier &verifier)
const {
1085 return VerifyTableStart(verifier) &&
1086 VerifyOffset(verifier,
VT_SHAPE) &&
1087 verifier.VerifyVector(shape()) &&
1089 verifier.VerifyVector(mesh_shape()) &&
1091 verifier.VerifyTable(layout()) &&
1092 verifier.EndTable();
1098 ::flatbuffers::FlatBufferBuilder &
fbb_;
1100 void add_shape(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> shape) {
1103 void add_mesh_shape(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> mesh_shape) {
1106 void add_layout(::flatbuffers::Offset<tt::target::ttnn::LayoutDesc> layout) {
1115 auto o = ::flatbuffers::Offset<TensorDesc>(end);
1121 ::flatbuffers::FlatBufferBuilder &_fbb,
1122 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> shape = 0,
1123 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> mesh_shape = 0,
1124 ::flatbuffers::Offset<tt::target::ttnn::LayoutDesc> layout = 0) {
1129 return builder_.
Finish();
1138 ::flatbuffers::FlatBufferBuilder &_fbb,
1139 const std::vector<int32_t> *shape =
nullptr,
1140 const std::vector<int32_t> *mesh_shape =
nullptr,
1141 ::flatbuffers::Offset<tt::target::ttnn::LayoutDesc> layout = 0) {
1142 auto shape__ = shape ? _fbb.CreateVector<int32_t>(*shape) : 0;
1143 auto mesh_shape__ = mesh_shape ? _fbb.CreateVector<int32_t>(*mesh_shape) : 0;
1155 return "tt.target.ttnn.TensorRef";
1157 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1166 return GetField<uint64_t>(
VT_SIZE, 0);
1168 const tt::target::ttnn::TensorDesc *
desc()
const {
1169 return GetPointer<const tt::target::ttnn::TensorDesc *>(VT_DESC);
1171 bool Verify(::flatbuffers::Verifier &verifier)
const {
1172 return VerifyTableStart(verifier) &&
1174 VerifyField<uint64_t>(verifier,
VT_SIZE, 8) &&
1175 VerifyOffset(verifier, VT_DESC) &&
1176 verifier.VerifyTable(desc()) &&
1177 verifier.EndTable();
1183 ::flatbuffers::FlatBufferBuilder &
fbb_;
1191 void add_desc(::flatbuffers::Offset<tt::target::ttnn::TensorDesc> desc) {
1192 fbb_.AddOffset(TensorRef::VT_DESC, desc);
1200 auto o = ::flatbuffers::Offset<TensorRef>(end);
1206 ::flatbuffers::FlatBufferBuilder &_fbb,
1207 uint32_t global_id = 0,
1209 ::flatbuffers::Offset<tt::target::ttnn::TensorDesc> desc = 0) {
1214 return builder_.
Finish();
1228 auto ptr =
reinterpret_cast<const tt::target::ttnn::ReplicateTensor *
>(obj);
1229 return verifier.VerifyTable(ptr);
1232 auto ptr =
reinterpret_cast<const tt::target::ttnn::ShardTensor *
>(obj);
1233 return verifier.VerifyTable(ptr);
1236 auto ptr =
reinterpret_cast<const tt::target::ttnn::ShardTensor2D *
>(obj);
1237 return verifier.VerifyTable(ptr);
1240 auto ptr =
reinterpret_cast<const tt::target::ttnn::AllGatherTensor *
>(obj);
1241 return verifier.VerifyTable(ptr);
1243 default:
return true;
1247 inline bool VerifyDistributedTensorConfigVector(::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset<void>> *values, const ::flatbuffers::Vector<DistributedTensorConfig> *types) {
1248 if (!values || !types)
return !values && !types;
1249 if (values->size() != types->size())
return false;
1250 for (::flatbuffers::uoffset_t i = 0; i < values->size(); ++i) {
VT_STRATEGY_TYPE
Definition: types_generated.h:636
VT_MESH_SHAPE
Definition: types_generated.h:1072
VT_MODE
Definition: types_generated.h:729
VT_STORAGE_TYPE
Definition: types_generated.h:908
VT_TENSOR_MEMORY_LAYOUT
Definition: types_generated.h:837
VT_BUFFER_TYPE
Definition: types_generated.h:838
VT_OOB_VAL
Definition: types_generated.h:999
VT_ORIENTATION
Definition: types_generated.h:728
VT_MEMORY_DESC
Definition: types_generated.h:1000
VT_MEMORY_CONFIG
Definition: types_generated.h:911
VT_LAYOUT
Definition: creation_generated.h:317
const MeshShardType(& EnumValuesMeshShardType())[4]
Definition: types_generated.h:170
inline ::flatbuffers::Offset< TensorDesc > CreateTensorDesc(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> shape=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> mesh_shape=0, ::flatbuffers::Offset< tt::target::ttnn::LayoutDesc > layout=0)
Definition: types_generated.h:1120
inline ::flatbuffers::Offset< ReplicateTensor > CreateReplicateTensor(::flatbuffers::FlatBufferBuilder &_fbb, uint32_t replication_factor=0)
Definition: types_generated.h:478
inline ::flatbuffers::Offset< ShardSpec > CreateShardSpec(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::CoreRangeSet > core_range_set=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> shape=0, tt::target::ttnn::ShardOrientation orientation=tt::target::ttnn::ShardOrientation::RowMajor, tt::target::ttnn::ShardMode mode=tt::target::ttnn::ShardMode::Physical, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> physical_shard_shape=0)
Definition: types_generated.h:791
MeshShardType
Definition: types_generated.h:161
ShardMode
Definition: types_generated.h:227
const char * EnumNameShardMode(ShardMode e)
Definition: types_generated.h:251
const ShardMode(& EnumValuesShardMode())[2]
Definition: types_generated.h:234
const char * EnumNameMeshShardType(MeshShardType e)
Definition: types_generated.h:191
const char * EnumNameDistributedTensorConfig(DistributedTensorConfig e)
Definition: types_generated.h:290
inline ::flatbuffers::Offset< ShardTensor > CreateShardTensor(::flatbuffers::FlatBufferBuilder &_fbb, uint32_t shard_dim=0)
Definition: types_generated.h:528
ShardOrientation
Definition: types_generated.h:197
const ShardOrientation(& EnumValuesShardOrientation())[2]
Definition: types_generated.h:204
const char *const * EnumNamesDistributedTensorConfig()
Definition: types_generated.h:278
StorageType
Definition: types_generated.h:98
inline ::flatbuffers::Offset< TensorRef > CreateTensorRef(::flatbuffers::FlatBufferBuilder &_fbb, uint32_t global_id=0, uint64_t size=0, ::flatbuffers::Offset< tt::target::ttnn::TensorDesc > desc=0)
Definition: types_generated.h:1205
const DistributedTensorConfig(& EnumValuesDistributedTensorConfig())[5]
Definition: types_generated.h:267
const char *const * EnumNamesShardOrientation()
Definition: types_generated.h:212
FLATBUFFERS_STRUCT_END(CoreCoord, 16)
inline ::flatbuffers::Offset< ShardSpec > CreateShardSpecDirect(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::CoreRangeSet > core_range_set=0, const std::vector< int32_t > *shape=nullptr, tt::target::ttnn::ShardOrientation orientation=tt::target::ttnn::ShardOrientation::RowMajor, tt::target::ttnn::ShardMode mode=tt::target::ttnn::ShardMode::Physical, const std::vector< int32_t > *physical_shard_shape=nullptr)
Definition: types_generated.h:812
FLATBUFFERS_MANUALLY_ALIGNED_STRUCT(8) CoreCoord FLATBUFFERS_FINAL_CLASS
Definition: types_generated.h:319
inline ::flatbuffers::Offset< AllGatherTensor > CreateAllGatherTensor(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:618
const char * EnumNameTensorMemoryLayout(TensorMemoryLayout e)
Definition: types_generated.h:92
inline ::flatbuffers::Offset< MemoryDesc > CreateMemoryDesc(::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::StorageType storage_type=tt::target::ttnn::StorageType::Host, 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)
Definition: types_generated.h:971
const char *const * EnumNamesStorageType()
Definition: types_generated.h:115
DistributedTensorConfig
Definition: types_generated.h:257
const char *const * EnumNamesShardMode()
Definition: types_generated.h:242
inline ::flatbuffers::Offset< CoreRangeSet > CreateCoreRangeSet(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset<::flatbuffers::Vector< const tt::target::ttnn::CoreRange * >> core_ranges=0)
Definition: types_generated.h:419
const TensorMemoryLayout(& EnumValuesTensorMemoryLayout())[4]
Definition: types_generated.h:71
TensorMemoryLayout
Definition: types_generated.h:62
inline ::flatbuffers::Offset< ShardTensor2D > CreateShardTensor2D(::flatbuffers::FlatBufferBuilder &_fbb, const tt::target::Dim2d *shard_mesh=nullptr)
Definition: types_generated.h:578
const char * EnumNameStorageType(StorageType e)
Definition: types_generated.h:125
inline ::flatbuffers::Offset< DistributionStrategy > CreateDistributionStrategy(::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::DistributedTensorConfig strategy_type=tt::target::ttnn::DistributedTensorConfig::NONE, ::flatbuffers::Offset< void > strategy=0)
Definition: types_generated.h:704
const char * EnumNameMeshShardDirection(MeshShardDirection e)
Definition: types_generated.h:155
const char *const * EnumNamesTensorMemoryLayout()
Definition: types_generated.h:81
inline ::flatbuffers::Offset< CoreRangeSet > CreateCoreRangeSetDirect(::flatbuffers::FlatBufferBuilder &_fbb, const std::vector< tt::target::ttnn::CoreRange > *core_ranges=nullptr)
Definition: types_generated.h:432
const MeshShardDirection(& EnumValuesMeshShardDirection())[2]
Definition: types_generated.h:138
bool VerifyDistributedTensorConfig(::flatbuffers::Verifier &verifier, const void *obj, DistributedTensorConfig type)
Definition: types_generated.h:1222
bool VerifyDistributedTensorConfigVector(::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset< void >> *values, const ::flatbuffers::Vector< DistributedTensorConfig > *types)
Definition: types_generated.h:1247
const char *const * EnumNamesMeshShardType()
Definition: types_generated.h:180
const char * EnumNameShardOrientation(ShardOrientation e)
Definition: types_generated.h:221
const StorageType(& EnumValuesStorageType())[3]
Definition: types_generated.h:106
inline ::flatbuffers::Offset< MemoryConfig > 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)
Definition: types_generated.h:884
inline ::flatbuffers::Offset< LayoutDesc > 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)
Definition: types_generated.h:1047
MeshShardDirection
Definition: types_generated.h:131
inline ::flatbuffers::Offset< TensorDesc > CreateTensorDescDirect(::flatbuffers::FlatBufferBuilder &_fbb, const std::vector< int32_t > *shape=nullptr, const std::vector< int32_t > *mesh_shape=nullptr, ::flatbuffers::Offset< tt::target::ttnn::LayoutDesc > layout=0)
Definition: types_generated.h:1137
const char *const * EnumNamesMeshShardDirection()
Definition: types_generated.h:146
OOBVal
Definition: types_generated.h:162
BufferType
Definition: types_generated.h:301
DataType
Definition: types_generated.h:81
Definition: debug_info_bfbs_hash_generated.h:4
Definition: debug_info_generated.h:36
Definition: types_generated.h:603
AllGatherTensorBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:607
::flatbuffers::Offset< AllGatherTensor > Finish()
Definition: types_generated.h:611
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:605
AllGatherTensor Table
Definition: types_generated.h:604
::flatbuffers::uoffset_t start_
Definition: types_generated.h:606
Definition: types_generated.h:624
AllGatherTensor type
Definition: types_generated.h:625
static constexpr auto Create
Definition: types_generated.h:626
Definition: types_generated.h:346
CoreCoord type
Definition: types_generated.h:347
Definition: types_generated.h:401
::flatbuffers::uoffset_t start_
Definition: types_generated.h:404
::flatbuffers::Offset< CoreRangeSet > Finish()
Definition: types_generated.h:412
CoreRangeSetBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:408
void add_core_ranges(::flatbuffers::Offset<::flatbuffers::Vector< const tt::target::ttnn::CoreRange * >> core_ranges)
Definition: types_generated.h:405
CoreRangeSet Table
Definition: types_generated.h:402
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:403
Definition: types_generated.h:427
CoreRangeSet type
Definition: types_generated.h:428
static constexpr auto Create
Definition: types_generated.h:429
Definition: types_generated.h:377
CoreRange type
Definition: types_generated.h:378
Definition: types_generated.h:296
static const DistributedTensorConfig enum_value
Definition: types_generated.h:297
Definition: types_generated.h:683
void add_strategy_type(tt::target::ttnn::DistributedTensorConfig strategy_type)
Definition: types_generated.h:687
DistributionStrategyBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:693
::flatbuffers::uoffset_t start_
Definition: types_generated.h:686
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:685
DistributionStrategy Table
Definition: types_generated.h:684
void add_strategy(::flatbuffers::Offset< void > strategy)
Definition: types_generated.h:690
::flatbuffers::Offset< DistributionStrategy > Finish()
Definition: types_generated.h:697
Definition: types_generated.h:714
static constexpr auto Create
Definition: types_generated.h:716
DistributionStrategy type
Definition: types_generated.h:715
Definition: binary_generated.h:31
TensorDescBuilder Builder
Definition: types_generated.h:1065
const tt::target::Dim2d * tile_shape() const
Definition: types_generated.h:917
tt::target::ttnn::TensorMemoryLayout tensor_memory_layout() const
Definition: types_generated.h:841
ShardSpecBuilder Builder
Definition: types_generated.h:720
tt::target::BufferType buffer_type() const
Definition: types_generated.h:844
ReplicateTensorBuilder Builder
Definition: types_generated.h:442
const tt::target::ttnn::ShardTensor2D * strategy_as_ShardTensor2D() const
Definition: types_generated.h:652
const tt::target::ttnn::MemoryConfig * memory_config() const
Definition: types_generated.h:923
tt::target::ttnn::ShardMode mode() const
Definition: types_generated.h:741
const void * strategy() const
Definition: types_generated.h:642
const tt::target::ttnn::CoreRangeSet * core_range_set() const
Definition: types_generated.h:732
const ::flatbuffers::Vector< int32_t > * mesh_shape() const
Definition: types_generated.h:1078
const ::flatbuffers::Vector< int32_t > * shape() const
Definition: types_generated.h:735
ShardTensorBuilder Builder
Definition: types_generated.h:492
uint32_t replication_factor() const
Definition: types_generated.h:450
tt::target::OOBVal oob_val() const
Definition: types_generated.h:1003
const tt::target::ttnn::AllGatherTensor * strategy_as_AllGatherTensor() const
Definition: types_generated.h:655
const tt::target::ttnn::DistributionStrategy * strategy() const
Definition: types_generated.h:1009
uint64_t size() const
Definition: types_generated.h:926
tt::target::DataType data_type() const
Definition: types_generated.h:920
TensorRefBuilder Builder
Definition: types_generated.h:1152
LayoutDescBuilder Builder
Definition: types_generated.h:993
DistributionStrategyBuilder Builder
Definition: types_generated.h:630
bool Verify(::flatbuffers::Verifier &verifier) const
Definition: types_generated.h:393
ShardTensor2DBuilder Builder
Definition: types_generated.h:542
MemoryConfigBuilder Builder
Definition: types_generated.h:831
MemoryDescBuilder Builder
Definition: types_generated.h:902
const tt::target::ttnn::MemoryDesc * memory_desc() const
Definition: types_generated.h:1006
uint32_t shard_dim() const
Definition: types_generated.h:500
const T * strategy_as() const
const tt::target::Dim2d * shard_mesh() const
Definition: types_generated.h:550
const ::flatbuffers::Vector< const tt::target::ttnn::CoreRange * > * core_ranges() const
Definition: types_generated.h:390
CoreRangeSetBuilder Builder
Definition: types_generated.h:382
const tt::target::ttnn::ReplicateTensor * strategy_as_ReplicateTensor() const
Definition: types_generated.h:646
const tt::target::ttnn::ShardTensor * strategy_as_ShardTensor() const
Definition: types_generated.h:649
const ::flatbuffers::Vector< int32_t > * physical_shard_shape() const
Definition: types_generated.h:744
static FLATBUFFERS_CONSTEXPR_CPP11 const char * GetFullyQualifiedName()
Definition: types_generated.h:384
tt::target::ttnn::ShardOrientation orientation() const
Definition: types_generated.h:738
AllGatherTensorBuilder Builder
Definition: types_generated.h:592
const tt::target::ttnn::TensorDesc * desc() const
Definition: types_generated.h:1168
tt::target::ttnn::DistributedTensorConfig strategy_type() const
Definition: types_generated.h:639
const tt::target::ttnn::LayoutDesc * layout() const
Definition: types_generated.h:1081
tt::target::ttnn::StorageType storage_type() const
Definition: types_generated.h:914
const tt::target::ttnn::ShardSpec * shard_spec() const
Definition: types_generated.h:847
uint32_t global_id() const
Definition: types_generated.h:1162
Definition: types_generated.h:1023
void add_memory_desc(::flatbuffers::Offset< tt::target::ttnn::MemoryDesc > memory_desc)
Definition: types_generated.h:1030
::flatbuffers::uoffset_t start_
Definition: types_generated.h:1026
LayoutDesc Table
Definition: types_generated.h:1024
void add_oob_val(tt::target::OOBVal oob_val)
Definition: types_generated.h:1027
LayoutDescBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:1036
::flatbuffers::Offset< LayoutDesc > Finish()
Definition: types_generated.h:1040
void add_strategy(::flatbuffers::Offset< tt::target::ttnn::DistributionStrategy > strategy)
Definition: types_generated.h:1033
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:1025
Definition: types_generated.h:1059
LayoutDesc type
Definition: types_generated.h:1060
static constexpr auto Create
Definition: types_generated.h:1061
Definition: types_generated.h:860
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:862
MemoryConfigBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:873
MemoryConfig Table
Definition: types_generated.h:861
void add_tensor_memory_layout(tt::target::ttnn::TensorMemoryLayout tensor_memory_layout)
Definition: types_generated.h:864
::flatbuffers::uoffset_t start_
Definition: types_generated.h:863
void add_shard_spec(::flatbuffers::Offset< tt::target::ttnn::ShardSpec > shard_spec)
Definition: types_generated.h:870
::flatbuffers::Offset< MemoryConfig > Finish()
Definition: types_generated.h:877
void add_buffer_type(tt::target::BufferType buffer_type)
Definition: types_generated.h:867
Definition: types_generated.h:896
static constexpr auto Create
Definition: types_generated.h:898
MemoryConfig type
Definition: types_generated.h:897
Definition: types_generated.h:941
MemoryDesc Table
Definition: types_generated.h:942
void add_memory_config(::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memory_config)
Definition: types_generated.h:954
MemoryDescBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:960
void add_size(uint64_t size)
Definition: types_generated.h:957
void add_storage_type(tt::target::ttnn::StorageType storage_type)
Definition: types_generated.h:945
void add_tile_shape(const tt::target::Dim2d *tile_shape)
Definition: types_generated.h:948
::flatbuffers::uoffset_t start_
Definition: types_generated.h:944
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:943
void add_data_type(tt::target::DataType data_type)
Definition: types_generated.h:951
::flatbuffers::Offset< MemoryDesc > Finish()
Definition: types_generated.h:964
Definition: types_generated.h:987
static constexpr auto Create
Definition: types_generated.h:989
MemoryDesc type
Definition: types_generated.h:988
Definition: types_generated.h:460
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:462
ReplicateTensorBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:467
ReplicateTensor Table
Definition: types_generated.h:461
void add_replication_factor(uint32_t replication_factor)
Definition: types_generated.h:464
::flatbuffers::Offset< ReplicateTensor > Finish()
Definition: types_generated.h:471
::flatbuffers::uoffset_t start_
Definition: types_generated.h:463
Definition: types_generated.h:486
static constexpr auto Create
Definition: types_generated.h:488
ReplicateTensor type
Definition: types_generated.h:487
Definition: types_generated.h:761
ShardSpecBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:780
ShardSpec Table
Definition: types_generated.h:762
void add_physical_shard_shape(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> physical_shard_shape)
Definition: types_generated.h:777
void add_orientation(tt::target::ttnn::ShardOrientation orientation)
Definition: types_generated.h:771
::flatbuffers::uoffset_t start_
Definition: types_generated.h:764
::flatbuffers::Offset< ShardSpec > Finish()
Definition: types_generated.h:784
void add_mode(tt::target::ttnn::ShardMode mode)
Definition: types_generated.h:774
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:763
void add_shape(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> shape)
Definition: types_generated.h:768
void add_core_range_set(::flatbuffers::Offset< tt::target::ttnn::CoreRangeSet > core_range_set)
Definition: types_generated.h:765
Definition: types_generated.h:807
ShardSpec type
Definition: types_generated.h:808
static constexpr auto Create
Definition: types_generated.h:809
Definition: types_generated.h:560
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:562
ShardTensor2DBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:567
ShardTensor2D Table
Definition: types_generated.h:561
::flatbuffers::uoffset_t start_
Definition: types_generated.h:563
::flatbuffers::Offset< ShardTensor2D > Finish()
Definition: types_generated.h:571
void add_shard_mesh(const tt::target::Dim2d *shard_mesh)
Definition: types_generated.h:564
Definition: types_generated.h:586
static constexpr auto Create
Definition: types_generated.h:588
ShardTensor2D type
Definition: types_generated.h:587
Definition: types_generated.h:510
ShardTensor Table
Definition: types_generated.h:511
::flatbuffers::Offset< ShardTensor > Finish()
Definition: types_generated.h:521
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:512
ShardTensorBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:517
void add_shard_dim(uint32_t shard_dim)
Definition: types_generated.h:514
::flatbuffers::uoffset_t start_
Definition: types_generated.h:513
Definition: types_generated.h:536
static constexpr auto Create
Definition: types_generated.h:538
ShardTensor type
Definition: types_generated.h:537
Definition: types_generated.h:1096
::flatbuffers::uoffset_t start_
Definition: types_generated.h:1099
void add_layout(::flatbuffers::Offset< tt::target::ttnn::LayoutDesc > layout)
Definition: types_generated.h:1106
TensorDescBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:1109
TensorDesc Table
Definition: types_generated.h:1097
void add_mesh_shape(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> mesh_shape)
Definition: types_generated.h:1103
::flatbuffers::Offset< TensorDesc > Finish()
Definition: types_generated.h:1113
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:1098
void add_shape(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> shape)
Definition: types_generated.h:1100
Definition: types_generated.h:1132
static constexpr auto Create
Definition: types_generated.h:1134
TensorDesc type
Definition: types_generated.h:1133
Definition: types_generated.h:1181
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:1183
void add_global_id(uint32_t global_id)
Definition: types_generated.h:1185
::flatbuffers::uoffset_t start_
Definition: types_generated.h:1184
void add_size(uint64_t size)
Definition: types_generated.h:1188
void add_desc(::flatbuffers::Offset< tt::target::ttnn::TensorDesc > desc)
Definition: types_generated.h:1191
::flatbuffers::Offset< TensorRef > Finish()
Definition: types_generated.h:1198
TensorRef Table
Definition: types_generated.h:1182
TensorRefBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:1194
Definition: types_generated.h:1217
static constexpr auto Create
Definition: types_generated.h:1219
TensorRef type
Definition: types_generated.h:1218