TT-MLIR
types_generated.h
Go to the documentation of this file.
1 // automatically generated by the FlatBuffers compiler, do not modify
2 
3 
4 #ifndef FLATBUFFERS_GENERATED_TYPES_TT_TARGET_TTNN_H_
5 #define FLATBUFFERS_GENERATED_TYPES_TT_TARGET_TTNN_H_
6 
7 #include "flatbuffers/flatbuffers.h"
8 
9 // Ensure the included flatbuffers.h is the same version as when this file was
10 // generated, otherwise it may not be compatible.
11 static_assert(FLATBUFFERS_VERSION_MAJOR == 24 &&
12  FLATBUFFERS_VERSION_MINOR == 3 &&
13  FLATBUFFERS_VERSION_REVISION == 25,
14  "Non-compatible flatbuffers version included");
15 
17 
18 namespace tt {
19 namespace target {
20 namespace ttnn {
21 
22 struct CoreCoord;
23 
24 struct CoreRange;
25 
26 struct CoreRangeSet;
27 struct CoreRangeSetBuilder;
28 
29 struct ReplicateTensor;
30 struct ReplicateTensorBuilder;
31 
32 struct ShardTensor;
33 struct ShardTensorBuilder;
34 
35 struct ShardTensor2D;
36 struct ShardTensor2DBuilder;
37 
38 struct AllGatherTensor;
39 struct AllGatherTensorBuilder;
40 
41 struct DistributionStrategy;
42 struct DistributionStrategyBuilder;
43 
44 struct ShardSpec;
45 struct ShardSpecBuilder;
46 
47 struct MemoryConfig;
48 struct MemoryConfigBuilder;
49 
50 struct MemoryDesc;
51 struct MemoryDescBuilder;
52 
53 struct LayoutDesc;
54 struct LayoutDescBuilder;
55 
56 struct TensorDesc;
57 struct TensorDescBuilder;
58 
59 struct TensorRef;
60 struct TensorRefBuilder;
61 
62 enum class TensorMemoryLayout : uint16_t {
63  Interleaved = 0,
64  HeightSharded = 1,
65  WidthSharded = 2,
66  BlockSharded = 3,
67  MIN = Interleaved,
69 };
70 
72  static const TensorMemoryLayout values[] = {
77  };
78  return values;
79 }
80 
81 inline const char * const *EnumNamesTensorMemoryLayout() {
82  static const char * const names[5] = {
83  "Interleaved",
84  "HeightSharded",
85  "WidthSharded",
86  "BlockSharded",
87  nullptr
88  };
89  return names;
90 }
91 
93  if (::flatbuffers::IsOutRange(e, TensorMemoryLayout::Interleaved, TensorMemoryLayout::BlockSharded)) return "";
94  const size_t index = static_cast<size_t>(e);
95  return EnumNamesTensorMemoryLayout()[index];
96 }
97 
98 enum class StorageType : uint16_t {
99  Host = 0,
100  Device = 1,
101  MultiDeviceHost = 2,
102  MIN = Host,
104 };
105 
106 inline const StorageType (&EnumValuesStorageType())[3] {
107  static const StorageType values[] = {
111  };
112  return values;
113 }
114 
115 inline const char * const *EnumNamesStorageType() {
116  static const char * const names[4] = {
117  "Host",
118  "Device",
119  "MultiDeviceHost",
120  nullptr
121  };
122  return names;
123 }
124 
125 inline const char *EnumNameStorageType(StorageType e) {
126  if (::flatbuffers::IsOutRange(e, StorageType::Host, StorageType::MultiDeviceHost)) return "";
127  const size_t index = static_cast<size_t>(e);
128  return EnumNamesStorageType()[index];
129 }
130 
131 enum class MeshShardDirection : uint32_t {
132  FullToShardShape = 0,
133  ShardToFullShape = 1,
136 };
137 
139  static const MeshShardDirection values[] = {
142  };
143  return values;
144 }
145 
146 inline const char * const *EnumNamesMeshShardDirection() {
147  static const char * const names[3] = {
148  "FullToShardShape",
149  "ShardToFullShape",
150  nullptr
151  };
152  return names;
153 }
154 
156  if (::flatbuffers::IsOutRange(e, MeshShardDirection::FullToShardShape, MeshShardDirection::ShardToFullShape)) return "";
157  const size_t index = static_cast<size_t>(e);
158  return EnumNamesMeshShardDirection()[index];
159 }
160 
161 enum class MeshShardType : uint32_t {
162  Identity = 0,
163  Replicate = 1,
164  Maximal = 2,
165  Devices = 3,
166  MIN = Identity,
167  MAX = Devices
168 };
169 
171  static const MeshShardType values[] = {
176  };
177  return values;
178 }
179 
180 inline const char * const *EnumNamesMeshShardType() {
181  static const char * const names[5] = {
182  "Identity",
183  "Replicate",
184  "Maximal",
185  "Devices",
186  nullptr
187  };
188  return names;
189 }
190 
191 inline const char *EnumNameMeshShardType(MeshShardType e) {
192  if (::flatbuffers::IsOutRange(e, MeshShardType::Identity, MeshShardType::Devices)) return "";
193  const size_t index = static_cast<size_t>(e);
194  return EnumNamesMeshShardType()[index];
195 }
196 
197 enum class ShardOrientation : uint32_t {
198  RowMajor = 0,
199  ColMajor = 1,
200  MIN = RowMajor,
201  MAX = ColMajor
202 };
203 
205  static const ShardOrientation values[] = {
208  };
209  return values;
210 }
211 
212 inline const char * const *EnumNamesShardOrientation() {
213  static const char * const names[3] = {
214  "RowMajor",
215  "ColMajor",
216  nullptr
217  };
218  return names;
219 }
220 
222  if (::flatbuffers::IsOutRange(e, ShardOrientation::RowMajor, ShardOrientation::ColMajor)) return "";
223  const size_t index = static_cast<size_t>(e);
224  return EnumNamesShardOrientation()[index];
225 }
226 
227 enum class ShardMode : uint32_t {
228  Physical = 0,
229  Logical = 1,
230  MIN = Physical,
231  MAX = Logical
232 };
233 
234 inline const ShardMode (&EnumValuesShardMode())[2] {
235  static const ShardMode values[] = {
238  };
239  return values;
240 }
241 
242 inline const char * const *EnumNamesShardMode() {
243  static const char * const names[3] = {
244  "Physical",
245  "Logical",
246  nullptr
247  };
248  return names;
249 }
250 
251 inline const char *EnumNameShardMode(ShardMode e) {
252  if (::flatbuffers::IsOutRange(e, ShardMode::Physical, ShardMode::Logical)) return "";
253  const size_t index = static_cast<size_t>(e);
254  return EnumNamesShardMode()[index];
255 }
256 
257 enum class DistributedTensorConfig : uint8_t {
258  NONE = 0,
259  ReplicateTensor = 1,
260  ShardTensor = 2,
261  ShardTensor2D = 3,
262  AllGatherTensor = 4,
263  MIN = NONE,
265 };
266 
268  static const DistributedTensorConfig values[] = {
274  };
275  return values;
276 }
277 
278 inline const char * const *EnumNamesDistributedTensorConfig() {
279  static const char * const names[6] = {
280  "NONE",
281  "ReplicateTensor",
282  "ShardTensor",
283  "ShardTensor2D",
284  "AllGatherTensor",
285  nullptr
286  };
287  return names;
288 }
289 
291  if (::flatbuffers::IsOutRange(e, DistributedTensorConfig::NONE, DistributedTensorConfig::AllGatherTensor)) return "";
292  const size_t index = static_cast<size_t>(e);
293  return EnumNamesDistributedTensorConfig()[index];
294 }
295 
296 template<typename T> struct DistributedTensorConfigTraits {
298 };
299 
300 template<> struct DistributedTensorConfigTraits<tt::target::ttnn::ReplicateTensor> {
302 };
303 
304 template<> struct DistributedTensorConfigTraits<tt::target::ttnn::ShardTensor> {
306 };
307 
308 template<> struct DistributedTensorConfigTraits<tt::target::ttnn::ShardTensor2D> {
310 };
311 
312 template<> struct DistributedTensorConfigTraits<tt::target::ttnn::AllGatherTensor> {
314 };
315 
316 bool VerifyDistributedTensorConfig(::flatbuffers::Verifier &verifier, const void *obj, DistributedTensorConfig type);
317 bool VerifyDistributedTensorConfigVector(::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset<void>> *values, const ::flatbuffers::Vector<DistributedTensorConfig> *types);
318 
320  private:
321  uint64_t x_;
322  uint64_t y_;
323 
324  public:
325  struct Traits;
326  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
327  return "tt.target.ttnn.CoreCoord";
328  }
329  CoreCoord()
330  : x_(0),
331  y_(0) {
332  }
333  CoreCoord(uint64_t _x, uint64_t _y)
334  : x_(::flatbuffers::EndianScalar(_x)),
335  y_(::flatbuffers::EndianScalar(_y)) {
336  }
337  uint64_t x() const {
338  return ::flatbuffers::EndianScalar(x_);
339  }
340  uint64_t y() const {
341  return ::flatbuffers::EndianScalar(y_);
342  }
343 };
344 FLATBUFFERS_STRUCT_END(CoreCoord, 16);
345 
347  using type = CoreCoord;
348 };
349 
351  private:
352  tt::target::ttnn::CoreCoord start_coord_;
353  tt::target::ttnn::CoreCoord end_coord_;
354 
355  public:
356  struct Traits;
357  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
358  return "tt.target.ttnn.CoreRange";
359  }
360  CoreRange()
361  : start_coord_(),
362  end_coord_() {
363  }
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) {
367  }
368  const tt::target::ttnn::CoreCoord &start_coord() const {
369  return start_coord_;
370  }
371  const tt::target::ttnn::CoreCoord &end_coord() const {
372  return end_coord_;
373  }
374 };
375 FLATBUFFERS_STRUCT_END(CoreRange, 32);
376 
378  using type = CoreRange;
379 };
380 
381 struct CoreRangeSet FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
383  struct Traits;
384  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
385  return "tt.target.ttnn.CoreRangeSet";
386  }
387  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
388  VT_CORE_RANGES = 4
389  };
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);
392  }
393  bool Verify(::flatbuffers::Verifier &verifier) const {
394  return VerifyTableStart(verifier) &&
395  VerifyOffset(verifier, VT_CORE_RANGES) &&
396  verifier.VerifyVector(core_ranges()) &&
397  verifier.EndTable();
398  }
399 };
400 
402  typedef CoreRangeSet Table;
403  ::flatbuffers::FlatBufferBuilder &fbb_;
404  ::flatbuffers::uoffset_t start_;
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);
407  }
408  explicit CoreRangeSetBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
409  : fbb_(_fbb) {
410  start_ = fbb_.StartTable();
411  }
412  ::flatbuffers::Offset<CoreRangeSet> Finish() {
413  const auto end = fbb_.EndTable(start_);
414  auto o = ::flatbuffers::Offset<CoreRangeSet>(end);
415  return o;
416  }
417 };
418 
419 inline ::flatbuffers::Offset<CoreRangeSet> CreateCoreRangeSet(
420  ::flatbuffers::FlatBufferBuilder &_fbb,
421  ::flatbuffers::Offset<::flatbuffers::Vector<const tt::target::ttnn::CoreRange *>> core_ranges = 0) {
422  CoreRangeSetBuilder builder_(_fbb);
423  builder_.add_core_ranges(core_ranges);
424  return builder_.Finish();
425 }
426 
428  using type = CoreRangeSet;
429  static auto constexpr Create = CreateCoreRangeSet;
430 };
431 
432 inline ::flatbuffers::Offset<CoreRangeSet> CreateCoreRangeSetDirect(
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;
437  _fbb,
438  core_ranges__);
439 }
440 
441 struct ReplicateTensor FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
443  struct Traits;
444  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
445  return "tt.target.ttnn.ReplicateTensor";
446  }
447  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
448  VT_REPLICATION_FACTOR = 4
449  };
450  uint32_t replication_factor() const {
451  return GetField<uint32_t>(VT_REPLICATION_FACTOR, 0);
452  }
453  bool Verify(::flatbuffers::Verifier &verifier) const {
454  return VerifyTableStart(verifier) &&
455  VerifyField<uint32_t>(verifier, VT_REPLICATION_FACTOR, 4) &&
456  verifier.EndTable();
457  }
458 };
459 
462  ::flatbuffers::FlatBufferBuilder &fbb_;
463  ::flatbuffers::uoffset_t start_;
464  void add_replication_factor(uint32_t replication_factor) {
465  fbb_.AddElement<uint32_t>(ReplicateTensor::VT_REPLICATION_FACTOR, replication_factor, 0);
466  }
467  explicit ReplicateTensorBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
468  : fbb_(_fbb) {
469  start_ = fbb_.StartTable();
470  }
471  ::flatbuffers::Offset<ReplicateTensor> Finish() {
472  const auto end = fbb_.EndTable(start_);
473  auto o = ::flatbuffers::Offset<ReplicateTensor>(end);
474  return o;
475  }
476 };
477 
478 inline ::flatbuffers::Offset<ReplicateTensor> CreateReplicateTensor(
479  ::flatbuffers::FlatBufferBuilder &_fbb,
480  uint32_t replication_factor = 0) {
481  ReplicateTensorBuilder builder_(_fbb);
482  builder_.add_replication_factor(replication_factor);
483  return builder_.Finish();
484 }
485 
488  static auto constexpr Create = CreateReplicateTensor;
489 };
490 
491 struct ShardTensor FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
493  struct Traits;
494  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
495  return "tt.target.ttnn.ShardTensor";
496  }
497  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
498  VT_SHARD_DIM = 4
499  };
500  uint32_t shard_dim() const {
501  return GetField<uint32_t>(VT_SHARD_DIM, 0);
502  }
503  bool Verify(::flatbuffers::Verifier &verifier) const {
504  return VerifyTableStart(verifier) &&
505  VerifyField<uint32_t>(verifier, VT_SHARD_DIM, 4) &&
506  verifier.EndTable();
507  }
508 };
509 
512  ::flatbuffers::FlatBufferBuilder &fbb_;
513  ::flatbuffers::uoffset_t start_;
514  void add_shard_dim(uint32_t shard_dim) {
515  fbb_.AddElement<uint32_t>(ShardTensor::VT_SHARD_DIM, shard_dim, 0);
516  }
517  explicit ShardTensorBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
518  : fbb_(_fbb) {
519  start_ = fbb_.StartTable();
520  }
521  ::flatbuffers::Offset<ShardTensor> Finish() {
522  const auto end = fbb_.EndTable(start_);
523  auto o = ::flatbuffers::Offset<ShardTensor>(end);
524  return o;
525  }
526 };
527 
528 inline ::flatbuffers::Offset<ShardTensor> CreateShardTensor(
529  ::flatbuffers::FlatBufferBuilder &_fbb,
530  uint32_t shard_dim = 0) {
531  ShardTensorBuilder builder_(_fbb);
532  builder_.add_shard_dim(shard_dim);
533  return builder_.Finish();
534 }
535 
537  using type = ShardTensor;
538  static auto constexpr Create = CreateShardTensor;
539 };
540 
541 struct ShardTensor2D FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
543  struct Traits;
544  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
545  return "tt.target.ttnn.ShardTensor2D";
546  }
547  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
548  VT_SHARD_MESH = 4
549  };
550  const tt::target::Dim2d *shard_mesh() const {
551  return GetStruct<const tt::target::Dim2d *>(VT_SHARD_MESH);
552  }
553  bool Verify(::flatbuffers::Verifier &verifier) const {
554  return VerifyTableStart(verifier) &&
555  VerifyField<tt::target::Dim2d>(verifier, VT_SHARD_MESH, 4) &&
556  verifier.EndTable();
557  }
558 };
559 
562  ::flatbuffers::FlatBufferBuilder &fbb_;
563  ::flatbuffers::uoffset_t start_;
564  void add_shard_mesh(const tt::target::Dim2d *shard_mesh) {
565  fbb_.AddStruct(ShardTensor2D::VT_SHARD_MESH, shard_mesh);
566  }
567  explicit ShardTensor2DBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
568  : fbb_(_fbb) {
569  start_ = fbb_.StartTable();
570  }
571  ::flatbuffers::Offset<ShardTensor2D> Finish() {
572  const auto end = fbb_.EndTable(start_);
573  auto o = ::flatbuffers::Offset<ShardTensor2D>(end);
574  return o;
575  }
576 };
577 
578 inline ::flatbuffers::Offset<ShardTensor2D> CreateShardTensor2D(
579  ::flatbuffers::FlatBufferBuilder &_fbb,
580  const tt::target::Dim2d *shard_mesh = nullptr) {
581  ShardTensor2DBuilder builder_(_fbb);
582  builder_.add_shard_mesh(shard_mesh);
583  return builder_.Finish();
584 }
585 
588  static auto constexpr Create = CreateShardTensor2D;
589 };
590 
591 struct AllGatherTensor FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
593  struct Traits;
594  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
595  return "tt.target.ttnn.AllGatherTensor";
596  }
597  bool Verify(::flatbuffers::Verifier &verifier) const {
598  return VerifyTableStart(verifier) &&
599  verifier.EndTable();
600  }
601 };
602 
605  ::flatbuffers::FlatBufferBuilder &fbb_;
606  ::flatbuffers::uoffset_t start_;
607  explicit AllGatherTensorBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
608  : fbb_(_fbb) {
609  start_ = fbb_.StartTable();
610  }
611  ::flatbuffers::Offset<AllGatherTensor> Finish() {
612  const auto end = fbb_.EndTable(start_);
613  auto o = ::flatbuffers::Offset<AllGatherTensor>(end);
614  return o;
615  }
616 };
617 
618 inline ::flatbuffers::Offset<AllGatherTensor> CreateAllGatherTensor(
619  ::flatbuffers::FlatBufferBuilder &_fbb) {
620  AllGatherTensorBuilder builder_(_fbb);
621  return builder_.Finish();
622 }
623 
626  static auto constexpr Create = CreateAllGatherTensor;
627 };
628 
629 struct DistributionStrategy FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
631  struct Traits;
632  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
633  return "tt.target.ttnn.DistributionStrategy";
634  }
635  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
637  VT_STRATEGY = 6
638  };
640  return static_cast<tt::target::ttnn::DistributedTensorConfig>(GetField<uint8_t>(VT_STRATEGY_TYPE, 0));
641  }
642  const void *strategy() const {
643  return GetPointer<const void *>(VT_STRATEGY);
644  }
645  template<typename T> const T *strategy_as() const;
646  const tt::target::ttnn::ReplicateTensor *strategy_as_ReplicateTensor() const {
647  return strategy_type() == tt::target::ttnn::DistributedTensorConfig::ReplicateTensor ? static_cast<const tt::target::ttnn::ReplicateTensor *>(strategy()) : nullptr;
648  }
649  const tt::target::ttnn::ShardTensor *strategy_as_ShardTensor() const {
650  return strategy_type() == tt::target::ttnn::DistributedTensorConfig::ShardTensor ? static_cast<const tt::target::ttnn::ShardTensor *>(strategy()) : nullptr;
651  }
652  const tt::target::ttnn::ShardTensor2D *strategy_as_ShardTensor2D() const {
653  return strategy_type() == tt::target::ttnn::DistributedTensorConfig::ShardTensor2D ? static_cast<const tt::target::ttnn::ShardTensor2D *>(strategy()) : nullptr;
654  }
655  const tt::target::ttnn::AllGatherTensor *strategy_as_AllGatherTensor() const {
656  return strategy_type() == tt::target::ttnn::DistributedTensorConfig::AllGatherTensor ? static_cast<const tt::target::ttnn::AllGatherTensor *>(strategy()) : nullptr;
657  }
658  bool Verify(::flatbuffers::Verifier &verifier) const {
659  return VerifyTableStart(verifier) &&
660  VerifyField<uint8_t>(verifier, VT_STRATEGY_TYPE, 1) &&
661  VerifyOffset(verifier, VT_STRATEGY) &&
662  VerifyDistributedTensorConfig(verifier, strategy(), strategy_type()) &&
663  verifier.EndTable();
664  }
665 };
666 
667 template<> inline const tt::target::ttnn::ReplicateTensor *DistributionStrategy::strategy_as<tt::target::ttnn::ReplicateTensor>() const {
668  return strategy_as_ReplicateTensor();
669 }
670 
671 template<> inline const tt::target::ttnn::ShardTensor *DistributionStrategy::strategy_as<tt::target::ttnn::ShardTensor>() const {
672  return strategy_as_ShardTensor();
673 }
674 
675 template<> inline const tt::target::ttnn::ShardTensor2D *DistributionStrategy::strategy_as<tt::target::ttnn::ShardTensor2D>() const {
676  return strategy_as_ShardTensor2D();
677 }
678 
679 template<> inline const tt::target::ttnn::AllGatherTensor *DistributionStrategy::strategy_as<tt::target::ttnn::AllGatherTensor>() const {
680  return strategy_as_AllGatherTensor();
681 }
682 
684  typedef DistributionStrategy Table;
685  ::flatbuffers::FlatBufferBuilder &fbb_;
686  ::flatbuffers::uoffset_t start_;
688  fbb_.AddElement<uint8_t>(DistributionStrategy::VT_STRATEGY_TYPE, static_cast<uint8_t>(strategy_type), 0);
689  }
690  void add_strategy(::flatbuffers::Offset<void> strategy) {
691  fbb_.AddOffset(DistributionStrategy::VT_STRATEGY, strategy);
692  }
693  explicit DistributionStrategyBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
694  : fbb_(_fbb) {
695  start_ = fbb_.StartTable();
696  }
697  ::flatbuffers::Offset<DistributionStrategy> Finish() {
698  const auto end = fbb_.EndTable(start_);
699  auto o = ::flatbuffers::Offset<DistributionStrategy>(end);
700  return o;
701  }
702 };
703 
704 inline ::flatbuffers::Offset<DistributionStrategy> CreateDistributionStrategy(
705  ::flatbuffers::FlatBufferBuilder &_fbb,
707  ::flatbuffers::Offset<void> strategy = 0) {
708  DistributionStrategyBuilder builder_(_fbb);
709  builder_.add_strategy(strategy);
710  builder_.add_strategy_type(strategy_type);
711  return builder_.Finish();
712 }
713 
715  using type = DistributionStrategy;
716  static auto constexpr Create = CreateDistributionStrategy;
717 };
718 
719 struct ShardSpec FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
721  struct Traits;
722  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
723  return "tt.target.ttnn.ShardSpec";
724  }
725  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
727  VT_SHAPE = 6,
729  VT_MODE = 10,
730  VT_PHYSICAL_SHARD_SHAPE = 12
731  };
732  const tt::target::ttnn::CoreRangeSet *core_range_set() const {
733  return GetPointer<const tt::target::ttnn::CoreRangeSet *>(VT_CORE_RANGE_SET);
734  }
735  const ::flatbuffers::Vector<int32_t> *shape() const {
736  return GetPointer<const ::flatbuffers::Vector<int32_t> *>(VT_SHAPE);
737  }
739  return static_cast<tt::target::ttnn::ShardOrientation>(GetField<uint32_t>(VT_ORIENTATION, 0));
740  }
742  return static_cast<tt::target::ttnn::ShardMode>(GetField<uint32_t>(VT_MODE, 0));
743  }
744  const ::flatbuffers::Vector<int32_t> *physical_shard_shape() const {
745  return GetPointer<const ::flatbuffers::Vector<int32_t> *>(VT_PHYSICAL_SHARD_SHAPE);
746  }
747  bool Verify(::flatbuffers::Verifier &verifier) const {
748  return VerifyTableStart(verifier) &&
749  VerifyOffset(verifier, VT_CORE_RANGE_SET) &&
750  verifier.VerifyTable(core_range_set()) &&
751  VerifyOffset(verifier, VT_SHAPE) &&
752  verifier.VerifyVector(shape()) &&
753  VerifyField<uint32_t>(verifier, VT_ORIENTATION, 4) &&
754  VerifyField<uint32_t>(verifier, VT_MODE, 4) &&
755  VerifyOffset(verifier, VT_PHYSICAL_SHARD_SHAPE) &&
756  verifier.VerifyVector(physical_shard_shape()) &&
757  verifier.EndTable();
758  }
759 };
760 
762  typedef ShardSpec Table;
763  ::flatbuffers::FlatBufferBuilder &fbb_;
764  ::flatbuffers::uoffset_t start_;
765  void add_core_range_set(::flatbuffers::Offset<tt::target::ttnn::CoreRangeSet> core_range_set) {
766  fbb_.AddOffset(ShardSpec::VT_CORE_RANGE_SET, core_range_set);
767  }
768  void add_shape(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> shape) {
769  fbb_.AddOffset(ShardSpec::VT_SHAPE, shape);
770  }
772  fbb_.AddElement<uint32_t>(ShardSpec::VT_ORIENTATION, static_cast<uint32_t>(orientation), 0);
773  }
775  fbb_.AddElement<uint32_t>(ShardSpec::VT_MODE, static_cast<uint32_t>(mode), 0);
776  }
777  void add_physical_shard_shape(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> physical_shard_shape) {
778  fbb_.AddOffset(ShardSpec::VT_PHYSICAL_SHARD_SHAPE, physical_shard_shape);
779  }
780  explicit ShardSpecBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
781  : fbb_(_fbb) {
782  start_ = fbb_.StartTable();
783  }
784  ::flatbuffers::Offset<ShardSpec> Finish() {
785  const auto end = fbb_.EndTable(start_);
786  auto o = ::flatbuffers::Offset<ShardSpec>(end);
787  return o;
788  }
789 };
790 
791 inline ::flatbuffers::Offset<ShardSpec> CreateShardSpec(
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) {
798  ShardSpecBuilder builder_(_fbb);
799  builder_.add_physical_shard_shape(physical_shard_shape);
800  builder_.add_mode(mode);
801  builder_.add_orientation(orientation);
802  builder_.add_shape(shape);
803  builder_.add_core_range_set(core_range_set);
804  return builder_.Finish();
805 }
806 
808  using type = ShardSpec;
809  static auto constexpr Create = CreateShardSpec;
810 };
811 
812 inline ::flatbuffers::Offset<ShardSpec> CreateShardSpecDirect(
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;
822  _fbb,
823  core_range_set,
824  shape__,
825  orientation,
826  mode,
827  physical_shard_shape__);
828 }
829 
830 struct MemoryConfig FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
832  struct Traits;
833  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
834  return "tt.target.ttnn.MemoryConfig";
835  }
836  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
839  VT_SHARD_SPEC = 8
840  };
842  return static_cast<tt::target::ttnn::TensorMemoryLayout>(GetField<uint16_t>(VT_TENSOR_MEMORY_LAYOUT, 0));
843  }
845  return static_cast<tt::target::BufferType>(GetField<uint16_t>(VT_BUFFER_TYPE, 0));
846  }
847  const tt::target::ttnn::ShardSpec *shard_spec() const {
848  return GetPointer<const tt::target::ttnn::ShardSpec *>(VT_SHARD_SPEC);
849  }
850  bool Verify(::flatbuffers::Verifier &verifier) const {
851  return VerifyTableStart(verifier) &&
852  VerifyField<uint16_t>(verifier, VT_TENSOR_MEMORY_LAYOUT, 2) &&
853  VerifyField<uint16_t>(verifier, VT_BUFFER_TYPE, 2) &&
854  VerifyOffset(verifier, VT_SHARD_SPEC) &&
855  verifier.VerifyTable(shard_spec()) &&
856  verifier.EndTable();
857  }
858 };
859 
861  typedef MemoryConfig Table;
862  ::flatbuffers::FlatBufferBuilder &fbb_;
863  ::flatbuffers::uoffset_t start_;
865  fbb_.AddElement<uint16_t>(MemoryConfig::VT_TENSOR_MEMORY_LAYOUT, static_cast<uint16_t>(tensor_memory_layout), 0);
866  }
868  fbb_.AddElement<uint16_t>(MemoryConfig::VT_BUFFER_TYPE, static_cast<uint16_t>(buffer_type), 0);
869  }
870  void add_shard_spec(::flatbuffers::Offset<tt::target::ttnn::ShardSpec> shard_spec) {
871  fbb_.AddOffset(MemoryConfig::VT_SHARD_SPEC, shard_spec);
872  }
873  explicit MemoryConfigBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
874  : fbb_(_fbb) {
875  start_ = fbb_.StartTable();
876  }
877  ::flatbuffers::Offset<MemoryConfig> Finish() {
878  const auto end = fbb_.EndTable(start_);
879  auto o = ::flatbuffers::Offset<MemoryConfig>(end);
880  return o;
881  }
882 };
883 
884 inline ::flatbuffers::Offset<MemoryConfig> CreateMemoryConfig(
885  ::flatbuffers::FlatBufferBuilder &_fbb,
888  ::flatbuffers::Offset<tt::target::ttnn::ShardSpec> shard_spec = 0) {
889  MemoryConfigBuilder builder_(_fbb);
890  builder_.add_shard_spec(shard_spec);
891  builder_.add_buffer_type(buffer_type);
892  builder_.add_tensor_memory_layout(tensor_memory_layout);
893  return builder_.Finish();
894 }
895 
897  using type = MemoryConfig;
898  static auto constexpr Create = CreateMemoryConfig;
899 };
900 
901 struct MemoryDesc FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
903  struct Traits;
904  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
905  return "tt.target.ttnn.MemoryDesc";
906  }
907  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
912  VT_SIZE = 12
913  };
915  return static_cast<tt::target::ttnn::StorageType>(GetField<uint16_t>(VT_STORAGE_TYPE, 0));
916  }
917  const tt::target::Dim2d *tile_shape() const {
918  return GetStruct<const tt::target::Dim2d *>(VT_TILE_SHAPE);
919  }
921  return static_cast<tt::target::DataType>(GetField<uint16_t>(VT_DATA_TYPE, 0));
922  }
923  const tt::target::ttnn::MemoryConfig *memory_config() const {
924  return GetPointer<const tt::target::ttnn::MemoryConfig *>(VT_MEMORY_CONFIG);
925  }
926  uint64_t size() const {
927  return GetField<uint64_t>(VT_SIZE, 0);
928  }
929  bool Verify(::flatbuffers::Verifier &verifier) const {
930  return VerifyTableStart(verifier) &&
931  VerifyField<uint16_t>(verifier, VT_STORAGE_TYPE, 2) &&
932  VerifyField<tt::target::Dim2d>(verifier, VT_TILE_SHAPE, 4) &&
933  VerifyField<uint16_t>(verifier, VT_DATA_TYPE, 2) &&
934  VerifyOffset(verifier, VT_MEMORY_CONFIG) &&
935  verifier.VerifyTable(memory_config()) &&
936  VerifyField<uint64_t>(verifier, VT_SIZE, 8) &&
937  verifier.EndTable();
938  }
939 };
940 
942  typedef MemoryDesc Table;
943  ::flatbuffers::FlatBufferBuilder &fbb_;
944  ::flatbuffers::uoffset_t start_;
946  fbb_.AddElement<uint16_t>(MemoryDesc::VT_STORAGE_TYPE, static_cast<uint16_t>(storage_type), 0);
947  }
948  void add_tile_shape(const tt::target::Dim2d *tile_shape) {
949  fbb_.AddStruct(MemoryDesc::VT_TILE_SHAPE, tile_shape);
950  }
952  fbb_.AddElement<uint16_t>(MemoryDesc::VT_DATA_TYPE, static_cast<uint16_t>(data_type), 0);
953  }
954  void add_memory_config(::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> memory_config) {
955  fbb_.AddOffset(MemoryDesc::VT_MEMORY_CONFIG, memory_config);
956  }
957  void add_size(uint64_t size) {
958  fbb_.AddElement<uint64_t>(MemoryDesc::VT_SIZE, size, 0);
959  }
960  explicit MemoryDescBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
961  : fbb_(_fbb) {
962  start_ = fbb_.StartTable();
963  }
964  ::flatbuffers::Offset<MemoryDesc> Finish() {
965  const auto end = fbb_.EndTable(start_);
966  auto o = ::flatbuffers::Offset<MemoryDesc>(end);
967  return o;
968  }
969 };
970 
971 inline ::flatbuffers::Offset<MemoryDesc> CreateMemoryDesc(
972  ::flatbuffers::FlatBufferBuilder &_fbb,
974  const tt::target::Dim2d *tile_shape = nullptr,
976  ::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> memory_config = 0,
977  uint64_t size = 0) {
978  MemoryDescBuilder builder_(_fbb);
979  builder_.add_size(size);
980  builder_.add_memory_config(memory_config);
981  builder_.add_tile_shape(tile_shape);
982  builder_.add_data_type(data_type);
983  builder_.add_storage_type(storage_type);
984  return builder_.Finish();
985 }
986 
988  using type = MemoryDesc;
989  static auto constexpr Create = CreateMemoryDesc;
990 };
991 
992 struct LayoutDesc FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
994  struct Traits;
995  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
996  return "tt.target.ttnn.LayoutDesc";
997  }
998  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1001  VT_STRATEGY = 8
1002  };
1004  return static_cast<tt::target::OOBVal>(GetField<uint16_t>(VT_OOB_VAL, 0));
1005  }
1006  const tt::target::ttnn::MemoryDesc *memory_desc() const {
1007  return GetPointer<const tt::target::ttnn::MemoryDesc *>(VT_MEMORY_DESC);
1008  }
1009  const tt::target::ttnn::DistributionStrategy *strategy() const {
1010  return GetPointer<const tt::target::ttnn::DistributionStrategy *>(VT_STRATEGY);
1011  }
1012  bool Verify(::flatbuffers::Verifier &verifier) const {
1013  return VerifyTableStart(verifier) &&
1014  VerifyField<uint16_t>(verifier, VT_OOB_VAL, 2) &&
1015  VerifyOffset(verifier, VT_MEMORY_DESC) &&
1016  verifier.VerifyTable(memory_desc()) &&
1017  VerifyOffset(verifier, VT_STRATEGY) &&
1018  verifier.VerifyTable(strategy()) &&
1019  verifier.EndTable();
1020  }
1021 };
1022 
1024  typedef LayoutDesc Table;
1025  ::flatbuffers::FlatBufferBuilder &fbb_;
1026  ::flatbuffers::uoffset_t start_;
1028  fbb_.AddElement<uint16_t>(LayoutDesc::VT_OOB_VAL, static_cast<uint16_t>(oob_val), 0);
1029  }
1030  void add_memory_desc(::flatbuffers::Offset<tt::target::ttnn::MemoryDesc> memory_desc) {
1031  fbb_.AddOffset(LayoutDesc::VT_MEMORY_DESC, memory_desc);
1032  }
1033  void add_strategy(::flatbuffers::Offset<tt::target::ttnn::DistributionStrategy> strategy) {
1034  fbb_.AddOffset(LayoutDesc::VT_STRATEGY, strategy);
1035  }
1036  explicit LayoutDescBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
1037  : fbb_(_fbb) {
1038  start_ = fbb_.StartTable();
1039  }
1040  ::flatbuffers::Offset<LayoutDesc> Finish() {
1041  const auto end = fbb_.EndTable(start_);
1042  auto o = ::flatbuffers::Offset<LayoutDesc>(end);
1043  return o;
1044  }
1045 };
1046 
1047 inline ::flatbuffers::Offset<LayoutDesc> CreateLayoutDesc(
1048  ::flatbuffers::FlatBufferBuilder &_fbb,
1050  ::flatbuffers::Offset<tt::target::ttnn::MemoryDesc> memory_desc = 0,
1051  ::flatbuffers::Offset<tt::target::ttnn::DistributionStrategy> strategy = 0) {
1052  LayoutDescBuilder builder_(_fbb);
1053  builder_.add_strategy(strategy);
1054  builder_.add_memory_desc(memory_desc);
1055  builder_.add_oob_val(oob_val);
1056  return builder_.Finish();
1057 }
1058 
1060  using type = LayoutDesc;
1061  static auto constexpr Create = CreateLayoutDesc;
1062 };
1063 
1064 struct TensorDesc FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
1066  struct Traits;
1067  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
1068  return "tt.target.ttnn.TensorDesc";
1069  }
1070  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1071  VT_SHAPE = 4,
1073  VT_LAYOUT = 8
1074  };
1075  const ::flatbuffers::Vector<int32_t> *shape() const {
1076  return GetPointer<const ::flatbuffers::Vector<int32_t> *>(VT_SHAPE);
1077  }
1078  const ::flatbuffers::Vector<int32_t> *mesh_shape() const {
1079  return GetPointer<const ::flatbuffers::Vector<int32_t> *>(VT_MESH_SHAPE);
1080  }
1081  const tt::target::ttnn::LayoutDesc *layout() const {
1082  return GetPointer<const tt::target::ttnn::LayoutDesc *>(VT_LAYOUT);
1083  }
1084  bool Verify(::flatbuffers::Verifier &verifier) const {
1085  return VerifyTableStart(verifier) &&
1086  VerifyOffset(verifier, VT_SHAPE) &&
1087  verifier.VerifyVector(shape()) &&
1088  VerifyOffset(verifier, VT_MESH_SHAPE) &&
1089  verifier.VerifyVector(mesh_shape()) &&
1090  VerifyOffset(verifier, VT_LAYOUT) &&
1091  verifier.VerifyTable(layout()) &&
1092  verifier.EndTable();
1093  }
1094 };
1095 
1097  typedef TensorDesc Table;
1098  ::flatbuffers::FlatBufferBuilder &fbb_;
1099  ::flatbuffers::uoffset_t start_;
1100  void add_shape(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> shape) {
1101  fbb_.AddOffset(TensorDesc::VT_SHAPE, shape);
1102  }
1103  void add_mesh_shape(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> mesh_shape) {
1104  fbb_.AddOffset(TensorDesc::VT_MESH_SHAPE, mesh_shape);
1105  }
1106  void add_layout(::flatbuffers::Offset<tt::target::ttnn::LayoutDesc> layout) {
1107  fbb_.AddOffset(TensorDesc::VT_LAYOUT, layout);
1108  }
1109  explicit TensorDescBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
1110  : fbb_(_fbb) {
1111  start_ = fbb_.StartTable();
1112  }
1113  ::flatbuffers::Offset<TensorDesc> Finish() {
1114  const auto end = fbb_.EndTable(start_);
1115  auto o = ::flatbuffers::Offset<TensorDesc>(end);
1116  return o;
1117  }
1118 };
1119 
1120 inline ::flatbuffers::Offset<TensorDesc> CreateTensorDesc(
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) {
1125  TensorDescBuilder builder_(_fbb);
1126  builder_.add_layout(layout);
1127  builder_.add_mesh_shape(mesh_shape);
1128  builder_.add_shape(shape);
1129  return builder_.Finish();
1130 }
1131 
1133  using type = TensorDesc;
1134  static auto constexpr Create = CreateTensorDesc;
1135 };
1136 
1137 inline ::flatbuffers::Offset<TensorDesc> CreateTensorDescDirect(
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;
1145  _fbb,
1146  shape__,
1147  mesh_shape__,
1148  layout);
1149 }
1150 
1151 struct TensorRef FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
1153  struct Traits;
1154  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
1155  return "tt.target.ttnn.TensorRef";
1156  }
1157  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1159  VT_SIZE = 6,
1160  VT_DESC = 8
1161  };
1162  uint32_t global_id() const {
1163  return GetField<uint32_t>(VT_GLOBAL_ID, 0);
1164  }
1165  uint64_t size() const {
1166  return GetField<uint64_t>(VT_SIZE, 0);
1167  }
1168  const tt::target::ttnn::TensorDesc *desc() const {
1169  return GetPointer<const tt::target::ttnn::TensorDesc *>(VT_DESC);
1170  }
1171  bool Verify(::flatbuffers::Verifier &verifier) const {
1172  return VerifyTableStart(verifier) &&
1173  VerifyField<uint32_t>(verifier, VT_GLOBAL_ID, 4) &&
1174  VerifyField<uint64_t>(verifier, VT_SIZE, 8) &&
1175  VerifyOffset(verifier, VT_DESC) &&
1176  verifier.VerifyTable(desc()) &&
1177  verifier.EndTable();
1178  }
1179 };
1180 
1182  typedef TensorRef Table;
1183  ::flatbuffers::FlatBufferBuilder &fbb_;
1184  ::flatbuffers::uoffset_t start_;
1185  void add_global_id(uint32_t global_id) {
1186  fbb_.AddElement<uint32_t>(TensorRef::VT_GLOBAL_ID, global_id, 0);
1187  }
1188  void add_size(uint64_t size) {
1189  fbb_.AddElement<uint64_t>(TensorRef::VT_SIZE, size, 0);
1190  }
1191  void add_desc(::flatbuffers::Offset<tt::target::ttnn::TensorDesc> desc) {
1192  fbb_.AddOffset(TensorRef::VT_DESC, desc);
1193  }
1194  explicit TensorRefBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
1195  : fbb_(_fbb) {
1196  start_ = fbb_.StartTable();
1197  }
1198  ::flatbuffers::Offset<TensorRef> Finish() {
1199  const auto end = fbb_.EndTable(start_);
1200  auto o = ::flatbuffers::Offset<TensorRef>(end);
1201  return o;
1202  }
1203 };
1204 
1205 inline ::flatbuffers::Offset<TensorRef> CreateTensorRef(
1206  ::flatbuffers::FlatBufferBuilder &_fbb,
1207  uint32_t global_id = 0,
1208  uint64_t size = 0,
1209  ::flatbuffers::Offset<tt::target::ttnn::TensorDesc> desc = 0) {
1210  TensorRefBuilder builder_(_fbb);
1211  builder_.add_size(size);
1212  builder_.add_desc(desc);
1213  builder_.add_global_id(global_id);
1214  return builder_.Finish();
1215 }
1216 
1218  using type = TensorRef;
1219  static auto constexpr Create = CreateTensorRef;
1220 };
1221 
1222 inline bool VerifyDistributedTensorConfig(::flatbuffers::Verifier &verifier, const void *obj, DistributedTensorConfig type) {
1223  switch (type) {
1225  return true;
1226  }
1228  auto ptr = reinterpret_cast<const tt::target::ttnn::ReplicateTensor *>(obj);
1229  return verifier.VerifyTable(ptr);
1230  }
1232  auto ptr = reinterpret_cast<const tt::target::ttnn::ShardTensor *>(obj);
1233  return verifier.VerifyTable(ptr);
1234  }
1236  auto ptr = reinterpret_cast<const tt::target::ttnn::ShardTensor2D *>(obj);
1237  return verifier.VerifyTable(ptr);
1238  }
1240  auto ptr = reinterpret_cast<const tt::target::ttnn::AllGatherTensor *>(obj);
1241  return verifier.VerifyTable(ptr);
1242  }
1243  default: return true;
1244  }
1245 }
1246 
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) {
1252  verifier, values->Get(i), types->GetEnum<DistributedTensorConfig>(i))) {
1253  return false;
1254  }
1255  }
1256  return true;
1257 }
1258 
1259 } // namespace ttnn
1260 } // namespace target
1261 } // namespace tt
1262 
1263 #endif // FLATBUFFERS_GENERATED_TYPES_TT_TARGET_TTNN_H_
VT_SHARD_SPEC
Definition: types_generated.h:122
VT_TILE_SHAPE
Definition: types_generated.h:361
VT_SHAPE
Definition: types_generated.h:360
VT_DATA_TYPE
Definition: types_generated.h:362
VT_CORE_RANGE_SET
Definition: types_generated.h:50
VT_GLOBAL_ID
Definition: types_generated.h:482
VT_SIZE
Definition: types_generated.h:193
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 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