TT-MLIR
creation_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_CREATION_TT_TARGET_TTNN_H_
5 #define FLATBUFFERS_GENERATED_CREATION_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 
18 
19 namespace tt {
20 namespace target {
21 namespace ttnn {
22 
23 struct ArangeOp;
24 struct ArangeOpBuilder;
25 
26 struct ConstantOp;
27 struct ConstantOpBuilder;
28 
29 struct ConstructTensorOp;
30 struct ConstructTensorOpBuilder;
31 
32 struct EmptyOp;
33 struct EmptyOpBuilder;
34 
35 struct FullOp;
36 struct FullOpBuilder;
37 
38 struct NamedFullOp;
39 struct NamedFullOpBuilder;
40 
41 enum class NamedFullOpType : uint32_t {
42  Zeros = 0,
43  Ones = 1,
44  MIN = Zeros,
45  MAX = Ones
46 };
47 
49  static const NamedFullOpType values[] = {
52  };
53  return values;
54 }
55 
56 inline const char * const *EnumNamesNamedFullOpType() {
57  static const char * const names[3] = {
58  "Zeros",
59  "Ones",
60  nullptr
61  };
62  return names;
63 }
64 
66  if (::flatbuffers::IsOutRange(e, NamedFullOpType::Zeros, NamedFullOpType::Ones)) return "";
67  const size_t index = static_cast<size_t>(e);
68  return EnumNamesNamedFullOpType()[index];
69 }
70 
71 struct ArangeOp FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
73  struct Traits;
74  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
75  VT_START = 4,
76  VT_END = 6,
77  VT_STEP = 8,
78  VT_DTYPE = 10,
79  VT_DEVICE = 12,
80  VT_MEMCFG = 14,
81  VT_OUT = 16
82  };
83  float start() const {
84  return GetField<float>(VT_START, 0.0f);
85  }
86  float end() const {
87  return GetField<float>(VT_END, 0.0f);
88  }
89  float step() const {
90  return GetField<float>(VT_STEP, 0.0f);
91  }
92  ::flatbuffers::Optional<tt::target::DataType> dtype() const {
93  return GetOptional<uint16_t, tt::target::DataType>(VT_DTYPE);
94  }
95  const tt::target::DeviceRef *device() const {
96  return GetPointer<const tt::target::DeviceRef *>(VT_DEVICE);
97  }
98  const tt::target::ttnn::MemoryConfig *memcfg() const {
99  return GetPointer<const tt::target::ttnn::MemoryConfig *>(VT_MEMCFG);
100  }
101  const tt::target::ttnn::TensorRef *out() const {
102  return GetPointer<const tt::target::ttnn::TensorRef *>(VT_OUT);
103  }
104  bool Verify(::flatbuffers::Verifier &verifier) const {
105  return VerifyTableStart(verifier) &&
106  VerifyField<float>(verifier, VT_START, 4) &&
107  VerifyField<float>(verifier, VT_END, 4) &&
108  VerifyField<float>(verifier, VT_STEP, 4) &&
109  VerifyField<uint16_t>(verifier, VT_DTYPE, 2) &&
110  VerifyOffset(verifier, VT_DEVICE) &&
111  verifier.VerifyTable(device()) &&
112  VerifyOffset(verifier, VT_MEMCFG) &&
113  verifier.VerifyTable(memcfg()) &&
114  VerifyOffset(verifier, VT_OUT) &&
115  verifier.VerifyTable(out()) &&
116  verifier.EndTable();
117  }
118 };
119 
121  typedef ArangeOp Table;
122  ::flatbuffers::FlatBufferBuilder &fbb_;
123  ::flatbuffers::uoffset_t start_;
124  void add_start(float start) {
125  fbb_.AddElement<float>(ArangeOp::VT_START, start, 0.0f);
126  }
127  void add_end(float end) {
128  fbb_.AddElement<float>(ArangeOp::VT_END, end, 0.0f);
129  }
130  void add_step(float step) {
131  fbb_.AddElement<float>(ArangeOp::VT_STEP, step, 0.0f);
132  }
134  fbb_.AddElement<uint16_t>(ArangeOp::VT_DTYPE, static_cast<uint16_t>(dtype));
135  }
136  void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
137  fbb_.AddOffset(ArangeOp::VT_DEVICE, device);
138  }
139  void add_memcfg(::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> memcfg) {
140  fbb_.AddOffset(ArangeOp::VT_MEMCFG, memcfg);
141  }
142  void add_out(::flatbuffers::Offset<tt::target::ttnn::TensorRef> out) {
143  fbb_.AddOffset(ArangeOp::VT_OUT, out);
144  }
145  explicit ArangeOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
146  : fbb_(_fbb) {
147  start_ = fbb_.StartTable();
148  }
149  ::flatbuffers::Offset<ArangeOp> Finish() {
150  const auto end = fbb_.EndTable(start_);
151  auto o = ::flatbuffers::Offset<ArangeOp>(end);
152  return o;
153  }
154 };
155 
156 inline ::flatbuffers::Offset<ArangeOp> CreateArangeOp(
157  ::flatbuffers::FlatBufferBuilder &_fbb,
158  float start = 0.0f,
159  float end = 0.0f,
160  float step = 0.0f,
161  ::flatbuffers::Optional<tt::target::DataType> dtype = ::flatbuffers::nullopt,
162  ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
163  ::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> memcfg = 0,
164  ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0) {
165  ArangeOpBuilder builder_(_fbb);
166  builder_.add_out(out);
167  builder_.add_memcfg(memcfg);
168  builder_.add_device(device);
169  builder_.add_step(step);
170  builder_.add_end(end);
171  builder_.add_start(start);
172  if(dtype) { builder_.add_dtype(*dtype); }
173  return builder_.Finish();
174 }
175 
177  using type = ArangeOp;
178  static auto constexpr Create = CreateArangeOp;
179 };
180 
181 struct ConstantOp FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
183  struct Traits;
184  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
185  VT_OUT = 4,
186  VT_DATA = 6
187  };
188  const tt::target::ttnn::TensorRef *out() const {
189  return GetPointer<const tt::target::ttnn::TensorRef *>(VT_OUT);
190  }
191  const ::flatbuffers::Vector<uint8_t> *data() const {
192  return GetPointer<const ::flatbuffers::Vector<uint8_t> *>(VT_DATA);
193  }
194  bool Verify(::flatbuffers::Verifier &verifier) const {
195  return VerifyTableStart(verifier) &&
196  VerifyOffset(verifier, VT_OUT) &&
197  verifier.VerifyTable(out()) &&
198  VerifyOffset(verifier, VT_DATA) &&
199  verifier.VerifyVector(data()) &&
200  verifier.EndTable();
201  }
202 };
203 
205  typedef ConstantOp Table;
206  ::flatbuffers::FlatBufferBuilder &fbb_;
207  ::flatbuffers::uoffset_t start_;
208  void add_out(::flatbuffers::Offset<tt::target::ttnn::TensorRef> out) {
209  fbb_.AddOffset(ConstantOp::VT_OUT, out);
210  }
211  void add_data(::flatbuffers::Offset<::flatbuffers::Vector<uint8_t>> data) {
212  fbb_.AddOffset(ConstantOp::VT_DATA, data);
213  }
214  explicit ConstantOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
215  : fbb_(_fbb) {
216  start_ = fbb_.StartTable();
217  }
218  ::flatbuffers::Offset<ConstantOp> Finish() {
219  const auto end = fbb_.EndTable(start_);
220  auto o = ::flatbuffers::Offset<ConstantOp>(end);
221  return o;
222  }
223 };
224 
225 inline ::flatbuffers::Offset<ConstantOp> CreateConstantOp(
226  ::flatbuffers::FlatBufferBuilder &_fbb,
227  ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
228  ::flatbuffers::Offset<::flatbuffers::Vector<uint8_t>> data = 0) {
229  ConstantOpBuilder builder_(_fbb);
230  builder_.add_data(data);
231  builder_.add_out(out);
232  return builder_.Finish();
233 }
234 
236  using type = ConstantOp;
237  static auto constexpr Create = CreateConstantOp;
238 };
239 
240 inline ::flatbuffers::Offset<ConstantOp> CreateConstantOpDirect(
241  ::flatbuffers::FlatBufferBuilder &_fbb,
242  ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
243  const std::vector<uint8_t> *data = nullptr) {
244  auto data__ = data ? _fbb.CreateVector<uint8_t>(*data) : 0;
246  _fbb,
247  out,
248  data__);
249 }
250 
251 struct ConstructTensorOp FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
253  struct Traits;
254  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
255  VT_SHAPE = 4,
256  VT_DTYPE = 6,
258  VT_OUT = 10
259  };
260  const ::flatbuffers::Vector<int64_t> *shape() const {
261  return GetPointer<const ::flatbuffers::Vector<int64_t> *>(VT_SHAPE);
262  }
264  return static_cast<tt::target::DataType>(GetField<uint16_t>(VT_DTYPE, 0));
265  }
267  return static_cast<tt::target::TensorLayout>(GetField<uint16_t>(VT_LAYOUT, 0));
268  }
269  const tt::target::ttnn::TensorRef *out() const {
270  return GetPointer<const tt::target::ttnn::TensorRef *>(VT_OUT);
271  }
272  bool Verify(::flatbuffers::Verifier &verifier) const {
273  return VerifyTableStart(verifier) &&
274  VerifyOffset(verifier, VT_SHAPE) &&
275  verifier.VerifyVector(shape()) &&
276  VerifyField<uint16_t>(verifier, VT_DTYPE, 2) &&
277  VerifyField<uint16_t>(verifier, VT_LAYOUT, 2) &&
278  VerifyOffset(verifier, VT_OUT) &&
279  verifier.VerifyTable(out()) &&
280  verifier.EndTable();
281  }
282 };
283 
286  ::flatbuffers::FlatBufferBuilder &fbb_;
287  ::flatbuffers::uoffset_t start_;
288  void add_shape(::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> shape) {
289  fbb_.AddOffset(ConstructTensorOp::VT_SHAPE, shape);
290  }
292  fbb_.AddElement<uint16_t>(ConstructTensorOp::VT_DTYPE, static_cast<uint16_t>(dtype), 0);
293  }
295  fbb_.AddElement<uint16_t>(ConstructTensorOp::VT_LAYOUT, static_cast<uint16_t>(layout), 0);
296  }
297  void add_out(::flatbuffers::Offset<tt::target::ttnn::TensorRef> out) {
298  fbb_.AddOffset(ConstructTensorOp::VT_OUT, out);
299  }
300  explicit ConstructTensorOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
301  : fbb_(_fbb) {
302  start_ = fbb_.StartTable();
303  }
304  ::flatbuffers::Offset<ConstructTensorOp> Finish() {
305  const auto end = fbb_.EndTable(start_);
306  auto o = ::flatbuffers::Offset<ConstructTensorOp>(end);
307  return o;
308  }
309 };
310 
311 inline ::flatbuffers::Offset<ConstructTensorOp> CreateConstructTensorOp(
312  ::flatbuffers::FlatBufferBuilder &_fbb,
313  ::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> shape = 0,
316  ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0) {
317  ConstructTensorOpBuilder builder_(_fbb);
318  builder_.add_out(out);
319  builder_.add_shape(shape);
320  builder_.add_layout(layout);
321  builder_.add_dtype(dtype);
322  return builder_.Finish();
323 }
324 
327  static auto constexpr Create = CreateConstructTensorOp;
328 };
329 
330 inline ::flatbuffers::Offset<ConstructTensorOp> CreateConstructTensorOpDirect(
331  ::flatbuffers::FlatBufferBuilder &_fbb,
332  const std::vector<int64_t> *shape = nullptr,
335  ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0) {
336  auto shape__ = shape ? _fbb.CreateVector<int64_t>(*shape) : 0;
338  _fbb,
339  shape__,
340  dtype,
341  layout,
342  out);
343 }
344 
345 struct EmptyOp FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
347  struct Traits;
348  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
349  VT_SHAPE = 4,
350  VT_DTYPE = 6,
351  VT_LAYOUT = 8,
353  VT_DEVICE = 12,
354  VT_MEMCFG = 14,
356  VT_OUT = 18
357  };
358  const ::flatbuffers::Vector<int64_t> *shape() const {
359  return GetPointer<const ::flatbuffers::Vector<int64_t> *>(VT_SHAPE);
360  }
362  return static_cast<tt::target::DataType>(GetField<uint16_t>(VT_DTYPE, 0));
363  }
365  return static_cast<tt::target::TensorLayout>(GetField<uint16_t>(VT_LAYOUT, 0));
366  }
367  uint32_t num_shards() const {
368  return GetField<uint32_t>(VT_NUM_SHARDS, 0);
369  }
370  const tt::target::DeviceRef *device() const {
371  return GetPointer<const tt::target::DeviceRef *>(VT_DEVICE);
372  }
373  const tt::target::ttnn::MemoryConfig *memcfg() const {
374  return GetPointer<const tt::target::ttnn::MemoryConfig *>(VT_MEMCFG);
375  }
376  const tt::target::ttnn::DistributionStrategy *strategy() const {
377  return GetPointer<const tt::target::ttnn::DistributionStrategy *>(VT_STRATEGY);
378  }
379  const tt::target::ttnn::TensorRef *out() const {
380  return GetPointer<const tt::target::ttnn::TensorRef *>(VT_OUT);
381  }
382  bool Verify(::flatbuffers::Verifier &verifier) const {
383  return VerifyTableStart(verifier) &&
384  VerifyOffset(verifier, VT_SHAPE) &&
385  verifier.VerifyVector(shape()) &&
386  VerifyField<uint16_t>(verifier, VT_DTYPE, 2) &&
387  VerifyField<uint16_t>(verifier, VT_LAYOUT, 2) &&
388  VerifyField<uint32_t>(verifier, VT_NUM_SHARDS, 4) &&
389  VerifyOffset(verifier, VT_DEVICE) &&
390  verifier.VerifyTable(device()) &&
391  VerifyOffset(verifier, VT_MEMCFG) &&
392  verifier.VerifyTable(memcfg()) &&
393  VerifyOffset(verifier, VT_STRATEGY) &&
394  verifier.VerifyTable(strategy()) &&
395  VerifyOffset(verifier, VT_OUT) &&
396  verifier.VerifyTable(out()) &&
397  verifier.EndTable();
398  }
399 };
400 
402  typedef EmptyOp Table;
403  ::flatbuffers::FlatBufferBuilder &fbb_;
404  ::flatbuffers::uoffset_t start_;
405  void add_shape(::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> shape) {
406  fbb_.AddOffset(EmptyOp::VT_SHAPE, shape);
407  }
409  fbb_.AddElement<uint16_t>(EmptyOp::VT_DTYPE, static_cast<uint16_t>(dtype), 0);
410  }
412  fbb_.AddElement<uint16_t>(EmptyOp::VT_LAYOUT, static_cast<uint16_t>(layout), 0);
413  }
414  void add_num_shards(uint32_t num_shards) {
415  fbb_.AddElement<uint32_t>(EmptyOp::VT_NUM_SHARDS, num_shards, 0);
416  }
417  void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
418  fbb_.AddOffset(EmptyOp::VT_DEVICE, device);
419  }
420  void add_memcfg(::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> memcfg) {
421  fbb_.AddOffset(EmptyOp::VT_MEMCFG, memcfg);
422  }
423  void add_strategy(::flatbuffers::Offset<tt::target::ttnn::DistributionStrategy> strategy) {
424  fbb_.AddOffset(EmptyOp::VT_STRATEGY, strategy);
425  }
426  void add_out(::flatbuffers::Offset<tt::target::ttnn::TensorRef> out) {
427  fbb_.AddOffset(EmptyOp::VT_OUT, out);
428  }
429  explicit EmptyOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
430  : fbb_(_fbb) {
431  start_ = fbb_.StartTable();
432  }
433  ::flatbuffers::Offset<EmptyOp> Finish() {
434  const auto end = fbb_.EndTable(start_);
435  auto o = ::flatbuffers::Offset<EmptyOp>(end);
436  return o;
437  }
438 };
439 
440 inline ::flatbuffers::Offset<EmptyOp> CreateEmptyOp(
441  ::flatbuffers::FlatBufferBuilder &_fbb,
442  ::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> shape = 0,
445  uint32_t num_shards = 0,
446  ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
447  ::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> memcfg = 0,
448  ::flatbuffers::Offset<tt::target::ttnn::DistributionStrategy> strategy = 0,
449  ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0) {
450  EmptyOpBuilder builder_(_fbb);
451  builder_.add_out(out);
452  builder_.add_strategy(strategy);
453  builder_.add_memcfg(memcfg);
454  builder_.add_device(device);
455  builder_.add_num_shards(num_shards);
456  builder_.add_shape(shape);
457  builder_.add_layout(layout);
458  builder_.add_dtype(dtype);
459  return builder_.Finish();
460 }
461 
463  using type = EmptyOp;
464  static auto constexpr Create = CreateEmptyOp;
465 };
466 
467 inline ::flatbuffers::Offset<EmptyOp> CreateEmptyOpDirect(
468  ::flatbuffers::FlatBufferBuilder &_fbb,
469  const std::vector<int64_t> *shape = nullptr,
472  uint32_t num_shards = 0,
473  ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
474  ::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> memcfg = 0,
475  ::flatbuffers::Offset<tt::target::ttnn::DistributionStrategy> strategy = 0,
476  ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0) {
477  auto shape__ = shape ? _fbb.CreateVector<int64_t>(*shape) : 0;
479  _fbb,
480  shape__,
481  dtype,
482  layout,
483  num_shards,
484  device,
485  memcfg,
486  strategy,
487  out);
488 }
489 
490 struct FullOp FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
492  struct Traits;
493  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
494  VT_DEVICE = 4,
496  VT_NUM_SHARDS = 8,
497  VT_STRATEGY = 10,
498  VT_OUT = 12
499  };
500  const tt::target::DeviceRef *device() const {
501  return GetPointer<const tt::target::DeviceRef *>(VT_DEVICE);
502  }
503  float fill_value() const {
504  return GetField<float>(VT_FILL_VALUE, 0.0f);
505  }
506  uint32_t num_shards() const {
507  return GetField<uint32_t>(VT_NUM_SHARDS, 0);
508  }
509  const tt::target::ttnn::DistributionStrategy *strategy() const {
510  return GetPointer<const tt::target::ttnn::DistributionStrategy *>(VT_STRATEGY);
511  }
512  const tt::target::ttnn::TensorRef *out() const {
513  return GetPointer<const tt::target::ttnn::TensorRef *>(VT_OUT);
514  }
515  bool Verify(::flatbuffers::Verifier &verifier) const {
516  return VerifyTableStart(verifier) &&
517  VerifyOffset(verifier, VT_DEVICE) &&
518  verifier.VerifyTable(device()) &&
519  VerifyField<float>(verifier, VT_FILL_VALUE, 4) &&
520  VerifyField<uint32_t>(verifier, VT_NUM_SHARDS, 4) &&
521  VerifyOffset(verifier, VT_STRATEGY) &&
522  verifier.VerifyTable(strategy()) &&
523  VerifyOffset(verifier, VT_OUT) &&
524  verifier.VerifyTable(out()) &&
525  verifier.EndTable();
526  }
527 };
528 
530  typedef FullOp Table;
531  ::flatbuffers::FlatBufferBuilder &fbb_;
532  ::flatbuffers::uoffset_t start_;
533  void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
534  fbb_.AddOffset(FullOp::VT_DEVICE, device);
535  }
536  void add_fill_value(float fill_value) {
537  fbb_.AddElement<float>(FullOp::VT_FILL_VALUE, fill_value, 0.0f);
538  }
539  void add_num_shards(uint32_t num_shards) {
540  fbb_.AddElement<uint32_t>(FullOp::VT_NUM_SHARDS, num_shards, 0);
541  }
542  void add_strategy(::flatbuffers::Offset<tt::target::ttnn::DistributionStrategy> strategy) {
543  fbb_.AddOffset(FullOp::VT_STRATEGY, strategy);
544  }
545  void add_out(::flatbuffers::Offset<tt::target::ttnn::TensorRef> out) {
546  fbb_.AddOffset(FullOp::VT_OUT, out);
547  }
548  explicit FullOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
549  : fbb_(_fbb) {
550  start_ = fbb_.StartTable();
551  }
552  ::flatbuffers::Offset<FullOp> Finish() {
553  const auto end = fbb_.EndTable(start_);
554  auto o = ::flatbuffers::Offset<FullOp>(end);
555  return o;
556  }
557 };
558 
559 inline ::flatbuffers::Offset<FullOp> CreateFullOp(
560  ::flatbuffers::FlatBufferBuilder &_fbb,
561  ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
562  float fill_value = 0.0f,
563  uint32_t num_shards = 0,
564  ::flatbuffers::Offset<tt::target::ttnn::DistributionStrategy> strategy = 0,
565  ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0) {
566  FullOpBuilder builder_(_fbb);
567  builder_.add_out(out);
568  builder_.add_strategy(strategy);
569  builder_.add_num_shards(num_shards);
570  builder_.add_fill_value(fill_value);
571  builder_.add_device(device);
572  return builder_.Finish();
573 }
574 
576  using type = FullOp;
577  static auto constexpr Create = CreateFullOp;
578 };
579 
580 struct NamedFullOp FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
582  struct Traits;
583  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
584  VT_TYPE = 4,
585  VT_SHAPE = 6,
586  VT_DTYPE = 8,
587  VT_LAYOUT = 10,
588  VT_DEVICE = 12,
589  VT_MEMCFG = 14,
590  VT_OUT = 16
591  };
593  return static_cast<tt::target::ttnn::NamedFullOpType>(GetField<uint32_t>(VT_TYPE, 0));
594  }
595  const ::flatbuffers::Vector<int64_t> *shape() const {
596  return GetPointer<const ::flatbuffers::Vector<int64_t> *>(VT_SHAPE);
597  }
598  ::flatbuffers::Optional<tt::target::DataType> dtype() const {
599  return GetOptional<uint16_t, tt::target::DataType>(VT_DTYPE);
600  }
601  ::flatbuffers::Optional<tt::target::TensorLayout> layout() const {
602  return GetOptional<uint16_t, tt::target::TensorLayout>(VT_LAYOUT);
603  }
604  const tt::target::DeviceRef *device() const {
605  return GetPointer<const tt::target::DeviceRef *>(VT_DEVICE);
606  }
607  const tt::target::ttnn::MemoryConfig *memcfg() const {
608  return GetPointer<const tt::target::ttnn::MemoryConfig *>(VT_MEMCFG);
609  }
610  const tt::target::ttnn::TensorRef *out() const {
611  return GetPointer<const tt::target::ttnn::TensorRef *>(VT_OUT);
612  }
613  bool Verify(::flatbuffers::Verifier &verifier) const {
614  return VerifyTableStart(verifier) &&
615  VerifyField<uint32_t>(verifier, VT_TYPE, 4) &&
616  VerifyOffset(verifier, VT_SHAPE) &&
617  verifier.VerifyVector(shape()) &&
618  VerifyField<uint16_t>(verifier, VT_DTYPE, 2) &&
619  VerifyField<uint16_t>(verifier, VT_LAYOUT, 2) &&
620  VerifyOffset(verifier, VT_DEVICE) &&
621  verifier.VerifyTable(device()) &&
622  VerifyOffset(verifier, VT_MEMCFG) &&
623  verifier.VerifyTable(memcfg()) &&
624  VerifyOffset(verifier, VT_OUT) &&
625  verifier.VerifyTable(out()) &&
626  verifier.EndTable();
627  }
628 };
629 
632  ::flatbuffers::FlatBufferBuilder &fbb_;
633  ::flatbuffers::uoffset_t start_;
635  fbb_.AddElement<uint32_t>(NamedFullOp::VT_TYPE, static_cast<uint32_t>(type), 0);
636  }
637  void add_shape(::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> shape) {
638  fbb_.AddOffset(NamedFullOp::VT_SHAPE, shape);
639  }
641  fbb_.AddElement<uint16_t>(NamedFullOp::VT_DTYPE, static_cast<uint16_t>(dtype));
642  }
644  fbb_.AddElement<uint16_t>(NamedFullOp::VT_LAYOUT, static_cast<uint16_t>(layout));
645  }
646  void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
647  fbb_.AddOffset(NamedFullOp::VT_DEVICE, device);
648  }
649  void add_memcfg(::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> memcfg) {
650  fbb_.AddOffset(NamedFullOp::VT_MEMCFG, memcfg);
651  }
652  void add_out(::flatbuffers::Offset<tt::target::ttnn::TensorRef> out) {
653  fbb_.AddOffset(NamedFullOp::VT_OUT, out);
654  }
655  explicit NamedFullOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
656  : fbb_(_fbb) {
657  start_ = fbb_.StartTable();
658  }
659  ::flatbuffers::Offset<NamedFullOp> Finish() {
660  const auto end = fbb_.EndTable(start_);
661  auto o = ::flatbuffers::Offset<NamedFullOp>(end);
662  return o;
663  }
664 };
665 
666 inline ::flatbuffers::Offset<NamedFullOp> CreateNamedFullOp(
667  ::flatbuffers::FlatBufferBuilder &_fbb,
669  ::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> shape = 0,
670  ::flatbuffers::Optional<tt::target::DataType> dtype = ::flatbuffers::nullopt,
671  ::flatbuffers::Optional<tt::target::TensorLayout> layout = ::flatbuffers::nullopt,
672  ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
673  ::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> memcfg = 0,
674  ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0) {
675  NamedFullOpBuilder builder_(_fbb);
676  builder_.add_out(out);
677  builder_.add_memcfg(memcfg);
678  builder_.add_device(device);
679  builder_.add_shape(shape);
680  builder_.add_type(type);
681  if(layout) { builder_.add_layout(*layout); }
682  if(dtype) { builder_.add_dtype(*dtype); }
683  return builder_.Finish();
684 }
685 
687  using type = NamedFullOp;
688  static auto constexpr Create = CreateNamedFullOp;
689 };
690 
691 inline ::flatbuffers::Offset<NamedFullOp> CreateNamedFullOpDirect(
692  ::flatbuffers::FlatBufferBuilder &_fbb,
694  const std::vector<int64_t> *shape = nullptr,
695  ::flatbuffers::Optional<tt::target::DataType> dtype = ::flatbuffers::nullopt,
696  ::flatbuffers::Optional<tt::target::TensorLayout> layout = ::flatbuffers::nullopt,
697  ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
698  ::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> memcfg = 0,
699  ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0) {
700  auto shape__ = shape ? _fbb.CreateVector<int64_t>(*shape) : 0;
702  _fbb,
703  type,
704  shape__,
705  dtype,
706  layout,
707  device,
708  memcfg,
709  out);
710 }
711 
712 } // namespace ttnn
713 } // namespace target
714 } // namespace tt
715 
716 #endif // FLATBUFFERS_GENERATED_CREATION_TT_TARGET_TTNN_H_
VT_DATA
Definition: program_generated.h:669
VT_LAYOUT
Definition: creation_generated.h:257
VT_DTYPE
Definition: creation_generated.h:78
VT_NUM_SHARDS
Definition: creation_generated.h:352
VT_STRATEGY
Definition: creation_generated.h:355
VT_SHAPE
Definition: creation_generated.h:255
VT_START
Definition: creation_generated.h:75
VT_TYPE
Definition: creation_generated.h:584
VT_DEVICE
Definition: creation_generated.h:79
VT_END
Definition: creation_generated.h:76
VT_FILL_VALUE
Definition: creation_generated.h:495
VT_OUT
Definition: creation_generated.h:185
VT_MEMCFG
Definition: creation_generated.h:80
VT_STEP
Definition: creation_generated.h:77
inline ::flatbuffers::Offset< ConstructTensorOp > CreateConstructTensorOpDirect(::flatbuffers::FlatBufferBuilder &_fbb, const std::vector< int64_t > *shape=nullptr, tt::target::DataType dtype=tt::target::DataType::Float32, tt::target::TensorLayout layout=tt::target::TensorLayout::RowMajor, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0)
Definition: creation_generated.h:330
inline ::flatbuffers::Offset< ConstantOp > CreateConstantOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0, ::flatbuffers::Offset<::flatbuffers::Vector< uint8_t >> data=0)
Definition: creation_generated.h:225
inline ::flatbuffers::Offset< NamedFullOp > CreateNamedFullOpDirect(::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::NamedFullOpType type=tt::target::ttnn::NamedFullOpType::Zeros, const std::vector< int64_t > *shape=nullptr, ::flatbuffers::Optional< tt::target::DataType > dtype=::flatbuffers::nullopt, ::flatbuffers::Optional< tt::target::TensorLayout > layout=::flatbuffers::nullopt, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memcfg=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0)
Definition: creation_generated.h:691
const char * EnumNameNamedFullOpType(NamedFullOpType e)
Definition: creation_generated.h:65
NamedFullOpType
Definition: creation_generated.h:41
const char *const * EnumNamesNamedFullOpType()
Definition: creation_generated.h:56
inline ::flatbuffers::Offset< NamedFullOp > CreateNamedFullOp(::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::NamedFullOpType type=tt::target::ttnn::NamedFullOpType::Zeros, ::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> shape=0, ::flatbuffers::Optional< tt::target::DataType > dtype=::flatbuffers::nullopt, ::flatbuffers::Optional< tt::target::TensorLayout > layout=::flatbuffers::nullopt, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memcfg=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0)
Definition: creation_generated.h:666
const NamedFullOpType(& EnumValuesNamedFullOpType())[2]
Definition: creation_generated.h:48
inline ::flatbuffers::Offset< FullOp > CreateFullOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, float fill_value=0.0f, uint32_t num_shards=0, ::flatbuffers::Offset< tt::target::ttnn::DistributionStrategy > strategy=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0)
Definition: creation_generated.h:559
inline ::flatbuffers::Offset< ConstructTensorOp > CreateConstructTensorOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> shape=0, tt::target::DataType dtype=tt::target::DataType::Float32, tt::target::TensorLayout layout=tt::target::TensorLayout::RowMajor, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0)
Definition: creation_generated.h:311
inline ::flatbuffers::Offset< EmptyOp > CreateEmptyOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> shape=0, tt::target::DataType dtype=tt::target::DataType::Float32, tt::target::TensorLayout layout=tt::target::TensorLayout::RowMajor, uint32_t num_shards=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memcfg=0, ::flatbuffers::Offset< tt::target::ttnn::DistributionStrategy > strategy=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0)
Definition: creation_generated.h:440
inline ::flatbuffers::Offset< EmptyOp > CreateEmptyOpDirect(::flatbuffers::FlatBufferBuilder &_fbb, const std::vector< int64_t > *shape=nullptr, tt::target::DataType dtype=tt::target::DataType::Float32, tt::target::TensorLayout layout=tt::target::TensorLayout::RowMajor, uint32_t num_shards=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memcfg=0, ::flatbuffers::Offset< tt::target::ttnn::DistributionStrategy > strategy=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0)
Definition: creation_generated.h:467
inline ::flatbuffers::Offset< ArangeOp > CreateArangeOp(::flatbuffers::FlatBufferBuilder &_fbb, float start=0.0f, float end=0.0f, float step=0.0f, ::flatbuffers::Optional< tt::target::DataType > dtype=::flatbuffers::nullopt, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memcfg=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0)
Definition: creation_generated.h:156
inline ::flatbuffers::Offset< ConstantOp > CreateConstantOpDirect(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0, const std::vector< uint8_t > *data=nullptr)
Definition: creation_generated.h:240
TensorLayout
Definition: types_generated.h:250
DataType
Definition: types_generated.h:81
Definition: debug_info_generated.h:18
Definition: debug_info_generated.h:36
Definition: creation_generated.h:120
void add_memcfg(::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memcfg)
Definition: creation_generated.h:139
ArangeOp Table
Definition: creation_generated.h:121
void add_start(float start)
Definition: creation_generated.h:124
void add_dtype(tt::target::DataType dtype)
Definition: creation_generated.h:133
void add_step(float step)
Definition: creation_generated.h:130
::flatbuffers::Offset< ArangeOp > Finish()
Definition: creation_generated.h:149
void add_out(::flatbuffers::Offset< tt::target::ttnn::TensorRef > out)
Definition: creation_generated.h:142
ArangeOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: creation_generated.h:145
::flatbuffers::FlatBufferBuilder & fbb_
Definition: creation_generated.h:122
::flatbuffers::uoffset_t start_
Definition: creation_generated.h:123
void add_end(float end)
Definition: creation_generated.h:127
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: creation_generated.h:136
Definition: creation_generated.h:176
ArangeOp type
Definition: creation_generated.h:177
static constexpr auto Create
Definition: creation_generated.h:178
Definition: creation_generated.h:204
ConstantOp Table
Definition: creation_generated.h:205
::flatbuffers::uoffset_t start_
Definition: creation_generated.h:207
::flatbuffers::FlatBufferBuilder & fbb_
Definition: creation_generated.h:206
::flatbuffers::Offset< ConstantOp > Finish()
Definition: creation_generated.h:218
ConstantOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: creation_generated.h:214
void add_data(::flatbuffers::Offset<::flatbuffers::Vector< uint8_t >> data)
Definition: creation_generated.h:211
void add_out(::flatbuffers::Offset< tt::target::ttnn::TensorRef > out)
Definition: creation_generated.h:208
Definition: creation_generated.h:235
ConstantOp type
Definition: creation_generated.h:236
static constexpr auto Create
Definition: creation_generated.h:237
Definition: creation_generated.h:284
ConstructTensorOp Table
Definition: creation_generated.h:285
void add_out(::flatbuffers::Offset< tt::target::ttnn::TensorRef > out)
Definition: creation_generated.h:297
void add_shape(::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> shape)
Definition: creation_generated.h:288
void add_dtype(tt::target::DataType dtype)
Definition: creation_generated.h:291
ConstructTensorOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: creation_generated.h:300
::flatbuffers::FlatBufferBuilder & fbb_
Definition: creation_generated.h:286
::flatbuffers::Offset< ConstructTensorOp > Finish()
Definition: creation_generated.h:304
::flatbuffers::uoffset_t start_
Definition: creation_generated.h:287
void add_layout(tt::target::TensorLayout layout)
Definition: creation_generated.h:294
Definition: creation_generated.h:325
ConstructTensorOp type
Definition: creation_generated.h:326
static constexpr auto Create
Definition: creation_generated.h:327
Definition: creation_generated.h:401
void add_memcfg(::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memcfg)
Definition: creation_generated.h:420
EmptyOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: creation_generated.h:429
void add_layout(tt::target::TensorLayout layout)
Definition: creation_generated.h:411
void add_shape(::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> shape)
Definition: creation_generated.h:405
::flatbuffers::Offset< EmptyOp > Finish()
Definition: creation_generated.h:433
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: creation_generated.h:417
::flatbuffers::FlatBufferBuilder & fbb_
Definition: creation_generated.h:403
void add_dtype(tt::target::DataType dtype)
Definition: creation_generated.h:408
void add_strategy(::flatbuffers::Offset< tt::target::ttnn::DistributionStrategy > strategy)
Definition: creation_generated.h:423
void add_out(::flatbuffers::Offset< tt::target::ttnn::TensorRef > out)
Definition: creation_generated.h:426
EmptyOp Table
Definition: creation_generated.h:402
void add_num_shards(uint32_t num_shards)
Definition: creation_generated.h:414
::flatbuffers::uoffset_t start_
Definition: creation_generated.h:404
Definition: creation_generated.h:462
static constexpr auto Create
Definition: creation_generated.h:464
EmptyOp type
Definition: creation_generated.h:463
Definition: binary_generated.h:31
tt::target::DataType dtype() const
Definition: creation_generated.h:263
::flatbuffers::Optional< tt::target::TensorLayout > layout() const
Definition: creation_generated.h:601
const tt::target::ttnn::TensorRef * out() const
Definition: creation_generated.h:101
float start() const
Definition: creation_generated.h:83
ConstructTensorOpBuilder Builder
Definition: creation_generated.h:252
uint32_t num_shards() const
Definition: creation_generated.h:367
const tt::target::ttnn::MemoryConfig * memcfg() const
Definition: creation_generated.h:98
const tt::target::ttnn::DistributionStrategy * strategy() const
Definition: creation_generated.h:376
const ::flatbuffers::Vector< int64_t > * shape() const
Definition: creation_generated.h:260
tt::target::TensorLayout layout() const
Definition: creation_generated.h:266
EmptyOpBuilder Builder
Definition: creation_generated.h:346
float fill_value() const
Definition: creation_generated.h:503
const ::flatbuffers::Vector< uint8_t > * data() const
Definition: creation_generated.h:191
::flatbuffers::Optional< tt::target::DataType > dtype() const
Definition: creation_generated.h:92
ArangeOpBuilder Builder
Definition: creation_generated.h:72
FullOpBuilder Builder
Definition: creation_generated.h:491
ConstantOpBuilder Builder
Definition: creation_generated.h:182
bool Verify(::flatbuffers::Verifier &verifier) const
Definition: creation_generated.h:104
float end() const
Definition: creation_generated.h:86
float step() const
Definition: creation_generated.h:89
tt::target::ttnn::NamedFullOpType type() const
Definition: creation_generated.h:592
const tt::target::DeviceRef * device() const
Definition: creation_generated.h:95
NamedFullOpBuilder Builder
Definition: creation_generated.h:581
Definition: creation_generated.h:529
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: creation_generated.h:533
::flatbuffers::FlatBufferBuilder & fbb_
Definition: creation_generated.h:531
void add_num_shards(uint32_t num_shards)
Definition: creation_generated.h:539
::flatbuffers::Offset< FullOp > Finish()
Definition: creation_generated.h:552
void add_strategy(::flatbuffers::Offset< tt::target::ttnn::DistributionStrategy > strategy)
Definition: creation_generated.h:542
void add_fill_value(float fill_value)
Definition: creation_generated.h:536
FullOp Table
Definition: creation_generated.h:530
void add_out(::flatbuffers::Offset< tt::target::ttnn::TensorRef > out)
Definition: creation_generated.h:545
FullOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: creation_generated.h:548
::flatbuffers::uoffset_t start_
Definition: creation_generated.h:532
Definition: creation_generated.h:575
FullOp type
Definition: creation_generated.h:576
static constexpr auto Create
Definition: creation_generated.h:577
Definition: creation_generated.h:630
void add_dtype(tt::target::DataType dtype)
Definition: creation_generated.h:640
void add_type(tt::target::ttnn::NamedFullOpType type)
Definition: creation_generated.h:634
void add_shape(::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> shape)
Definition: creation_generated.h:637
void add_layout(tt::target::TensorLayout layout)
Definition: creation_generated.h:643
::flatbuffers::FlatBufferBuilder & fbb_
Definition: creation_generated.h:632
NamedFullOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: creation_generated.h:655
::flatbuffers::Offset< NamedFullOp > Finish()
Definition: creation_generated.h:659
void add_memcfg(::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memcfg)
Definition: creation_generated.h:649
void add_out(::flatbuffers::Offset< tt::target::ttnn::TensorRef > out)
Definition: creation_generated.h:652
NamedFullOp Table
Definition: creation_generated.h:631
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: creation_generated.h:646
::flatbuffers::uoffset_t start_
Definition: creation_generated.h:633
Definition: creation_generated.h:686
static constexpr auto Create
Definition: creation_generated.h:688
NamedFullOp type
Definition: creation_generated.h:687