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  SingleBank = 1,
65  HeightSharded = 2,
66  WidthSharded = 3,
67  BlockSharded = 4,
68  MIN = Interleaved,
70 };
71 
73  static const TensorMemoryLayout values[] = {
79  };
80  return values;
81 }
82 
83 inline const char * const *EnumNamesTensorMemoryLayout() {
84  static const char * const names[6] = {
85  "Interleaved",
86  "SingleBank",
87  "HeightSharded",
88  "WidthSharded",
89  "BlockSharded",
90  nullptr
91  };
92  return names;
93 }
94 
96  if (::flatbuffers::IsOutRange(e, TensorMemoryLayout::Interleaved, TensorMemoryLayout::BlockSharded)) return "";
97  const size_t index = static_cast<size_t>(e);
98  return EnumNamesTensorMemoryLayout()[index];
99 }
100 
101 enum class StorageType : uint16_t {
102  Host = 0,
103  Device = 1,
104  MultiDeviceHost = 2,
105  MIN = Host,
107 };
108 
109 inline const StorageType (&EnumValuesStorageType())[3] {
110  static const StorageType values[] = {
114  };
115  return values;
116 }
117 
118 inline const char * const *EnumNamesStorageType() {
119  static const char * const names[4] = {
120  "Host",
121  "Device",
122  "MultiDeviceHost",
123  nullptr
124  };
125  return names;
126 }
127 
128 inline const char *EnumNameStorageType(StorageType e) {
129  if (::flatbuffers::IsOutRange(e, StorageType::Host, StorageType::MultiDeviceHost)) return "";
130  const size_t index = static_cast<size_t>(e);
131  return EnumNamesStorageType()[index];
132 }
133 
134 enum class MeshShardDirection : uint32_t {
135  FullToShardShape = 0,
136  ShardToFullShape = 1,
139 };
140 
142  static const MeshShardDirection values[] = {
145  };
146  return values;
147 }
148 
149 inline const char * const *EnumNamesMeshShardDirection() {
150  static const char * const names[3] = {
151  "FullToShardShape",
152  "ShardToFullShape",
153  nullptr
154  };
155  return names;
156 }
157 
159  if (::flatbuffers::IsOutRange(e, MeshShardDirection::FullToShardShape, MeshShardDirection::ShardToFullShape)) return "";
160  const size_t index = static_cast<size_t>(e);
161  return EnumNamesMeshShardDirection()[index];
162 }
163 
164 enum class MeshShardType : uint32_t {
165  Identity = 0,
166  Replicate = 1,
167  Maximal = 2,
168  Devices = 3,
169  MIN = Identity,
170  MAX = Devices
171 };
172 
174  static const MeshShardType values[] = {
179  };
180  return values;
181 }
182 
183 inline const char * const *EnumNamesMeshShardType() {
184  static const char * const names[5] = {
185  "Identity",
186  "Replicate",
187  "Maximal",
188  "Devices",
189  nullptr
190  };
191  return names;
192 }
193 
194 inline const char *EnumNameMeshShardType(MeshShardType e) {
195  if (::flatbuffers::IsOutRange(e, MeshShardType::Identity, MeshShardType::Devices)) return "";
196  const size_t index = static_cast<size_t>(e);
197  return EnumNamesMeshShardType()[index];
198 }
199 
200 enum class ShardOrientation : uint32_t {
201  RowMajor = 0,
202  ColMajor = 1,
203  MIN = RowMajor,
204  MAX = ColMajor
205 };
206 
208  static const ShardOrientation values[] = {
211  };
212  return values;
213 }
214 
215 inline const char * const *EnumNamesShardOrientation() {
216  static const char * const names[3] = {
217  "RowMajor",
218  "ColMajor",
219  nullptr
220  };
221  return names;
222 }
223 
225  if (::flatbuffers::IsOutRange(e, ShardOrientation::RowMajor, ShardOrientation::ColMajor)) return "";
226  const size_t index = static_cast<size_t>(e);
227  return EnumNamesShardOrientation()[index];
228 }
229 
230 enum class ShardMode : uint32_t {
231  Physical = 0,
232  Logical = 1,
233  MIN = Physical,
234  MAX = Logical
235 };
236 
237 inline const ShardMode (&EnumValuesShardMode())[2] {
238  static const ShardMode values[] = {
241  };
242  return values;
243 }
244 
245 inline const char * const *EnumNamesShardMode() {
246  static const char * const names[3] = {
247  "Physical",
248  "Logical",
249  nullptr
250  };
251  return names;
252 }
253 
254 inline const char *EnumNameShardMode(ShardMode e) {
255  if (::flatbuffers::IsOutRange(e, ShardMode::Physical, ShardMode::Logical)) return "";
256  const size_t index = static_cast<size_t>(e);
257  return EnumNamesShardMode()[index];
258 }
259 
260 enum class DistributedTensorConfig : uint8_t {
261  NONE = 0,
262  ReplicateTensor = 1,
263  ShardTensor = 2,
264  ShardTensor2D = 3,
265  AllGatherTensor = 4,
266  MIN = NONE,
268 };
269 
271  static const DistributedTensorConfig values[] = {
277  };
278  return values;
279 }
280 
281 inline const char * const *EnumNamesDistributedTensorConfig() {
282  static const char * const names[6] = {
283  "NONE",
284  "ReplicateTensor",
285  "ShardTensor",
286  "ShardTensor2D",
287  "AllGatherTensor",
288  nullptr
289  };
290  return names;
291 }
292 
294  if (::flatbuffers::IsOutRange(e, DistributedTensorConfig::NONE, DistributedTensorConfig::AllGatherTensor)) return "";
295  const size_t index = static_cast<size_t>(e);
296  return EnumNamesDistributedTensorConfig()[index];
297 }
298 
299 template<typename T> struct DistributedTensorConfigTraits {
301 };
302 
303 template<> struct DistributedTensorConfigTraits<tt::target::ttnn::ReplicateTensor> {
305 };
306 
307 template<> struct DistributedTensorConfigTraits<tt::target::ttnn::ShardTensor> {
309 };
310 
311 template<> struct DistributedTensorConfigTraits<tt::target::ttnn::ShardTensor2D> {
313 };
314 
315 template<> struct DistributedTensorConfigTraits<tt::target::ttnn::AllGatherTensor> {
317 };
318 
319 bool VerifyDistributedTensorConfig(::flatbuffers::Verifier &verifier, const void *obj, DistributedTensorConfig type);
320 bool VerifyDistributedTensorConfigVector(::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset<void>> *values, const ::flatbuffers::Vector<DistributedTensorConfig> *types);
321 
323  private:
324  uint64_t x_;
325  uint64_t y_;
326 
327  public:
328  struct Traits;
329  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
330  return "tt.target.ttnn.CoreCoord";
331  }
332  CoreCoord()
333  : x_(0),
334  y_(0) {
335  }
336  CoreCoord(uint64_t _x, uint64_t _y)
337  : x_(::flatbuffers::EndianScalar(_x)),
338  y_(::flatbuffers::EndianScalar(_y)) {
339  }
340  uint64_t x() const {
341  return ::flatbuffers::EndianScalar(x_);
342  }
343  uint64_t y() const {
344  return ::flatbuffers::EndianScalar(y_);
345  }
346 };
347 FLATBUFFERS_STRUCT_END(CoreCoord, 16);
348 
350  using type = CoreCoord;
351 };
352 
354  private:
355  tt::target::ttnn::CoreCoord start_coord_;
356  tt::target::ttnn::CoreCoord end_coord_;
357 
358  public:
359  struct Traits;
360  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
361  return "tt.target.ttnn.CoreRange";
362  }
363  CoreRange()
364  : start_coord_(),
365  end_coord_() {
366  }
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) {
370  }
371  const tt::target::ttnn::CoreCoord &start_coord() const {
372  return start_coord_;
373  }
374  const tt::target::ttnn::CoreCoord &end_coord() const {
375  return end_coord_;
376  }
377 };
378 FLATBUFFERS_STRUCT_END(CoreRange, 32);
379 
381  using type = CoreRange;
382 };
383 
384 struct CoreRangeSet FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
386  struct Traits;
387  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
388  return "tt.target.ttnn.CoreRangeSet";
389  }
390  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
391  VT_CORE_RANGES = 4
392  };
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);
395  }
396  bool Verify(::flatbuffers::Verifier &verifier) const {
397  return VerifyTableStart(verifier) &&
398  VerifyOffset(verifier, VT_CORE_RANGES) &&
399  verifier.VerifyVector(core_ranges()) &&
400  verifier.EndTable();
401  }
402 };
403 
405  typedef CoreRangeSet Table;
406  ::flatbuffers::FlatBufferBuilder &fbb_;
407  ::flatbuffers::uoffset_t start_;
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);
410  }
411  explicit CoreRangeSetBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
412  : fbb_(_fbb) {
413  start_ = fbb_.StartTable();
414  }
415  ::flatbuffers::Offset<CoreRangeSet> Finish() {
416  const auto end = fbb_.EndTable(start_);
417  auto o = ::flatbuffers::Offset<CoreRangeSet>(end);
418  return o;
419  }
420 };
421 
422 inline ::flatbuffers::Offset<CoreRangeSet> CreateCoreRangeSet(
423  ::flatbuffers::FlatBufferBuilder &_fbb,
424  ::flatbuffers::Offset<::flatbuffers::Vector<const tt::target::ttnn::CoreRange *>> core_ranges = 0) {
425  CoreRangeSetBuilder builder_(_fbb);
426  builder_.add_core_ranges(core_ranges);
427  return builder_.Finish();
428 }
429 
431  using type = CoreRangeSet;
432  static auto constexpr Create = CreateCoreRangeSet;
433 };
434 
435 inline ::flatbuffers::Offset<CoreRangeSet> CreateCoreRangeSetDirect(
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;
440  _fbb,
441  core_ranges__);
442 }
443 
444 struct ReplicateTensor FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
446  struct Traits;
447  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
448  return "tt.target.ttnn.ReplicateTensor";
449  }
450  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
451  VT_REPLICATION_FACTOR = 4
452  };
453  uint32_t replication_factor() const {
454  return GetField<uint32_t>(VT_REPLICATION_FACTOR, 0);
455  }
456  bool Verify(::flatbuffers::Verifier &verifier) const {
457  return VerifyTableStart(verifier) &&
458  VerifyField<uint32_t>(verifier, VT_REPLICATION_FACTOR, 4) &&
459  verifier.EndTable();
460  }
461 };
462 
465  ::flatbuffers::FlatBufferBuilder &fbb_;
466  ::flatbuffers::uoffset_t start_;
467  void add_replication_factor(uint32_t replication_factor) {
468  fbb_.AddElement<uint32_t>(ReplicateTensor::VT_REPLICATION_FACTOR, replication_factor, 0);
469  }
470  explicit ReplicateTensorBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
471  : fbb_(_fbb) {
472  start_ = fbb_.StartTable();
473  }
474  ::flatbuffers::Offset<ReplicateTensor> Finish() {
475  const auto end = fbb_.EndTable(start_);
476  auto o = ::flatbuffers::Offset<ReplicateTensor>(end);
477  return o;
478  }
479 };
480 
481 inline ::flatbuffers::Offset<ReplicateTensor> CreateReplicateTensor(
482  ::flatbuffers::FlatBufferBuilder &_fbb,
483  uint32_t replication_factor = 0) {
484  ReplicateTensorBuilder builder_(_fbb);
485  builder_.add_replication_factor(replication_factor);
486  return builder_.Finish();
487 }
488 
491  static auto constexpr Create = CreateReplicateTensor;
492 };
493 
494 struct ShardTensor FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
496  struct Traits;
497  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
498  return "tt.target.ttnn.ShardTensor";
499  }
500  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
501  VT_SHARD_DIM = 4
502  };
503  uint32_t shard_dim() const {
504  return GetField<uint32_t>(VT_SHARD_DIM, 0);
505  }
506  bool Verify(::flatbuffers::Verifier &verifier) const {
507  return VerifyTableStart(verifier) &&
508  VerifyField<uint32_t>(verifier, VT_SHARD_DIM, 4) &&
509  verifier.EndTable();
510  }
511 };
512 
515  ::flatbuffers::FlatBufferBuilder &fbb_;
516  ::flatbuffers::uoffset_t start_;
517  void add_shard_dim(uint32_t shard_dim) {
518  fbb_.AddElement<uint32_t>(ShardTensor::VT_SHARD_DIM, shard_dim, 0);
519  }
520  explicit ShardTensorBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
521  : fbb_(_fbb) {
522  start_ = fbb_.StartTable();
523  }
524  ::flatbuffers::Offset<ShardTensor> Finish() {
525  const auto end = fbb_.EndTable(start_);
526  auto o = ::flatbuffers::Offset<ShardTensor>(end);
527  return o;
528  }
529 };
530 
531 inline ::flatbuffers::Offset<ShardTensor> CreateShardTensor(
532  ::flatbuffers::FlatBufferBuilder &_fbb,
533  uint32_t shard_dim = 0) {
534  ShardTensorBuilder builder_(_fbb);
535  builder_.add_shard_dim(shard_dim);
536  return builder_.Finish();
537 }
538 
540  using type = ShardTensor;
541  static auto constexpr Create = CreateShardTensor;
542 };
543 
544 struct ShardTensor2D FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
546  struct Traits;
547  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
548  return "tt.target.ttnn.ShardTensor2D";
549  }
550  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
551  VT_SHARD_MESH = 4
552  };
553  const tt::target::Dim2d *shard_mesh() const {
554  return GetStruct<const tt::target::Dim2d *>(VT_SHARD_MESH);
555  }
556  bool Verify(::flatbuffers::Verifier &verifier) const {
557  return VerifyTableStart(verifier) &&
558  VerifyField<tt::target::Dim2d>(verifier, VT_SHARD_MESH, 4) &&
559  verifier.EndTable();
560  }
561 };
562 
565  ::flatbuffers::FlatBufferBuilder &fbb_;
566  ::flatbuffers::uoffset_t start_;
567  void add_shard_mesh(const tt::target::Dim2d *shard_mesh) {
568  fbb_.AddStruct(ShardTensor2D::VT_SHARD_MESH, shard_mesh);
569  }
570  explicit ShardTensor2DBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
571  : fbb_(_fbb) {
572  start_ = fbb_.StartTable();
573  }
574  ::flatbuffers::Offset<ShardTensor2D> Finish() {
575  const auto end = fbb_.EndTable(start_);
576  auto o = ::flatbuffers::Offset<ShardTensor2D>(end);
577  return o;
578  }
579 };
580 
581 inline ::flatbuffers::Offset<ShardTensor2D> CreateShardTensor2D(
582  ::flatbuffers::FlatBufferBuilder &_fbb,
583  const tt::target::Dim2d *shard_mesh = nullptr) {
584  ShardTensor2DBuilder builder_(_fbb);
585  builder_.add_shard_mesh(shard_mesh);
586  return builder_.Finish();
587 }
588 
591  static auto constexpr Create = CreateShardTensor2D;
592 };
593 
594 struct AllGatherTensor FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
596  struct Traits;
597  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
598  return "tt.target.ttnn.AllGatherTensor";
599  }
600  bool Verify(::flatbuffers::Verifier &verifier) const {
601  return VerifyTableStart(verifier) &&
602  verifier.EndTable();
603  }
604 };
605 
608  ::flatbuffers::FlatBufferBuilder &fbb_;
609  ::flatbuffers::uoffset_t start_;
610  explicit AllGatherTensorBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
611  : fbb_(_fbb) {
612  start_ = fbb_.StartTable();
613  }
614  ::flatbuffers::Offset<AllGatherTensor> Finish() {
615  const auto end = fbb_.EndTable(start_);
616  auto o = ::flatbuffers::Offset<AllGatherTensor>(end);
617  return o;
618  }
619 };
620 
621 inline ::flatbuffers::Offset<AllGatherTensor> CreateAllGatherTensor(
622  ::flatbuffers::FlatBufferBuilder &_fbb) {
623  AllGatherTensorBuilder builder_(_fbb);
624  return builder_.Finish();
625 }
626 
629  static auto constexpr Create = CreateAllGatherTensor;
630 };
631 
632 struct DistributionStrategy FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
634  struct Traits;
635  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
636  return "tt.target.ttnn.DistributionStrategy";
637  }
638  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
640  VT_STRATEGY = 6
641  };
643  return static_cast<tt::target::ttnn::DistributedTensorConfig>(GetField<uint8_t>(VT_STRATEGY_TYPE, 0));
644  }
645  const void *strategy() const {
646  return GetPointer<const void *>(VT_STRATEGY);
647  }
648  template<typename T> const T *strategy_as() const;
649  const tt::target::ttnn::ReplicateTensor *strategy_as_ReplicateTensor() const {
650  return strategy_type() == tt::target::ttnn::DistributedTensorConfig::ReplicateTensor ? static_cast<const tt::target::ttnn::ReplicateTensor *>(strategy()) : nullptr;
651  }
652  const tt::target::ttnn::ShardTensor *strategy_as_ShardTensor() const {
653  return strategy_type() == tt::target::ttnn::DistributedTensorConfig::ShardTensor ? static_cast<const tt::target::ttnn::ShardTensor *>(strategy()) : nullptr;
654  }
655  const tt::target::ttnn::ShardTensor2D *strategy_as_ShardTensor2D() const {
656  return strategy_type() == tt::target::ttnn::DistributedTensorConfig::ShardTensor2D ? static_cast<const tt::target::ttnn::ShardTensor2D *>(strategy()) : nullptr;
657  }
658  const tt::target::ttnn::AllGatherTensor *strategy_as_AllGatherTensor() const {
659  return strategy_type() == tt::target::ttnn::DistributedTensorConfig::AllGatherTensor ? static_cast<const tt::target::ttnn::AllGatherTensor *>(strategy()) : nullptr;
660  }
661  bool Verify(::flatbuffers::Verifier &verifier) const {
662  return VerifyTableStart(verifier) &&
663  VerifyField<uint8_t>(verifier, VT_STRATEGY_TYPE, 1) &&
664  VerifyOffset(verifier, VT_STRATEGY) &&
665  VerifyDistributedTensorConfig(verifier, strategy(), strategy_type()) &&
666  verifier.EndTable();
667  }
668 };
669 
670 template<> inline const tt::target::ttnn::ReplicateTensor *DistributionStrategy::strategy_as<tt::target::ttnn::ReplicateTensor>() const {
671  return strategy_as_ReplicateTensor();
672 }
673 
674 template<> inline const tt::target::ttnn::ShardTensor *DistributionStrategy::strategy_as<tt::target::ttnn::ShardTensor>() const {
675  return strategy_as_ShardTensor();
676 }
677 
678 template<> inline const tt::target::ttnn::ShardTensor2D *DistributionStrategy::strategy_as<tt::target::ttnn::ShardTensor2D>() const {
679  return strategy_as_ShardTensor2D();
680 }
681 
682 template<> inline const tt::target::ttnn::AllGatherTensor *DistributionStrategy::strategy_as<tt::target::ttnn::AllGatherTensor>() const {
683  return strategy_as_AllGatherTensor();
684 }
685 
687  typedef DistributionStrategy Table;
688  ::flatbuffers::FlatBufferBuilder &fbb_;
689  ::flatbuffers::uoffset_t start_;
691  fbb_.AddElement<uint8_t>(DistributionStrategy::VT_STRATEGY_TYPE, static_cast<uint8_t>(strategy_type), 0);
692  }
693  void add_strategy(::flatbuffers::Offset<void> strategy) {
694  fbb_.AddOffset(DistributionStrategy::VT_STRATEGY, strategy);
695  }
696  explicit DistributionStrategyBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
697  : fbb_(_fbb) {
698  start_ = fbb_.StartTable();
699  }
700  ::flatbuffers::Offset<DistributionStrategy> Finish() {
701  const auto end = fbb_.EndTable(start_);
702  auto o = ::flatbuffers::Offset<DistributionStrategy>(end);
703  return o;
704  }
705 };
706 
707 inline ::flatbuffers::Offset<DistributionStrategy> CreateDistributionStrategy(
708  ::flatbuffers::FlatBufferBuilder &_fbb,
710  ::flatbuffers::Offset<void> strategy = 0) {
711  DistributionStrategyBuilder builder_(_fbb);
712  builder_.add_strategy(strategy);
713  builder_.add_strategy_type(strategy_type);
714  return builder_.Finish();
715 }
716 
718  using type = DistributionStrategy;
719  static auto constexpr Create = CreateDistributionStrategy;
720 };
721 
722 struct ShardSpec FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
724  struct Traits;
725  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
726  return "tt.target.ttnn.ShardSpec";
727  }
728  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
730  VT_SHAPE = 6,
732  VT_MODE = 10,
733  VT_PHYSICAL_SHARD_SHAPE = 12
734  };
735  const tt::target::ttnn::CoreRangeSet *core_range_set() const {
736  return GetPointer<const tt::target::ttnn::CoreRangeSet *>(VT_CORE_RANGE_SET);
737  }
738  const ::flatbuffers::Vector<int32_t> *shape() const {
739  return GetPointer<const ::flatbuffers::Vector<int32_t> *>(VT_SHAPE);
740  }
742  return static_cast<tt::target::ttnn::ShardOrientation>(GetField<uint32_t>(VT_ORIENTATION, 0));
743  }
745  return static_cast<tt::target::ttnn::ShardMode>(GetField<uint32_t>(VT_MODE, 0));
746  }
747  const ::flatbuffers::Vector<int32_t> *physical_shard_shape() const {
748  return GetPointer<const ::flatbuffers::Vector<int32_t> *>(VT_PHYSICAL_SHARD_SHAPE);
749  }
750  bool Verify(::flatbuffers::Verifier &verifier) const {
751  return VerifyTableStart(verifier) &&
752  VerifyOffset(verifier, VT_CORE_RANGE_SET) &&
753  verifier.VerifyTable(core_range_set()) &&
754  VerifyOffset(verifier, VT_SHAPE) &&
755  verifier.VerifyVector(shape()) &&
756  VerifyField<uint32_t>(verifier, VT_ORIENTATION, 4) &&
757  VerifyField<uint32_t>(verifier, VT_MODE, 4) &&
758  VerifyOffset(verifier, VT_PHYSICAL_SHARD_SHAPE) &&
759  verifier.VerifyVector(physical_shard_shape()) &&
760  verifier.EndTable();
761  }
762 };
763 
765  typedef ShardSpec Table;
766  ::flatbuffers::FlatBufferBuilder &fbb_;
767  ::flatbuffers::uoffset_t start_;
768  void add_core_range_set(::flatbuffers::Offset<tt::target::ttnn::CoreRangeSet> core_range_set) {
769  fbb_.AddOffset(ShardSpec::VT_CORE_RANGE_SET, core_range_set);
770  }
771  void add_shape(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> shape) {
772  fbb_.AddOffset(ShardSpec::VT_SHAPE, shape);
773  }
775  fbb_.AddElement<uint32_t>(ShardSpec::VT_ORIENTATION, static_cast<uint32_t>(orientation), 0);
776  }
778  fbb_.AddElement<uint32_t>(ShardSpec::VT_MODE, static_cast<uint32_t>(mode), 0);
779  }
780  void add_physical_shard_shape(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> physical_shard_shape) {
781  fbb_.AddOffset(ShardSpec::VT_PHYSICAL_SHARD_SHAPE, physical_shard_shape);
782  }
783  explicit ShardSpecBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
784  : fbb_(_fbb) {
785  start_ = fbb_.StartTable();
786  }
787  ::flatbuffers::Offset<ShardSpec> Finish() {
788  const auto end = fbb_.EndTable(start_);
789  auto o = ::flatbuffers::Offset<ShardSpec>(end);
790  return o;
791  }
792 };
793 
794 inline ::flatbuffers::Offset<ShardSpec> CreateShardSpec(
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) {
801  ShardSpecBuilder builder_(_fbb);
802  builder_.add_physical_shard_shape(physical_shard_shape);
803  builder_.add_mode(mode);
804  builder_.add_orientation(orientation);
805  builder_.add_shape(shape);
806  builder_.add_core_range_set(core_range_set);
807  return builder_.Finish();
808 }
809 
811  using type = ShardSpec;
812  static auto constexpr Create = CreateShardSpec;
813 };
814 
815 inline ::flatbuffers::Offset<ShardSpec> CreateShardSpecDirect(
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;
825  _fbb,
826  core_range_set,
827  shape__,
828  orientation,
829  mode,
830  physical_shard_shape__);
831 }
832 
833 struct MemoryConfig FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
835  struct Traits;
836  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
837  return "tt.target.ttnn.MemoryConfig";
838  }
839  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
842  VT_SHARD_SPEC = 8
843  };
845  return static_cast<tt::target::ttnn::TensorMemoryLayout>(GetField<uint16_t>(VT_TENSOR_MEMORY_LAYOUT, 0));
846  }
848  return static_cast<tt::target::BufferType>(GetField<uint16_t>(VT_BUFFER_TYPE, 0));
849  }
850  const tt::target::ttnn::ShardSpec *shard_spec() const {
851  return GetPointer<const tt::target::ttnn::ShardSpec *>(VT_SHARD_SPEC);
852  }
853  bool Verify(::flatbuffers::Verifier &verifier) const {
854  return VerifyTableStart(verifier) &&
855  VerifyField<uint16_t>(verifier, VT_TENSOR_MEMORY_LAYOUT, 2) &&
856  VerifyField<uint16_t>(verifier, VT_BUFFER_TYPE, 2) &&
857  VerifyOffset(verifier, VT_SHARD_SPEC) &&
858  verifier.VerifyTable(shard_spec()) &&
859  verifier.EndTable();
860  }
861 };
862 
864  typedef MemoryConfig Table;
865  ::flatbuffers::FlatBufferBuilder &fbb_;
866  ::flatbuffers::uoffset_t start_;
868  fbb_.AddElement<uint16_t>(MemoryConfig::VT_TENSOR_MEMORY_LAYOUT, static_cast<uint16_t>(tensor_memory_layout), 0);
869  }
871  fbb_.AddElement<uint16_t>(MemoryConfig::VT_BUFFER_TYPE, static_cast<uint16_t>(buffer_type), 0);
872  }
873  void add_shard_spec(::flatbuffers::Offset<tt::target::ttnn::ShardSpec> shard_spec) {
874  fbb_.AddOffset(MemoryConfig::VT_SHARD_SPEC, shard_spec);
875  }
876  explicit MemoryConfigBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
877  : fbb_(_fbb) {
878  start_ = fbb_.StartTable();
879  }
880  ::flatbuffers::Offset<MemoryConfig> Finish() {
881  const auto end = fbb_.EndTable(start_);
882  auto o = ::flatbuffers::Offset<MemoryConfig>(end);
883  return o;
884  }
885 };
886 
887 inline ::flatbuffers::Offset<MemoryConfig> CreateMemoryConfig(
888  ::flatbuffers::FlatBufferBuilder &_fbb,
891  ::flatbuffers::Offset<tt::target::ttnn::ShardSpec> shard_spec = 0) {
892  MemoryConfigBuilder builder_(_fbb);
893  builder_.add_shard_spec(shard_spec);
894  builder_.add_buffer_type(buffer_type);
895  builder_.add_tensor_memory_layout(tensor_memory_layout);
896  return builder_.Finish();
897 }
898 
900  using type = MemoryConfig;
901  static auto constexpr Create = CreateMemoryConfig;
902 };
903 
904 struct MemoryDesc FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
906  struct Traits;
907  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
908  return "tt.target.ttnn.MemoryDesc";
909  }
910  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
915  VT_SIZE = 12
916  };
918  return static_cast<tt::target::ttnn::StorageType>(GetField<uint16_t>(VT_STORAGE_TYPE, 0));
919  }
920  const tt::target::Dim2d *tile_shape() const {
921  return GetStruct<const tt::target::Dim2d *>(VT_TILE_SHAPE);
922  }
924  return static_cast<tt::target::DataType>(GetField<uint16_t>(VT_DATA_TYPE, 0));
925  }
926  const tt::target::ttnn::MemoryConfig *memory_config() const {
927  return GetPointer<const tt::target::ttnn::MemoryConfig *>(VT_MEMORY_CONFIG);
928  }
929  uint64_t size() const {
930  return GetField<uint64_t>(VT_SIZE, 0);
931  }
932  bool Verify(::flatbuffers::Verifier &verifier) const {
933  return VerifyTableStart(verifier) &&
934  VerifyField<uint16_t>(verifier, VT_STORAGE_TYPE, 2) &&
935  VerifyField<tt::target::Dim2d>(verifier, VT_TILE_SHAPE, 4) &&
936  VerifyField<uint16_t>(verifier, VT_DATA_TYPE, 2) &&
937  VerifyOffset(verifier, VT_MEMORY_CONFIG) &&
938  verifier.VerifyTable(memory_config()) &&
939  VerifyField<uint64_t>(verifier, VT_SIZE, 8) &&
940  verifier.EndTable();
941  }
942 };
943 
945  typedef MemoryDesc Table;
946  ::flatbuffers::FlatBufferBuilder &fbb_;
947  ::flatbuffers::uoffset_t start_;
949  fbb_.AddElement<uint16_t>(MemoryDesc::VT_STORAGE_TYPE, static_cast<uint16_t>(storage_type), 0);
950  }
951  void add_tile_shape(const tt::target::Dim2d *tile_shape) {
952  fbb_.AddStruct(MemoryDesc::VT_TILE_SHAPE, tile_shape);
953  }
955  fbb_.AddElement<uint16_t>(MemoryDesc::VT_DATA_TYPE, static_cast<uint16_t>(data_type), 0);
956  }
957  void add_memory_config(::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> memory_config) {
958  fbb_.AddOffset(MemoryDesc::VT_MEMORY_CONFIG, memory_config);
959  }
960  void add_size(uint64_t size) {
961  fbb_.AddElement<uint64_t>(MemoryDesc::VT_SIZE, size, 0);
962  }
963  explicit MemoryDescBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
964  : fbb_(_fbb) {
965  start_ = fbb_.StartTable();
966  }
967  ::flatbuffers::Offset<MemoryDesc> Finish() {
968  const auto end = fbb_.EndTable(start_);
969  auto o = ::flatbuffers::Offset<MemoryDesc>(end);
970  return o;
971  }
972 };
973 
974 inline ::flatbuffers::Offset<MemoryDesc> CreateMemoryDesc(
975  ::flatbuffers::FlatBufferBuilder &_fbb,
977  const tt::target::Dim2d *tile_shape = nullptr,
979  ::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> memory_config = 0,
980  uint64_t size = 0) {
981  MemoryDescBuilder builder_(_fbb);
982  builder_.add_size(size);
983  builder_.add_memory_config(memory_config);
984  builder_.add_tile_shape(tile_shape);
985  builder_.add_data_type(data_type);
986  builder_.add_storage_type(storage_type);
987  return builder_.Finish();
988 }
989 
991  using type = MemoryDesc;
992  static auto constexpr Create = CreateMemoryDesc;
993 };
994 
995 struct LayoutDesc FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
997  struct Traits;
998  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
999  return "tt.target.ttnn.LayoutDesc";
1000  }
1001  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1004  VT_STRATEGY = 8
1005  };
1007  return static_cast<tt::target::OOBVal>(GetField<uint16_t>(VT_OOB_VAL, 0));
1008  }
1009  const tt::target::ttnn::MemoryDesc *memory_desc() const {
1010  return GetPointer<const tt::target::ttnn::MemoryDesc *>(VT_MEMORY_DESC);
1011  }
1012  const tt::target::ttnn::DistributionStrategy *strategy() const {
1013  return GetPointer<const tt::target::ttnn::DistributionStrategy *>(VT_STRATEGY);
1014  }
1015  bool Verify(::flatbuffers::Verifier &verifier) const {
1016  return VerifyTableStart(verifier) &&
1017  VerifyField<uint16_t>(verifier, VT_OOB_VAL, 2) &&
1018  VerifyOffset(verifier, VT_MEMORY_DESC) &&
1019  verifier.VerifyTable(memory_desc()) &&
1020  VerifyOffset(verifier, VT_STRATEGY) &&
1021  verifier.VerifyTable(strategy()) &&
1022  verifier.EndTable();
1023  }
1024 };
1025 
1027  typedef LayoutDesc Table;
1028  ::flatbuffers::FlatBufferBuilder &fbb_;
1029  ::flatbuffers::uoffset_t start_;
1031  fbb_.AddElement<uint16_t>(LayoutDesc::VT_OOB_VAL, static_cast<uint16_t>(oob_val), 0);
1032  }
1033  void add_memory_desc(::flatbuffers::Offset<tt::target::ttnn::MemoryDesc> memory_desc) {
1034  fbb_.AddOffset(LayoutDesc::VT_MEMORY_DESC, memory_desc);
1035  }
1036  void add_strategy(::flatbuffers::Offset<tt::target::ttnn::DistributionStrategy> strategy) {
1037  fbb_.AddOffset(LayoutDesc::VT_STRATEGY, strategy);
1038  }
1039  explicit LayoutDescBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
1040  : fbb_(_fbb) {
1041  start_ = fbb_.StartTable();
1042  }
1043  ::flatbuffers::Offset<LayoutDesc> Finish() {
1044  const auto end = fbb_.EndTable(start_);
1045  auto o = ::flatbuffers::Offset<LayoutDesc>(end);
1046  return o;
1047  }
1048 };
1049 
1050 inline ::flatbuffers::Offset<LayoutDesc> CreateLayoutDesc(
1051  ::flatbuffers::FlatBufferBuilder &_fbb,
1053  ::flatbuffers::Offset<tt::target::ttnn::MemoryDesc> memory_desc = 0,
1054  ::flatbuffers::Offset<tt::target::ttnn::DistributionStrategy> strategy = 0) {
1055  LayoutDescBuilder builder_(_fbb);
1056  builder_.add_strategy(strategy);
1057  builder_.add_memory_desc(memory_desc);
1058  builder_.add_oob_val(oob_val);
1059  return builder_.Finish();
1060 }
1061 
1063  using type = LayoutDesc;
1064  static auto constexpr Create = CreateLayoutDesc;
1065 };
1066 
1067 struct TensorDesc FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
1069  struct Traits;
1070  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
1071  return "tt.target.ttnn.TensorDesc";
1072  }
1073  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1074  VT_SHAPE = 4,
1076  VT_LAYOUT = 8
1077  };
1078  const ::flatbuffers::Vector<int32_t> *shape() const {
1079  return GetPointer<const ::flatbuffers::Vector<int32_t> *>(VT_SHAPE);
1080  }
1081  const ::flatbuffers::Vector<int32_t> *mesh_shape() const {
1082  return GetPointer<const ::flatbuffers::Vector<int32_t> *>(VT_MESH_SHAPE);
1083  }
1084  const tt::target::ttnn::LayoutDesc *layout() const {
1085  return GetPointer<const tt::target::ttnn::LayoutDesc *>(VT_LAYOUT);
1086  }
1087  bool Verify(::flatbuffers::Verifier &verifier) const {
1088  return VerifyTableStart(verifier) &&
1089  VerifyOffset(verifier, VT_SHAPE) &&
1090  verifier.VerifyVector(shape()) &&
1091  VerifyOffset(verifier, VT_MESH_SHAPE) &&
1092  verifier.VerifyVector(mesh_shape()) &&
1093  VerifyOffset(verifier, VT_LAYOUT) &&
1094  verifier.VerifyTable(layout()) &&
1095  verifier.EndTable();
1096  }
1097 };
1098 
1100  typedef TensorDesc Table;
1101  ::flatbuffers::FlatBufferBuilder &fbb_;
1102  ::flatbuffers::uoffset_t start_;
1103  void add_shape(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> shape) {
1104  fbb_.AddOffset(TensorDesc::VT_SHAPE, shape);
1105  }
1106  void add_mesh_shape(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> mesh_shape) {
1107  fbb_.AddOffset(TensorDesc::VT_MESH_SHAPE, mesh_shape);
1108  }
1109  void add_layout(::flatbuffers::Offset<tt::target::ttnn::LayoutDesc> layout) {
1110  fbb_.AddOffset(TensorDesc::VT_LAYOUT, layout);
1111  }
1112  explicit TensorDescBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
1113  : fbb_(_fbb) {
1114  start_ = fbb_.StartTable();
1115  }
1116  ::flatbuffers::Offset<TensorDesc> Finish() {
1117  const auto end = fbb_.EndTable(start_);
1118  auto o = ::flatbuffers::Offset<TensorDesc>(end);
1119  return o;
1120  }
1121 };
1122 
1123 inline ::flatbuffers::Offset<TensorDesc> CreateTensorDesc(
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) {
1128  TensorDescBuilder builder_(_fbb);
1129  builder_.add_layout(layout);
1130  builder_.add_mesh_shape(mesh_shape);
1131  builder_.add_shape(shape);
1132  return builder_.Finish();
1133 }
1134 
1136  using type = TensorDesc;
1137  static auto constexpr Create = CreateTensorDesc;
1138 };
1139 
1140 inline ::flatbuffers::Offset<TensorDesc> CreateTensorDescDirect(
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;
1148  _fbb,
1149  shape__,
1150  mesh_shape__,
1151  layout);
1152 }
1153 
1154 struct TensorRef FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
1156  struct Traits;
1157  static FLATBUFFERS_CONSTEXPR_CPP11 const char *GetFullyQualifiedName() {
1158  return "tt.target.ttnn.TensorRef";
1159  }
1160  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1162  VT_SIZE = 6,
1163  VT_DESC = 8
1164  };
1165  uint32_t global_id() const {
1166  return GetField<uint32_t>(VT_GLOBAL_ID, 0);
1167  }
1168  uint64_t size() const {
1169  return GetField<uint64_t>(VT_SIZE, 0);
1170  }
1171  const tt::target::ttnn::TensorDesc *desc() const {
1172  return GetPointer<const tt::target::ttnn::TensorDesc *>(VT_DESC);
1173  }
1174  bool Verify(::flatbuffers::Verifier &verifier) const {
1175  return VerifyTableStart(verifier) &&
1176  VerifyField<uint32_t>(verifier, VT_GLOBAL_ID, 4) &&
1177  VerifyField<uint64_t>(verifier, VT_SIZE, 8) &&
1178  VerifyOffset(verifier, VT_DESC) &&
1179  verifier.VerifyTable(desc()) &&
1180  verifier.EndTable();
1181  }
1182 };
1183 
1185  typedef TensorRef Table;
1186  ::flatbuffers::FlatBufferBuilder &fbb_;
1187  ::flatbuffers::uoffset_t start_;
1188  void add_global_id(uint32_t global_id) {
1189  fbb_.AddElement<uint32_t>(TensorRef::VT_GLOBAL_ID, global_id, 0);
1190  }
1191  void add_size(uint64_t size) {
1192  fbb_.AddElement<uint64_t>(TensorRef::VT_SIZE, size, 0);
1193  }
1194  void add_desc(::flatbuffers::Offset<tt::target::ttnn::TensorDesc> desc) {
1195  fbb_.AddOffset(TensorRef::VT_DESC, desc);
1196  }
1197  explicit TensorRefBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
1198  : fbb_(_fbb) {
1199  start_ = fbb_.StartTable();
1200  }
1201  ::flatbuffers::Offset<TensorRef> Finish() {
1202  const auto end = fbb_.EndTable(start_);
1203  auto o = ::flatbuffers::Offset<TensorRef>(end);
1204  return o;
1205  }
1206 };
1207 
1208 inline ::flatbuffers::Offset<TensorRef> CreateTensorRef(
1209  ::flatbuffers::FlatBufferBuilder &_fbb,
1210  uint32_t global_id = 0,
1211  uint64_t size = 0,
1212  ::flatbuffers::Offset<tt::target::ttnn::TensorDesc> desc = 0) {
1213  TensorRefBuilder builder_(_fbb);
1214  builder_.add_size(size);
1215  builder_.add_desc(desc);
1216  builder_.add_global_id(global_id);
1217  return builder_.Finish();
1218 }
1219 
1221  using type = TensorRef;
1222  static auto constexpr Create = CreateTensorRef;
1223 };
1224 
1225 inline bool VerifyDistributedTensorConfig(::flatbuffers::Verifier &verifier, const void *obj, DistributedTensorConfig type) {
1226  switch (type) {
1228  return true;
1229  }
1231  auto ptr = reinterpret_cast<const tt::target::ttnn::ReplicateTensor *>(obj);
1232  return verifier.VerifyTable(ptr);
1233  }
1235  auto ptr = reinterpret_cast<const tt::target::ttnn::ShardTensor *>(obj);
1236  return verifier.VerifyTable(ptr);
1237  }
1239  auto ptr = reinterpret_cast<const tt::target::ttnn::ShardTensor2D *>(obj);
1240  return verifier.VerifyTable(ptr);
1241  }
1243  auto ptr = reinterpret_cast<const tt::target::ttnn::AllGatherTensor *>(obj);
1244  return verifier.VerifyTable(ptr);
1245  }
1246  default: return true;
1247  }
1248 }
1249 
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) {
1255  verifier, values->Get(i), types->GetEnum<DistributedTensorConfig>(i))) {
1256  return false;
1257  }
1258  }
1259  return true;
1260 }
1261 
1262 } // namespace ttnn
1263 } // namespace target
1264 } // namespace tt
1265 
1266 #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: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 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