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;
84 static const char *
const names[6] = {
97 const size_t index =
static_cast<size_t>(e);
119 static const char *
const names[4] = {
130 const size_t index =
static_cast<size_t>(e);
150 static const char *
const names[3] = {
160 const size_t index =
static_cast<size_t>(e);
184 static const char *
const names[5] = {
196 const size_t index =
static_cast<size_t>(e);
216 static const char *
const names[3] = {
226 const size_t index =
static_cast<size_t>(e);
246 static const char *
const names[3] = {
256 const size_t index =
static_cast<size_t>(e);
282 static const char *
const names[6] = {
295 const size_t index =
static_cast<size_t>(e);
320 bool VerifyDistributedTensorConfigVector(::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset<void>> *values, const ::flatbuffers::Vector<DistributedTensorConfig> *types);
329 static FLATBUFFERS_CONSTEXPR_CPP11
const char *GetFullyQualifiedName() {
330 return "tt.target.ttnn.CoreCoord";
336 CoreCoord(uint64_t _x, uint64_t _y)
337 : x_(::flatbuffers::EndianScalar(_x)),
338 y_(::flatbuffers::EndianScalar(_y)) {
341 return ::flatbuffers::EndianScalar(x_);
344 return ::flatbuffers::EndianScalar(y_);
355 tt::target::ttnn::CoreCoord start_coord_;
356 tt::target::ttnn::CoreCoord end_coord_;
360 static FLATBUFFERS_CONSTEXPR_CPP11
const char *GetFullyQualifiedName() {
361 return "tt.target.ttnn.CoreRange";
367 CoreRange(
const tt::target::ttnn::CoreCoord &_start_coord,
const tt::target::ttnn::CoreCoord &_end_coord)
368 : start_coord_(_start_coord),
369 end_coord_(_end_coord) {
371 const tt::target::ttnn::CoreCoord &start_coord()
const {
374 const tt::target::ttnn::CoreCoord &end_coord()
const {
388 return "tt.target.ttnn.CoreRangeSet";
390 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
393 const ::flatbuffers::Vector<const tt::target::ttnn::CoreRange *> *
core_ranges()
const {
394 return GetPointer<const ::flatbuffers::Vector<const tt::target::ttnn::CoreRange *> *>(VT_CORE_RANGES);
396 bool Verify(::flatbuffers::Verifier &verifier)
const {
397 return VerifyTableStart(verifier) &&
398 VerifyOffset(verifier, VT_CORE_RANGES) &&
399 verifier.VerifyVector(core_ranges()) &&
406 ::flatbuffers::FlatBufferBuilder &
fbb_;
408 void add_core_ranges(::flatbuffers::Offset<::flatbuffers::Vector<const tt::target::ttnn::CoreRange *>> core_ranges) {
409 fbb_.AddOffset(CoreRangeSet::VT_CORE_RANGES, core_ranges);
415 ::flatbuffers::Offset<CoreRangeSet>
Finish() {
417 auto o = ::flatbuffers::Offset<CoreRangeSet>(end);
423 ::flatbuffers::FlatBufferBuilder &_fbb,
424 ::flatbuffers::Offset<::flatbuffers::Vector<const tt::target::ttnn::CoreRange *>> core_ranges = 0) {
436 ::flatbuffers::FlatBufferBuilder &_fbb,
437 const std::vector<tt::target::ttnn::CoreRange> *core_ranges =
nullptr) {
438 auto core_ranges__ = core_ranges ? _fbb.CreateVectorOfStructs<tt::target::ttnn::CoreRange>(*core_ranges) : 0;
448 return "tt.target.ttnn.ReplicateTensor";
450 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
451 VT_REPLICATION_FACTOR = 4
454 return GetField<uint32_t>(VT_REPLICATION_FACTOR, 0);
456 bool Verify(::flatbuffers::Verifier &verifier)
const {
457 return VerifyTableStart(verifier) &&
458 VerifyField<uint32_t>(verifier, VT_REPLICATION_FACTOR, 4) &&
465 ::flatbuffers::FlatBufferBuilder &
fbb_;
468 fbb_.AddElement<uint32_t>(ReplicateTensor::VT_REPLICATION_FACTOR, replication_factor, 0);
474 ::flatbuffers::Offset<ReplicateTensor>
Finish() {
476 auto o = ::flatbuffers::Offset<ReplicateTensor>(end);
482 ::flatbuffers::FlatBufferBuilder &_fbb,
483 uint32_t replication_factor = 0) {
498 return "tt.target.ttnn.ShardTensor";
500 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
504 return GetField<uint32_t>(VT_SHARD_DIM, 0);
506 bool Verify(::flatbuffers::Verifier &verifier)
const {
507 return VerifyTableStart(verifier) &&
508 VerifyField<uint32_t>(verifier, VT_SHARD_DIM, 4) &&
515 ::flatbuffers::FlatBufferBuilder &
fbb_;
518 fbb_.AddElement<uint32_t>(ShardTensor::VT_SHARD_DIM, shard_dim, 0);
524 ::flatbuffers::Offset<ShardTensor>
Finish() {
526 auto o = ::flatbuffers::Offset<ShardTensor>(end);
532 ::flatbuffers::FlatBufferBuilder &_fbb,
533 uint32_t shard_dim = 0) {
548 return "tt.target.ttnn.ShardTensor2D";
550 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
554 return GetStruct<const tt::target::Dim2d *>(VT_SHARD_MESH);
556 bool Verify(::flatbuffers::Verifier &verifier)
const {
557 return VerifyTableStart(verifier) &&
558 VerifyField<tt::target::Dim2d>(verifier, VT_SHARD_MESH, 4) &&
565 ::flatbuffers::FlatBufferBuilder &
fbb_;
568 fbb_.AddStruct(ShardTensor2D::VT_SHARD_MESH, shard_mesh);
574 ::flatbuffers::Offset<ShardTensor2D>
Finish() {
576 auto o = ::flatbuffers::Offset<ShardTensor2D>(end);
582 ::flatbuffers::FlatBufferBuilder &_fbb,
583 const tt::target::Dim2d *shard_mesh =
nullptr) {
598 return "tt.target.ttnn.AllGatherTensor";
600 bool Verify(::flatbuffers::Verifier &verifier)
const {
601 return VerifyTableStart(verifier) &&
608 ::flatbuffers::FlatBufferBuilder &
fbb_;
614 ::flatbuffers::Offset<AllGatherTensor>
Finish() {
616 auto o = ::flatbuffers::Offset<AllGatherTensor>(end);
622 ::flatbuffers::FlatBufferBuilder &_fbb) {
636 return "tt.target.ttnn.DistributionStrategy";
638 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
646 return GetPointer<const void *>(VT_STRATEGY);
661 bool Verify(::flatbuffers::Verifier &verifier)
const {
662 return VerifyTableStart(verifier) &&
664 VerifyOffset(verifier, VT_STRATEGY) &&
670 template<>
inline const tt::target::ttnn::ReplicateTensor *DistributionStrategy::strategy_as<tt::target::ttnn::ReplicateTensor>()
const {
671 return strategy_as_ReplicateTensor();
674 template<>
inline const tt::target::ttnn::ShardTensor *DistributionStrategy::strategy_as<tt::target::ttnn::ShardTensor>()
const {
675 return strategy_as_ShardTensor();
678 template<>
inline const tt::target::ttnn::ShardTensor2D *DistributionStrategy::strategy_as<tt::target::ttnn::ShardTensor2D>()
const {
679 return strategy_as_ShardTensor2D();
682 template<>
inline const tt::target::ttnn::AllGatherTensor *DistributionStrategy::strategy_as<tt::target::ttnn::AllGatherTensor>()
const {
683 return strategy_as_AllGatherTensor();
688 ::flatbuffers::FlatBufferBuilder &
fbb_;
694 fbb_.AddOffset(DistributionStrategy::VT_STRATEGY, strategy);
700 ::flatbuffers::Offset<DistributionStrategy>
Finish() {
702 auto o = ::flatbuffers::Offset<DistributionStrategy>(end);
708 ::flatbuffers::FlatBufferBuilder &_fbb,
710 ::flatbuffers::Offset<void> strategy = 0) {
718 using type = DistributionStrategy;
726 return "tt.target.ttnn.ShardSpec";
728 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
733 VT_PHYSICAL_SHARD_SHAPE = 12
738 const ::flatbuffers::Vector<int32_t> *
shape()
const {
739 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_SHAPE);
748 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(VT_PHYSICAL_SHARD_SHAPE);
750 bool Verify(::flatbuffers::Verifier &verifier)
const {
751 return VerifyTableStart(verifier) &&
753 verifier.VerifyTable(core_range_set()) &&
755 verifier.VerifyVector(shape()) &&
757 VerifyField<uint32_t>(verifier,
VT_MODE, 4) &&
758 VerifyOffset(verifier, VT_PHYSICAL_SHARD_SHAPE) &&
759 verifier.VerifyVector(physical_shard_shape()) &&
766 ::flatbuffers::FlatBufferBuilder &
fbb_;
771 void add_shape(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> shape) {
781 fbb_.AddOffset(ShardSpec::VT_PHYSICAL_SHARD_SHAPE, physical_shard_shape);
787 ::flatbuffers::Offset<ShardSpec>
Finish() {
789 auto o = ::flatbuffers::Offset<ShardSpec>(end);
795 ::flatbuffers::FlatBufferBuilder &_fbb,
796 ::flatbuffers::Offset<tt::target::ttnn::CoreRangeSet> core_range_set = 0,
797 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> shape = 0,
800 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> physical_shard_shape = 0) {
816 ::flatbuffers::FlatBufferBuilder &_fbb,
817 ::flatbuffers::Offset<tt::target::ttnn::CoreRangeSet> core_range_set = 0,
818 const std::vector<int32_t> *shape =
nullptr,
821 const std::vector<int32_t> *physical_shard_shape =
nullptr) {
822 auto shape__ = shape ? _fbb.CreateVector<int32_t>(*shape) : 0;
823 auto physical_shard_shape__ = physical_shard_shape ? _fbb.CreateVector<int32_t>(*physical_shard_shape) : 0;
830 physical_shard_shape__);
837 return "tt.target.ttnn.MemoryConfig";
839 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
851 return GetPointer<const tt::target::ttnn::ShardSpec *>(
VT_SHARD_SPEC);
853 bool Verify(::flatbuffers::Verifier &verifier)
const {
854 return VerifyTableStart(verifier) &&
858 verifier.VerifyTable(shard_spec()) &&
865 ::flatbuffers::FlatBufferBuilder &
fbb_;
873 void add_shard_spec(::flatbuffers::Offset<tt::target::ttnn::ShardSpec> shard_spec) {
880 ::flatbuffers::Offset<MemoryConfig>
Finish() {
882 auto o = ::flatbuffers::Offset<MemoryConfig>(end);
888 ::flatbuffers::FlatBufferBuilder &_fbb,
891 ::flatbuffers::Offset<tt::target::ttnn::ShardSpec> shard_spec = 0) {
908 return "tt.target.ttnn.MemoryDesc";
910 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
927 return GetPointer<const tt::target::ttnn::MemoryConfig *>(
VT_MEMORY_CONFIG);
930 return GetField<uint64_t>(
VT_SIZE, 0);
932 bool Verify(::flatbuffers::Verifier &verifier)
const {
933 return VerifyTableStart(verifier) &&
935 VerifyField<tt::target::Dim2d>(verifier,
VT_TILE_SHAPE, 4) &&
938 verifier.VerifyTable(memory_config()) &&
939 VerifyField<uint64_t>(verifier,
VT_SIZE, 8) &&
946 ::flatbuffers::FlatBufferBuilder &
fbb_;
967 ::flatbuffers::Offset<MemoryDesc>
Finish() {
969 auto o = ::flatbuffers::Offset<MemoryDesc>(end);
975 ::flatbuffers::FlatBufferBuilder &_fbb,
977 const tt::target::Dim2d *tile_shape =
nullptr,
979 ::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> memory_config = 0,
999 return "tt.target.ttnn.LayoutDesc";
1001 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1010 return GetPointer<const tt::target::ttnn::MemoryDesc *>(
VT_MEMORY_DESC);
1012 const tt::target::ttnn::DistributionStrategy *
strategy()
const {
1013 return GetPointer<const tt::target::ttnn::DistributionStrategy *>(VT_STRATEGY);
1015 bool Verify(::flatbuffers::Verifier &verifier)
const {
1016 return VerifyTableStart(verifier) &&
1017 VerifyField<uint16_t>(verifier,
VT_OOB_VAL, 2) &&
1019 verifier.VerifyTable(memory_desc()) &&
1020 VerifyOffset(verifier, VT_STRATEGY) &&
1021 verifier.VerifyTable(strategy()) &&
1022 verifier.EndTable();
1028 ::flatbuffers::FlatBufferBuilder &
fbb_;
1036 void add_strategy(::flatbuffers::Offset<tt::target::ttnn::DistributionStrategy> strategy) {
1037 fbb_.AddOffset(LayoutDesc::VT_STRATEGY, strategy);
1045 auto o = ::flatbuffers::Offset<LayoutDesc>(end);
1051 ::flatbuffers::FlatBufferBuilder &_fbb,
1053 ::flatbuffers::Offset<tt::target::ttnn::MemoryDesc> memory_desc = 0,
1054 ::flatbuffers::Offset<tt::target::ttnn::DistributionStrategy> strategy = 0) {
1059 return builder_.
Finish();
1071 return "tt.target.ttnn.TensorDesc";
1073 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1078 const ::flatbuffers::Vector<int32_t> *
shape()
const {
1079 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_SHAPE);
1082 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_MESH_SHAPE);
1084 const tt::target::ttnn::LayoutDesc *
layout()
const {
1085 return GetPointer<const tt::target::ttnn::LayoutDesc *>(
VT_LAYOUT);
1087 bool Verify(::flatbuffers::Verifier &verifier)
const {
1088 return VerifyTableStart(verifier) &&
1089 VerifyOffset(verifier,
VT_SHAPE) &&
1090 verifier.VerifyVector(shape()) &&
1092 verifier.VerifyVector(mesh_shape()) &&
1094 verifier.VerifyTable(layout()) &&
1095 verifier.EndTable();
1101 ::flatbuffers::FlatBufferBuilder &
fbb_;
1103 void add_shape(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> shape) {
1106 void add_mesh_shape(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> mesh_shape) {
1109 void add_layout(::flatbuffers::Offset<tt::target::ttnn::LayoutDesc> layout) {
1118 auto o = ::flatbuffers::Offset<TensorDesc>(end);
1124 ::flatbuffers::FlatBufferBuilder &_fbb,
1125 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> shape = 0,
1126 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> mesh_shape = 0,
1127 ::flatbuffers::Offset<tt::target::ttnn::LayoutDesc> layout = 0) {
1132 return builder_.
Finish();
1141 ::flatbuffers::FlatBufferBuilder &_fbb,
1142 const std::vector<int32_t> *shape =
nullptr,
1143 const std::vector<int32_t> *mesh_shape =
nullptr,
1144 ::flatbuffers::Offset<tt::target::ttnn::LayoutDesc> layout = 0) {
1145 auto shape__ = shape ? _fbb.CreateVector<int32_t>(*shape) : 0;
1146 auto mesh_shape__ = mesh_shape ? _fbb.CreateVector<int32_t>(*mesh_shape) : 0;
1158 return "tt.target.ttnn.TensorRef";
1160 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1169 return GetField<uint64_t>(
VT_SIZE, 0);
1171 const tt::target::ttnn::TensorDesc *
desc()
const {
1172 return GetPointer<const tt::target::ttnn::TensorDesc *>(VT_DESC);
1174 bool Verify(::flatbuffers::Verifier &verifier)
const {
1175 return VerifyTableStart(verifier) &&
1177 VerifyField<uint64_t>(verifier,
VT_SIZE, 8) &&
1178 VerifyOffset(verifier, VT_DESC) &&
1179 verifier.VerifyTable(desc()) &&
1180 verifier.EndTable();
1186 ::flatbuffers::FlatBufferBuilder &
fbb_;
1194 void add_desc(::flatbuffers::Offset<tt::target::ttnn::TensorDesc> desc) {
1195 fbb_.AddOffset(TensorRef::VT_DESC, desc);
1203 auto o = ::flatbuffers::Offset<TensorRef>(end);
1209 ::flatbuffers::FlatBufferBuilder &_fbb,
1210 uint32_t global_id = 0,
1212 ::flatbuffers::Offset<tt::target::ttnn::TensorDesc> desc = 0) {
1217 return builder_.
Finish();
1231 auto ptr =
reinterpret_cast<const tt::target::ttnn::ReplicateTensor *
>(obj);
1232 return verifier.VerifyTable(ptr);
1235 auto ptr =
reinterpret_cast<const tt::target::ttnn::ShardTensor *
>(obj);
1236 return verifier.VerifyTable(ptr);
1239 auto ptr =
reinterpret_cast<const tt::target::ttnn::ShardTensor2D *
>(obj);
1240 return verifier.VerifyTable(ptr);
1243 auto ptr =
reinterpret_cast<const tt::target::ttnn::AllGatherTensor *
>(obj);
1244 return verifier.VerifyTable(ptr);
1246 default:
return true;
1250 inline bool VerifyDistributedTensorConfigVector(::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset<void>> *values, const ::flatbuffers::Vector<DistributedTensorConfig> *types) {
1251 if (!values || !types)
return !values && !types;
1252 if (values->size() != types->size())
return false;
1253 for (::flatbuffers::uoffset_t i = 0; i < values->size(); ++i) {
VT_STRATEGY_TYPE
Definition: types_generated.h:639
VT_MESH_SHAPE
Definition: types_generated.h:1075
VT_MODE
Definition: types_generated.h:732
VT_STORAGE_TYPE
Definition: types_generated.h:911
VT_TENSOR_MEMORY_LAYOUT
Definition: types_generated.h:840
VT_BUFFER_TYPE
Definition: types_generated.h:841
VT_OOB_VAL
Definition: types_generated.h:1002
VT_ORIENTATION
Definition: types_generated.h:731
VT_MEMORY_DESC
Definition: types_generated.h:1003
VT_MEMORY_CONFIG
Definition: types_generated.h:914
VT_LAYOUT
Definition: creation_generated.h:317
const MeshShardType(& EnumValuesMeshShardType())[4]
Definition: types_generated.h:173
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:1123
inline ::flatbuffers::Offset< ReplicateTensor > CreateReplicateTensor(::flatbuffers::FlatBufferBuilder &_fbb, uint32_t replication_factor=0)
Definition: types_generated.h:481
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:794
MeshShardType
Definition: types_generated.h:164
ShardMode
Definition: types_generated.h:230
const char * EnumNameShardMode(ShardMode e)
Definition: types_generated.h:254
const ShardMode(& EnumValuesShardMode())[2]
Definition: types_generated.h:237
const char * EnumNameMeshShardType(MeshShardType e)
Definition: types_generated.h:194
const char * EnumNameDistributedTensorConfig(DistributedTensorConfig e)
Definition: types_generated.h:293
inline ::flatbuffers::Offset< ShardTensor > CreateShardTensor(::flatbuffers::FlatBufferBuilder &_fbb, uint32_t shard_dim=0)
Definition: types_generated.h:531
ShardOrientation
Definition: types_generated.h:200
const ShardOrientation(& EnumValuesShardOrientation())[2]
Definition: types_generated.h:207
const char *const * EnumNamesDistributedTensorConfig()
Definition: types_generated.h:281
StorageType
Definition: types_generated.h:101
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:1208
const DistributedTensorConfig(& EnumValuesDistributedTensorConfig())[5]
Definition: types_generated.h:270
const char *const * EnumNamesShardOrientation()
Definition: types_generated.h:215
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:815
FLATBUFFERS_MANUALLY_ALIGNED_STRUCT(8) CoreCoord FLATBUFFERS_FINAL_CLASS
Definition: types_generated.h:322
inline ::flatbuffers::Offset< AllGatherTensor > CreateAllGatherTensor(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:621
const char * EnumNameTensorMemoryLayout(TensorMemoryLayout e)
Definition: types_generated.h:95
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:974
const char *const * EnumNamesStorageType()
Definition: types_generated.h:118
const TensorMemoryLayout(& EnumValuesTensorMemoryLayout())[5]
Definition: types_generated.h:72
DistributedTensorConfig
Definition: types_generated.h:260
const char *const * EnumNamesShardMode()
Definition: types_generated.h:245
inline ::flatbuffers::Offset< CoreRangeSet > CreateCoreRangeSet(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset<::flatbuffers::Vector< const tt::target::ttnn::CoreRange * >> core_ranges=0)
Definition: types_generated.h:422
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:581
const char * EnumNameStorageType(StorageType e)
Definition: types_generated.h:128
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:707
const char * EnumNameMeshShardDirection(MeshShardDirection e)
Definition: types_generated.h:158
const char *const * EnumNamesTensorMemoryLayout()
Definition: types_generated.h:83
inline ::flatbuffers::Offset< CoreRangeSet > CreateCoreRangeSetDirect(::flatbuffers::FlatBufferBuilder &_fbb, const std::vector< tt::target::ttnn::CoreRange > *core_ranges=nullptr)
Definition: types_generated.h:435
const MeshShardDirection(& EnumValuesMeshShardDirection())[2]
Definition: types_generated.h:141
bool VerifyDistributedTensorConfig(::flatbuffers::Verifier &verifier, const void *obj, DistributedTensorConfig type)
Definition: types_generated.h:1225
bool VerifyDistributedTensorConfigVector(::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset< void >> *values, const ::flatbuffers::Vector< DistributedTensorConfig > *types)
Definition: types_generated.h:1250
const char *const * EnumNamesMeshShardType()
Definition: types_generated.h:183
const char * EnumNameShardOrientation(ShardOrientation e)
Definition: types_generated.h:224
const StorageType(& EnumValuesStorageType())[3]
Definition: types_generated.h:109
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:887
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:1050
MeshShardDirection
Definition: types_generated.h:134
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:1140
const char *const * EnumNamesMeshShardDirection()
Definition: types_generated.h:149
OOBVal
Definition: types_generated.h:144
BufferType
Definition: types_generated.h:283
DataType
Definition: types_generated.h:81
Definition: debug_info_generated.h:18
Definition: debug_info_generated.h:36
Definition: types_generated.h:606
AllGatherTensorBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:610
::flatbuffers::Offset< AllGatherTensor > Finish()
Definition: types_generated.h:614
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:608
AllGatherTensor Table
Definition: types_generated.h:607
::flatbuffers::uoffset_t start_
Definition: types_generated.h:609
Definition: types_generated.h:627
AllGatherTensor type
Definition: types_generated.h:628
static constexpr auto Create
Definition: types_generated.h:629
Definition: types_generated.h:349
CoreCoord type
Definition: types_generated.h:350
Definition: types_generated.h:404
::flatbuffers::uoffset_t start_
Definition: types_generated.h:407
::flatbuffers::Offset< CoreRangeSet > Finish()
Definition: types_generated.h:415
CoreRangeSetBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:411
void add_core_ranges(::flatbuffers::Offset<::flatbuffers::Vector< const tt::target::ttnn::CoreRange * >> core_ranges)
Definition: types_generated.h:408
CoreRangeSet Table
Definition: types_generated.h:405
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:406
Definition: types_generated.h:430
CoreRangeSet type
Definition: types_generated.h:431
static constexpr auto Create
Definition: types_generated.h:432
Definition: types_generated.h:380
CoreRange type
Definition: types_generated.h:381
Definition: types_generated.h:299
static const DistributedTensorConfig enum_value
Definition: types_generated.h:300
Definition: types_generated.h:686
void add_strategy_type(tt::target::ttnn::DistributedTensorConfig strategy_type)
Definition: types_generated.h:690
DistributionStrategyBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:696
::flatbuffers::uoffset_t start_
Definition: types_generated.h:689
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:688
DistributionStrategy Table
Definition: types_generated.h:687
void add_strategy(::flatbuffers::Offset< void > strategy)
Definition: types_generated.h:693
::flatbuffers::Offset< DistributionStrategy > Finish()
Definition: types_generated.h:700
Definition: types_generated.h:717
static constexpr auto Create
Definition: types_generated.h:719
DistributionStrategy type
Definition: types_generated.h:718
Definition: binary_generated.h:31
TensorDescBuilder Builder
Definition: types_generated.h:1068
const tt::target::Dim2d * tile_shape() const
Definition: types_generated.h:920
tt::target::ttnn::TensorMemoryLayout tensor_memory_layout() const
Definition: types_generated.h:844
ShardSpecBuilder Builder
Definition: types_generated.h:723
tt::target::BufferType buffer_type() const
Definition: types_generated.h:847
ReplicateTensorBuilder Builder
Definition: types_generated.h:445
const tt::target::ttnn::ShardTensor2D * strategy_as_ShardTensor2D() const
Definition: types_generated.h:655
const tt::target::ttnn::MemoryConfig * memory_config() const
Definition: types_generated.h:926
tt::target::ttnn::ShardMode mode() const
Definition: types_generated.h:744
const void * strategy() const
Definition: types_generated.h:645
const tt::target::ttnn::CoreRangeSet * core_range_set() const
Definition: types_generated.h:735
const ::flatbuffers::Vector< int32_t > * mesh_shape() const
Definition: types_generated.h:1081
const ::flatbuffers::Vector< int32_t > * shape() const
Definition: types_generated.h:738
ShardTensorBuilder Builder
Definition: types_generated.h:495
uint32_t replication_factor() const
Definition: types_generated.h:453
tt::target::OOBVal oob_val() const
Definition: types_generated.h:1006
const tt::target::ttnn::AllGatherTensor * strategy_as_AllGatherTensor() const
Definition: types_generated.h:658
const tt::target::ttnn::DistributionStrategy * strategy() const
Definition: types_generated.h:1012
uint64_t size() const
Definition: types_generated.h:929
tt::target::DataType data_type() const
Definition: types_generated.h:923
TensorRefBuilder Builder
Definition: types_generated.h:1155
LayoutDescBuilder Builder
Definition: types_generated.h:996
DistributionStrategyBuilder Builder
Definition: types_generated.h:633
bool Verify(::flatbuffers::Verifier &verifier) const
Definition: types_generated.h:396
ShardTensor2DBuilder Builder
Definition: types_generated.h:545
MemoryConfigBuilder Builder
Definition: types_generated.h:834
MemoryDescBuilder Builder
Definition: types_generated.h:905
const tt::target::ttnn::MemoryDesc * memory_desc() const
Definition: types_generated.h:1009
uint32_t shard_dim() const
Definition: types_generated.h:503
const T * strategy_as() const
const tt::target::Dim2d * shard_mesh() const
Definition: types_generated.h:553
const ::flatbuffers::Vector< const tt::target::ttnn::CoreRange * > * core_ranges() const
Definition: types_generated.h:393
CoreRangeSetBuilder Builder
Definition: types_generated.h:385
const tt::target::ttnn::ReplicateTensor * strategy_as_ReplicateTensor() const
Definition: types_generated.h:649
const tt::target::ttnn::ShardTensor * strategy_as_ShardTensor() const
Definition: types_generated.h:652
const ::flatbuffers::Vector< int32_t > * physical_shard_shape() const
Definition: types_generated.h:747
static FLATBUFFERS_CONSTEXPR_CPP11 const char * GetFullyQualifiedName()
Definition: types_generated.h:387
tt::target::ttnn::ShardOrientation orientation() const
Definition: types_generated.h:741
AllGatherTensorBuilder Builder
Definition: types_generated.h:595
const tt::target::ttnn::TensorDesc * desc() const
Definition: types_generated.h:1171
tt::target::ttnn::DistributedTensorConfig strategy_type() const
Definition: types_generated.h:642
const tt::target::ttnn::LayoutDesc * layout() const
Definition: types_generated.h:1084
tt::target::ttnn::StorageType storage_type() const
Definition: types_generated.h:917
const tt::target::ttnn::ShardSpec * shard_spec() const
Definition: types_generated.h:850
uint32_t global_id() const
Definition: types_generated.h:1165
Definition: types_generated.h:1026
void add_memory_desc(::flatbuffers::Offset< tt::target::ttnn::MemoryDesc > memory_desc)
Definition: types_generated.h:1033
::flatbuffers::uoffset_t start_
Definition: types_generated.h:1029
LayoutDesc Table
Definition: types_generated.h:1027
void add_oob_val(tt::target::OOBVal oob_val)
Definition: types_generated.h:1030
LayoutDescBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:1039
::flatbuffers::Offset< LayoutDesc > Finish()
Definition: types_generated.h:1043
void add_strategy(::flatbuffers::Offset< tt::target::ttnn::DistributionStrategy > strategy)
Definition: types_generated.h:1036
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:1028
Definition: types_generated.h:1062
LayoutDesc type
Definition: types_generated.h:1063
static constexpr auto Create
Definition: types_generated.h:1064
Definition: types_generated.h:863
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:865
MemoryConfigBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:876
MemoryConfig Table
Definition: types_generated.h:864
void add_tensor_memory_layout(tt::target::ttnn::TensorMemoryLayout tensor_memory_layout)
Definition: types_generated.h:867
::flatbuffers::uoffset_t start_
Definition: types_generated.h:866
void add_shard_spec(::flatbuffers::Offset< tt::target::ttnn::ShardSpec > shard_spec)
Definition: types_generated.h:873
::flatbuffers::Offset< MemoryConfig > Finish()
Definition: types_generated.h:880
void add_buffer_type(tt::target::BufferType buffer_type)
Definition: types_generated.h:870
Definition: types_generated.h:899
static constexpr auto Create
Definition: types_generated.h:901
MemoryConfig type
Definition: types_generated.h:900
Definition: types_generated.h:944
MemoryDesc Table
Definition: types_generated.h:945
void add_memory_config(::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memory_config)
Definition: types_generated.h:957
MemoryDescBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:963
void add_size(uint64_t size)
Definition: types_generated.h:960
void add_storage_type(tt::target::ttnn::StorageType storage_type)
Definition: types_generated.h:948
void add_tile_shape(const tt::target::Dim2d *tile_shape)
Definition: types_generated.h:951
::flatbuffers::uoffset_t start_
Definition: types_generated.h:947
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:946
void add_data_type(tt::target::DataType data_type)
Definition: types_generated.h:954
::flatbuffers::Offset< MemoryDesc > Finish()
Definition: types_generated.h:967
Definition: types_generated.h:990
static constexpr auto Create
Definition: types_generated.h:992
MemoryDesc type
Definition: types_generated.h:991
Definition: types_generated.h:463
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:465
ReplicateTensorBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:470
ReplicateTensor Table
Definition: types_generated.h:464
void add_replication_factor(uint32_t replication_factor)
Definition: types_generated.h:467
::flatbuffers::Offset< ReplicateTensor > Finish()
Definition: types_generated.h:474
::flatbuffers::uoffset_t start_
Definition: types_generated.h:466
Definition: types_generated.h:489
static constexpr auto Create
Definition: types_generated.h:491
ReplicateTensor type
Definition: types_generated.h:490
Definition: types_generated.h:764
ShardSpecBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:783
ShardSpec Table
Definition: types_generated.h:765
void add_physical_shard_shape(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> physical_shard_shape)
Definition: types_generated.h:780
void add_orientation(tt::target::ttnn::ShardOrientation orientation)
Definition: types_generated.h:774
::flatbuffers::uoffset_t start_
Definition: types_generated.h:767
::flatbuffers::Offset< ShardSpec > Finish()
Definition: types_generated.h:787
void add_mode(tt::target::ttnn::ShardMode mode)
Definition: types_generated.h:777
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:766
void add_shape(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> shape)
Definition: types_generated.h:771
void add_core_range_set(::flatbuffers::Offset< tt::target::ttnn::CoreRangeSet > core_range_set)
Definition: types_generated.h:768
Definition: types_generated.h:810
ShardSpec type
Definition: types_generated.h:811
static constexpr auto Create
Definition: types_generated.h:812
Definition: types_generated.h:563
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:565
ShardTensor2DBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:570
ShardTensor2D Table
Definition: types_generated.h:564
::flatbuffers::uoffset_t start_
Definition: types_generated.h:566
::flatbuffers::Offset< ShardTensor2D > Finish()
Definition: types_generated.h:574
void add_shard_mesh(const tt::target::Dim2d *shard_mesh)
Definition: types_generated.h:567
Definition: types_generated.h:589
static constexpr auto Create
Definition: types_generated.h:591
ShardTensor2D type
Definition: types_generated.h:590
Definition: types_generated.h:513
ShardTensor Table
Definition: types_generated.h:514
::flatbuffers::Offset< ShardTensor > Finish()
Definition: types_generated.h:524
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:515
ShardTensorBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:520
void add_shard_dim(uint32_t shard_dim)
Definition: types_generated.h:517
::flatbuffers::uoffset_t start_
Definition: types_generated.h:516
Definition: types_generated.h:539
static constexpr auto Create
Definition: types_generated.h:541
ShardTensor type
Definition: types_generated.h:540
Definition: types_generated.h:1099
::flatbuffers::uoffset_t start_
Definition: types_generated.h:1102
void add_layout(::flatbuffers::Offset< tt::target::ttnn::LayoutDesc > layout)
Definition: types_generated.h:1109
TensorDescBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:1112
TensorDesc Table
Definition: types_generated.h:1100
void add_mesh_shape(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> mesh_shape)
Definition: types_generated.h:1106
::flatbuffers::Offset< TensorDesc > Finish()
Definition: types_generated.h:1116
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:1101
void add_shape(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> shape)
Definition: types_generated.h:1103
Definition: types_generated.h:1135
static constexpr auto Create
Definition: types_generated.h:1137
TensorDesc type
Definition: types_generated.h:1136
Definition: types_generated.h:1184
::flatbuffers::FlatBufferBuilder & fbb_
Definition: types_generated.h:1186
void add_global_id(uint32_t global_id)
Definition: types_generated.h:1188
::flatbuffers::uoffset_t start_
Definition: types_generated.h:1187
void add_size(uint64_t size)
Definition: types_generated.h:1191
void add_desc(::flatbuffers::Offset< tt::target::ttnn::TensorDesc > desc)
Definition: types_generated.h:1194
::flatbuffers::Offset< TensorRef > Finish()
Definition: types_generated.h:1201
TensorRef Table
Definition: types_generated.h:1185
TensorRefBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: types_generated.h:1197
Definition: types_generated.h:1220
static constexpr auto Create
Definition: types_generated.h:1222
TensorRef type
Definition: types_generated.h:1221