4 #ifndef FLATBUFFERS_GENERATED_PROGRAM_TT_TARGET_TTNN_H_
5 #define FLATBUFFERS_GENERATED_PROGRAM_TT_TARGET_TTNN_H_
7 #include "flatbuffers/flatbuffers.h"
11 static_assert(FLATBUFFERS_VERSION_MAJOR == 24 &&
12 FLATBUFFERS_VERSION_MINOR == 3 &&
13 FLATBUFFERS_VERSION_REVISION == 25,
14 "Non-compatible flatbuffers version included");
24 struct GetDeviceOpBuilder;
27 struct ToMemoryConfigOpBuilder;
30 struct ToLayoutOpBuilder;
33 struct TypecastOpBuilder;
36 struct ToDeviceOpBuilder;
39 struct UpdateCacheOpBuilder;
42 struct FillCacheOpBuilder;
45 struct FromDeviceOpBuilder;
48 struct EmptyOpBuilder;
57 struct ArangeOpBuilder;
60 struct ClampOpParamsBuilder;
63 struct EltwiseOpWithFloatParamsBuilder;
66 struct EltwiseOpBuilder;
69 struct ReductionOpBuilder;
72 struct EmbeddingOpBuilder;
75 struct EmbeddingBackwardOpBuilder;
78 struct SoftmaxOpBuilder;
81 struct TransposeOpBuilder;
84 struct ConcatOpBuilder;
87 struct ReshapeOpBuilder;
90 struct RepeatOpBuilder;
93 struct SliceOpBuilder;
96 struct LinearOpBuilder;
99 struct MatmulOpBuilder;
102 struct Conv2dOpBuilder;
105 struct ConvTranspose2dOpBuilder;
108 struct MaxPool2dOpBuilder;
111 struct DeallocateOpBuilder;
114 struct AllGatherOpBuilder;
117 struct PermuteOpBuilder;
120 struct ReduceScatterOpBuilder;
123 struct MeshShardOpBuilder;
126 struct OperationBuilder;
129 struct ProgramBuilder;
237 static const char *
const names[48] = {
292 const size_t index =
static_cast<size_t>(e);
314 static const char *
const names[4] = {
317 "EltwiseOpWithFloatParams",
325 const size_t index =
static_cast<size_t>(e);
342 bool VerifyEltwiseOpParamsVector(::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset<void>> *values, const ::flatbuffers::Vector<EltwiseOpParams> *types);
362 static const char *
const names[4] = {
373 const size_t index =
static_cast<size_t>(e);
416 static const OpType values[] = {
455 static const char *
const names[34] = {
471 "EmbeddingBackwardOp",
496 const size_t index =
static_cast<size_t>(e);
633 bool VerifyOpTypeVector(::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset<void>> *values, const ::flatbuffers::Vector<OpType> *types);
638 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
643 const tt::target::Dim2d *
mesh()
const {
644 return GetStruct<const tt::target::Dim2d *>(
VT_MESH);
646 const ::flatbuffers::Vector<uint32_t> *
chip_ids()
const {
647 return GetPointer<const ::flatbuffers::Vector<uint32_t> *>(
VT_CHIP_IDS);
649 const tt::target::DeviceRef *
out()
const {
650 return GetPointer<const tt::target::DeviceRef *>(
VT_OUT);
652 bool Verify(::flatbuffers::Verifier &verifier)
const {
653 return VerifyTableStart(verifier) &&
654 VerifyField<tt::target::Dim2d>(verifier,
VT_MESH, 4) &&
656 verifier.VerifyVector(chip_ids()) &&
657 VerifyOffset(verifier,
VT_OUT) &&
658 verifier.VerifyTable(out()) &&
665 ::flatbuffers::FlatBufferBuilder &
fbb_;
670 void add_chip_ids(::flatbuffers::Offset<::flatbuffers::Vector<uint32_t>> chip_ids) {
673 void add_out(::flatbuffers::Offset<tt::target::DeviceRef> out) {
680 ::flatbuffers::Offset<GetDeviceOp>
Finish() {
682 auto o = ::flatbuffers::Offset<GetDeviceOp>(end);
688 ::flatbuffers::FlatBufferBuilder &_fbb,
689 const tt::target::Dim2d *mesh =
nullptr,
690 ::flatbuffers::Offset<::flatbuffers::Vector<uint32_t>> chip_ids = 0,
691 ::flatbuffers::Offset<tt::target::DeviceRef> out = 0) {
705 ::flatbuffers::FlatBufferBuilder &_fbb,
706 const tt::target::Dim2d *mesh =
nullptr,
707 const std::vector<uint32_t> *chip_ids =
nullptr,
708 ::flatbuffers::Offset<tt::target::DeviceRef> out = 0) {
709 auto chip_ids__ = chip_ids ? _fbb.CreateVector<uint32_t>(*chip_ids) : 0;
720 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
725 const tt::target::TensorRef *
in0()
const {
726 return GetPointer<const tt::target::TensorRef *>(
VT_IN0);
728 const tt::target::MemoryConfigDesc *
memcfg()
const {
729 return GetPointer<const tt::target::MemoryConfigDesc *>(
VT_MEMCFG);
731 const tt::target::TensorRef *
out()
const {
732 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
734 bool Verify(::flatbuffers::Verifier &verifier)
const {
735 return VerifyTableStart(verifier) &&
736 VerifyOffset(verifier,
VT_IN0) &&
737 verifier.VerifyTable(in0()) &&
739 verifier.VerifyTable(memcfg()) &&
740 VerifyOffset(verifier,
VT_OUT) &&
741 verifier.VerifyTable(out()) &&
748 ::flatbuffers::FlatBufferBuilder &
fbb_;
750 void add_in0(::flatbuffers::Offset<tt::target::TensorRef> in0) {
753 void add_memcfg(::flatbuffers::Offset<tt::target::MemoryConfigDesc> memcfg) {
756 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
763 ::flatbuffers::Offset<ToMemoryConfigOp>
Finish() {
765 auto o = ::flatbuffers::Offset<ToMemoryConfigOp>(end);
771 ::flatbuffers::FlatBufferBuilder &_fbb,
772 ::flatbuffers::Offset<tt::target::TensorRef> in0 = 0,
773 ::flatbuffers::Offset<tt::target::MemoryConfigDesc> memcfg = 0,
774 ::flatbuffers::Offset<tt::target::TensorRef> out = 0) {
790 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
798 const tt::target::TensorRef *
in()
const {
799 return GetPointer<const tt::target::TensorRef *>(
VT_IN);
804 ::flatbuffers::Optional<tt::target::DataType>
dtype()
const {
805 return GetOptional<uint16_t, tt::target::DataType>(
VT_DTYPE);
807 const tt::target::MemoryConfigDesc *
memcfg()
const {
808 return GetPointer<const tt::target::MemoryConfigDesc *>(
VT_MEMCFG);
810 const tt::target::DeviceRef *
device()
const {
811 return GetPointer<const tt::target::DeviceRef *>(
VT_DEVICE);
813 const tt::target::TensorRef *
out()
const {
814 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
816 bool Verify(::flatbuffers::Verifier &verifier)
const {
817 return VerifyTableStart(verifier) &&
818 VerifyOffset(verifier,
VT_IN) &&
819 verifier.VerifyTable(in()) &&
820 VerifyField<uint16_t>(verifier,
VT_LAYOUT, 2) &&
821 VerifyField<uint16_t>(verifier,
VT_DTYPE, 2) &&
823 verifier.VerifyTable(memcfg()) &&
825 verifier.VerifyTable(device()) &&
826 VerifyOffset(verifier,
VT_OUT) &&
827 verifier.VerifyTable(out()) &&
834 ::flatbuffers::FlatBufferBuilder &
fbb_;
836 void add_in(::flatbuffers::Offset<tt::target::TensorRef> in) {
845 void add_memcfg(::flatbuffers::Offset<tt::target::MemoryConfigDesc> memcfg) {
848 void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
851 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
858 ::flatbuffers::Offset<ToLayoutOp>
Finish() {
860 auto o = ::flatbuffers::Offset<ToLayoutOp>(end);
866 ::flatbuffers::FlatBufferBuilder &_fbb,
867 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
869 ::flatbuffers::Optional<tt::target::DataType> dtype = ::flatbuffers::nullopt,
870 ::flatbuffers::Offset<tt::target::MemoryConfigDesc> memcfg = 0,
871 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
872 ::flatbuffers::Offset<tt::target::TensorRef> out = 0) {
878 if(dtype) { builder_.
add_dtype(*dtype); }
891 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
896 const tt::target::TensorRef *
in()
const {
897 return GetPointer<const tt::target::TensorRef *>(
VT_IN);
902 const tt::target::TensorRef *
out()
const {
903 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
905 bool Verify(::flatbuffers::Verifier &verifier)
const {
906 return VerifyTableStart(verifier) &&
907 VerifyOffset(verifier,
VT_IN) &&
908 verifier.VerifyTable(in()) &&
909 VerifyField<uint16_t>(verifier,
VT_DTYPE, 2) &&
910 VerifyOffset(verifier,
VT_OUT) &&
911 verifier.VerifyTable(out()) &&
918 ::flatbuffers::FlatBufferBuilder &
fbb_;
920 void add_in(::flatbuffers::Offset<tt::target::TensorRef> in) {
926 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
933 ::flatbuffers::Offset<TypecastOp>
Finish() {
935 auto o = ::flatbuffers::Offset<TypecastOp>(end);
941 ::flatbuffers::FlatBufferBuilder &_fbb,
942 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
944 ::flatbuffers::Offset<tt::target::TensorRef> out = 0) {
960 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
966 const tt::target::TensorRef *
in()
const {
967 return GetPointer<const tt::target::TensorRef *>(
VT_IN);
969 const tt::target::DeviceRef *
device()
const {
970 return GetPointer<const tt::target::DeviceRef *>(
VT_DEVICE);
972 const tt::target::MemoryConfigDesc *
memcfg()
const {
973 return GetPointer<const tt::target::MemoryConfigDesc *>(
VT_MEMCFG);
975 const tt::target::TensorRef *
out()
const {
976 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
978 bool Verify(::flatbuffers::Verifier &verifier)
const {
979 return VerifyTableStart(verifier) &&
980 VerifyOffset(verifier,
VT_IN) &&
981 verifier.VerifyTable(in()) &&
983 verifier.VerifyTable(device()) &&
985 verifier.VerifyTable(memcfg()) &&
986 VerifyOffset(verifier,
VT_OUT) &&
987 verifier.VerifyTable(out()) &&
994 ::flatbuffers::FlatBufferBuilder &
fbb_;
996 void add_in(::flatbuffers::Offset<tt::target::TensorRef> in) {
999 void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
1002 void add_memcfg(::flatbuffers::Offset<tt::target::MemoryConfigDesc> memcfg) {
1005 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
1014 auto o = ::flatbuffers::Offset<ToDeviceOp>(end);
1020 ::flatbuffers::FlatBufferBuilder &_fbb,
1021 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
1022 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
1023 ::flatbuffers::Offset<tt::target::MemoryConfigDesc> memcfg = 0,
1024 ::flatbuffers::Offset<tt::target::TensorRef> out = 0) {
1030 return builder_.
Finish();
1041 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1045 VT_BATCH_OFFSET = 10
1047 const tt::target::TensorRef *
cache()
const {
1048 return GetPointer<const tt::target::TensorRef *>(
VT_CACHE);
1050 const tt::target::TensorRef *
input()
const {
1051 return GetPointer<const tt::target::TensorRef *>(
VT_INPUT);
1057 return GetField<uint32_t>(VT_BATCH_OFFSET, 0);
1059 bool Verify(::flatbuffers::Verifier &verifier)
const {
1060 return VerifyTableStart(verifier) &&
1061 VerifyOffset(verifier,
VT_CACHE) &&
1062 verifier.VerifyTable(cache()) &&
1063 VerifyOffset(verifier,
VT_INPUT) &&
1064 verifier.VerifyTable(input()) &&
1066 verifier.VerifyTable(update_index()) &&
1067 VerifyField<uint32_t>(verifier, VT_BATCH_OFFSET, 4) &&
1068 verifier.EndTable();
1074 ::flatbuffers::FlatBufferBuilder &
fbb_;
1076 void add_cache(::flatbuffers::Offset<tt::target::TensorRef> cache) {
1079 void add_input(::flatbuffers::Offset<tt::target::TensorRef> input) {
1086 fbb_.AddElement<uint32_t>(UpdateCacheOp::VT_BATCH_OFFSET, batch_offset, 0);
1092 ::flatbuffers::Offset<UpdateCacheOp>
Finish() {
1094 auto o = ::flatbuffers::Offset<UpdateCacheOp>(end);
1100 ::flatbuffers::FlatBufferBuilder &_fbb,
1101 ::flatbuffers::Offset<tt::target::TensorRef> cache = 0,
1102 ::flatbuffers::Offset<tt::target::TensorRef> input = 0,
1103 ::flatbuffers::Offset<tt::target::TensorRef> update_index = 0,
1104 uint32_t batch_offset = 0) {
1110 return builder_.
Finish();
1121 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1126 const tt::target::TensorRef *
cache()
const {
1127 return GetPointer<const tt::target::TensorRef *>(
VT_CACHE);
1129 const tt::target::TensorRef *
input()
const {
1130 return GetPointer<const tt::target::TensorRef *>(
VT_INPUT);
1133 return GetField<uint32_t>(VT_BATCH_OFFSET, 0);
1135 bool Verify(::flatbuffers::Verifier &verifier)
const {
1136 return VerifyTableStart(verifier) &&
1137 VerifyOffset(verifier,
VT_CACHE) &&
1138 verifier.VerifyTable(cache()) &&
1139 VerifyOffset(verifier,
VT_INPUT) &&
1140 verifier.VerifyTable(input()) &&
1141 VerifyField<uint32_t>(verifier, VT_BATCH_OFFSET, 4) &&
1142 verifier.EndTable();
1148 ::flatbuffers::FlatBufferBuilder &
fbb_;
1150 void add_cache(::flatbuffers::Offset<tt::target::TensorRef> cache) {
1153 void add_input(::flatbuffers::Offset<tt::target::TensorRef> input) {
1157 fbb_.AddElement<uint32_t>(FillCacheOp::VT_BATCH_OFFSET, batch_offset, 0);
1163 ::flatbuffers::Offset<FillCacheOp>
Finish() {
1165 auto o = ::flatbuffers::Offset<FillCacheOp>(end);
1171 ::flatbuffers::FlatBufferBuilder &_fbb,
1172 ::flatbuffers::Offset<tt::target::TensorRef> cache = 0,
1173 ::flatbuffers::Offset<tt::target::TensorRef> input = 0,
1174 uint32_t batch_offset = 0) {
1179 return builder_.
Finish();
1190 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1194 const tt::target::TensorRef *
in()
const {
1195 return GetPointer<const tt::target::TensorRef *>(
VT_IN);
1197 const tt::target::TensorRef *
out()
const {
1198 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
1200 bool Verify(::flatbuffers::Verifier &verifier)
const {
1201 return VerifyTableStart(verifier) &&
1202 VerifyOffset(verifier,
VT_IN) &&
1203 verifier.VerifyTable(in()) &&
1204 VerifyOffset(verifier,
VT_OUT) &&
1205 verifier.VerifyTable(out()) &&
1206 verifier.EndTable();
1212 ::flatbuffers::FlatBufferBuilder &
fbb_;
1214 void add_in(::flatbuffers::Offset<tt::target::TensorRef> in) {
1217 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
1224 ::flatbuffers::Offset<FromDeviceOp>
Finish() {
1226 auto o = ::flatbuffers::Offset<FromDeviceOp>(end);
1232 ::flatbuffers::FlatBufferBuilder &_fbb,
1233 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
1234 ::flatbuffers::Offset<tt::target::TensorRef> out = 0) {
1238 return builder_.
Finish();
1249 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1259 const ::flatbuffers::Vector<int64_t> *
shape()
const {
1260 return GetPointer<const ::flatbuffers::Vector<int64_t> *>(
VT_SHAPE);
1271 const tt::target::DeviceRef *
device()
const {
1272 return GetPointer<const tt::target::DeviceRef *>(
VT_DEVICE);
1274 const tt::target::MemoryConfigDesc *
memcfg()
const {
1275 return GetPointer<const tt::target::MemoryConfigDesc *>(
VT_MEMCFG);
1277 const tt::target::DistributionStrategy *
strategy()
const {
1278 return GetPointer<const tt::target::DistributionStrategy *>(
VT_STRATEGY);
1280 const tt::target::TensorRef *
out()
const {
1281 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
1283 bool Verify(::flatbuffers::Verifier &verifier)
const {
1284 return VerifyTableStart(verifier) &&
1285 VerifyOffset(verifier,
VT_SHAPE) &&
1286 verifier.VerifyVector(shape()) &&
1287 VerifyField<uint16_t>(verifier,
VT_DTYPE, 2) &&
1288 VerifyField<uint16_t>(verifier,
VT_LAYOUT, 2) &&
1291 verifier.VerifyTable(device()) &&
1293 verifier.VerifyTable(memcfg()) &&
1295 verifier.VerifyTable(strategy()) &&
1296 VerifyOffset(verifier,
VT_OUT) &&
1297 verifier.VerifyTable(out()) &&
1298 verifier.EndTable();
1304 ::flatbuffers::FlatBufferBuilder &
fbb_;
1306 void add_shape(::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> shape) {
1318 void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
1321 void add_memcfg(::flatbuffers::Offset<tt::target::MemoryConfigDesc> memcfg) {
1324 void add_strategy(::flatbuffers::Offset<tt::target::DistributionStrategy> strategy) {
1327 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
1336 auto o = ::flatbuffers::Offset<EmptyOp>(end);
1342 ::flatbuffers::FlatBufferBuilder &_fbb,
1343 ::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> shape = 0,
1346 uint32_t num_shards = 0,
1347 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
1348 ::flatbuffers::Offset<tt::target::MemoryConfigDesc> memcfg = 0,
1349 ::flatbuffers::Offset<tt::target::DistributionStrategy> strategy = 0,
1350 ::flatbuffers::Offset<tt::target::TensorRef> out = 0) {
1360 return builder_.
Finish();
1369 ::flatbuffers::FlatBufferBuilder &_fbb,
1370 const std::vector<int64_t> *shape =
nullptr,
1373 uint32_t num_shards = 0,
1374 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
1375 ::flatbuffers::Offset<tt::target::MemoryConfigDesc> memcfg = 0,
1376 ::flatbuffers::Offset<tt::target::DistributionStrategy> strategy = 0,
1377 ::flatbuffers::Offset<tt::target::TensorRef> out = 0) {
1378 auto shape__ = shape ? _fbb.CreateVector<int64_t>(*shape) : 0;
1394 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1402 const ::flatbuffers::Vector<int64_t> *
shape()
const {
1403 return GetPointer<const ::flatbuffers::Vector<int64_t> *>(
VT_SHAPE);
1405 ::flatbuffers::Optional<tt::target::DataType>
dtype()
const {
1406 return GetOptional<uint16_t, tt::target::DataType>(
VT_DTYPE);
1408 ::flatbuffers::Optional<tt::target::TensorLayout>
layout()
const {
1409 return GetOptional<uint16_t, tt::target::TensorLayout>(
VT_LAYOUT);
1411 const tt::target::DeviceRef *
device()
const {
1412 return GetPointer<const tt::target::DeviceRef *>(
VT_DEVICE);
1414 const tt::target::MemoryConfigDesc *
memcfg()
const {
1415 return GetPointer<const tt::target::MemoryConfigDesc *>(
VT_MEMCFG);
1417 const tt::target::TensorRef *
out()
const {
1418 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
1420 bool Verify(::flatbuffers::Verifier &verifier)
const {
1421 return VerifyTableStart(verifier) &&
1422 VerifyOffset(verifier,
VT_SHAPE) &&
1423 verifier.VerifyVector(shape()) &&
1424 VerifyField<uint16_t>(verifier,
VT_DTYPE, 2) &&
1425 VerifyField<uint16_t>(verifier,
VT_LAYOUT, 2) &&
1427 verifier.VerifyTable(device()) &&
1429 verifier.VerifyTable(memcfg()) &&
1430 VerifyOffset(verifier,
VT_OUT) &&
1431 verifier.VerifyTable(out()) &&
1432 verifier.EndTable();
1438 ::flatbuffers::FlatBufferBuilder &
fbb_;
1440 void add_shape(::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> shape) {
1449 void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
1452 void add_memcfg(::flatbuffers::Offset<tt::target::MemoryConfigDesc> memcfg) {
1455 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
1464 auto o = ::flatbuffers::Offset<OnesOp>(end);
1470 ::flatbuffers::FlatBufferBuilder &_fbb,
1471 ::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> shape = 0,
1472 ::flatbuffers::Optional<tt::target::DataType> dtype = ::flatbuffers::nullopt,
1473 ::flatbuffers::Optional<tt::target::TensorLayout> layout = ::flatbuffers::nullopt,
1474 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
1475 ::flatbuffers::Offset<tt::target::MemoryConfigDesc> memcfg = 0,
1476 ::flatbuffers::Offset<tt::target::TensorRef> out = 0) {
1483 if(dtype) { builder_.
add_dtype(*dtype); }
1484 return builder_.
Finish();
1493 ::flatbuffers::FlatBufferBuilder &_fbb,
1494 const std::vector<int64_t> *shape =
nullptr,
1495 ::flatbuffers::Optional<tt::target::DataType> dtype = ::flatbuffers::nullopt,
1496 ::flatbuffers::Optional<tt::target::TensorLayout> layout = ::flatbuffers::nullopt,
1497 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
1498 ::flatbuffers::Offset<tt::target::MemoryConfigDesc> memcfg = 0,
1499 ::flatbuffers::Offset<tt::target::TensorRef> out = 0) {
1500 auto shape__ = shape ? _fbb.CreateVector<int64_t>(*shape) : 0;
1514 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1521 const tt::target::DeviceRef *
device()
const {
1522 return GetPointer<const tt::target::DeviceRef *>(
VT_DEVICE);
1530 const tt::target::DistributionStrategy *
strategy()
const {
1531 return GetPointer<const tt::target::DistributionStrategy *>(
VT_STRATEGY);
1533 const tt::target::TensorRef *
out()
const {
1534 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
1536 bool Verify(::flatbuffers::Verifier &verifier)
const {
1537 return VerifyTableStart(verifier) &&
1539 verifier.VerifyTable(device()) &&
1543 verifier.VerifyTable(strategy()) &&
1544 VerifyOffset(verifier,
VT_OUT) &&
1545 verifier.VerifyTable(out()) &&
1546 verifier.EndTable();
1552 ::flatbuffers::FlatBufferBuilder &
fbb_;
1554 void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
1563 void add_strategy(::flatbuffers::Offset<tt::target::DistributionStrategy> strategy) {
1566 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
1575 auto o = ::flatbuffers::Offset<FullOp>(end);
1581 ::flatbuffers::FlatBufferBuilder &_fbb,
1582 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
1583 float fill_value = 0.0f,
1584 uint32_t num_shards = 0,
1585 ::flatbuffers::Offset<tt::target::DistributionStrategy> strategy = 0,
1586 ::flatbuffers::Offset<tt::target::TensorRef> out = 0) {
1593 return builder_.
Finish();
1604 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1614 return GetField<float>(
VT_START, 0.0f);
1617 return GetField<float>(
VT_END, 0.0f);
1620 return GetField<float>(
VT_STEP, 0.0f);
1622 ::flatbuffers::Optional<tt::target::DataType>
dtype()
const {
1623 return GetOptional<uint16_t, tt::target::DataType>(
VT_DTYPE);
1625 const tt::target::DeviceRef *
device()
const {
1626 return GetPointer<const tt::target::DeviceRef *>(
VT_DEVICE);
1628 const tt::target::MemoryConfigDesc *
memcfg()
const {
1629 return GetPointer<const tt::target::MemoryConfigDesc *>(
VT_MEMCFG);
1631 const tt::target::TensorRef *
out()
const {
1632 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
1634 bool Verify(::flatbuffers::Verifier &verifier)
const {
1635 return VerifyTableStart(verifier) &&
1636 VerifyField<float>(verifier,
VT_START, 4) &&
1637 VerifyField<float>(verifier,
VT_END, 4) &&
1638 VerifyField<float>(verifier,
VT_STEP, 4) &&
1639 VerifyField<uint16_t>(verifier,
VT_DTYPE, 2) &&
1641 verifier.VerifyTable(device()) &&
1643 verifier.VerifyTable(memcfg()) &&
1644 VerifyOffset(verifier,
VT_OUT) &&
1645 verifier.VerifyTable(out()) &&
1646 verifier.EndTable();
1652 ::flatbuffers::FlatBufferBuilder &
fbb_;
1666 void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
1669 void add_memcfg(::flatbuffers::Offset<tt::target::MemoryConfigDesc> memcfg) {
1672 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
1681 auto o = ::flatbuffers::Offset<ArangeOp>(end);
1687 ::flatbuffers::FlatBufferBuilder &_fbb,
1691 ::flatbuffers::Optional<tt::target::DataType> dtype = ::flatbuffers::nullopt,
1692 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
1693 ::flatbuffers::Offset<tt::target::MemoryConfigDesc> memcfg = 0,
1694 ::flatbuffers::Offset<tt::target::TensorRef> out = 0) {
1702 if(dtype) { builder_.
add_dtype(*dtype); }
1703 return builder_.
Finish();
1714 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1719 return GetField<float>(
VT_MIN, 0.0f);
1722 return GetField<float>(VT_MAX, 0.0f);
1724 bool Verify(::flatbuffers::Verifier &verifier)
const {
1725 return VerifyTableStart(verifier) &&
1726 VerifyField<float>(verifier,
VT_MIN, 4) &&
1727 VerifyField<float>(verifier, VT_MAX, 4) &&
1728 verifier.EndTable();
1734 ::flatbuffers::FlatBufferBuilder &
fbb_;
1740 fbb_.AddElement<
float>(ClampOpParams::VT_MAX, max, 0.0f);
1746 ::flatbuffers::Offset<ClampOpParams>
Finish() {
1748 auto o = ::flatbuffers::Offset<ClampOpParams>(end);
1754 ::flatbuffers::FlatBufferBuilder &_fbb,
1760 return builder_.
Finish();
1771 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1775 return GetField<float>(VT_PARAMETER, 0.0f);
1777 bool Verify(::flatbuffers::Verifier &verifier)
const {
1778 return VerifyTableStart(verifier) &&
1779 VerifyField<float>(verifier, VT_PARAMETER, 4) &&
1780 verifier.EndTable();
1786 ::flatbuffers::FlatBufferBuilder &
fbb_;
1789 fbb_.AddElement<
float>(EltwiseOpWithFloatParams::VT_PARAMETER, parameter, 0.0f);
1795 ::flatbuffers::Offset<EltwiseOpWithFloatParams>
Finish() {
1797 auto o = ::flatbuffers::Offset<EltwiseOpWithFloatParams>(end);
1803 ::flatbuffers::FlatBufferBuilder &_fbb,
1804 float parameter = 0.0f) {
1807 return builder_.
Finish();
1818 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1828 const ::flatbuffers::Vector<::flatbuffers::Offset<tt::target::TensorRef>> *
ins()
const {
1829 return GetPointer<const ::flatbuffers::Vector<::flatbuffers::Offset<tt::target::TensorRef>> *>(
VT_INS);
1831 const tt::target::TensorRef *
out()
const {
1832 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
1838 return GetPointer<const void *>(VT_PARAMS);
1847 bool Verify(::flatbuffers::Verifier &verifier)
const {
1848 return VerifyTableStart(verifier) &&
1849 VerifyField<uint32_t>(verifier,
VT_TYPE, 4) &&
1850 VerifyOffset(verifier,
VT_INS) &&
1851 verifier.VerifyVector(ins()) &&
1852 verifier.VerifyVectorOfTables(ins()) &&
1853 VerifyOffset(verifier,
VT_OUT) &&
1854 verifier.VerifyTable(out()) &&
1856 VerifyOffset(verifier, VT_PARAMS) &&
1858 verifier.EndTable();
1862 template<>
inline const tt::target::ttnn::ClampOpParams *EltwiseOp::params_as<tt::target::ttnn::ClampOpParams>()
const {
1863 return params_as_ClampOpParams();
1866 template<>
inline const tt::target::ttnn::EltwiseOpWithFloatParams *EltwiseOp::params_as<tt::target::ttnn::EltwiseOpWithFloatParams>()
const {
1867 return params_as_EltwiseOpWithFloatParams();
1872 ::flatbuffers::FlatBufferBuilder &
fbb_;
1877 void add_ins(::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<tt::target::TensorRef>>> ins) {
1880 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
1887 fbb_.AddOffset(EltwiseOp::VT_PARAMS, params);
1895 auto o = ::flatbuffers::Offset<EltwiseOp>(end);
1901 ::flatbuffers::FlatBufferBuilder &_fbb,
1903 ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<tt::target::TensorRef>>> ins = 0,
1904 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
1906 ::flatbuffers::Offset<void> params = 0) {
1913 return builder_.
Finish();
1922 ::flatbuffers::FlatBufferBuilder &_fbb,
1924 const std::vector<::flatbuffers::Offset<tt::target::TensorRef>> *ins =
nullptr,
1925 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
1927 ::flatbuffers::Offset<void> params = 0) {
1928 auto ins__ = ins ? _fbb.CreateVector<::flatbuffers::Offset<tt::target::TensorRef>>(*ins) : 0;
1941 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
1951 const tt::target::TensorRef *
in()
const {
1952 return GetPointer<const tt::target::TensorRef *>(
VT_IN);
1954 const tt::target::TensorRef *
out()
const {
1955 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
1957 const ::flatbuffers::Vector<int32_t> *
dim_arg()
const {
1958 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_DIM_ARG);
1961 return GetField<uint8_t>(VT_KEEP_DIM, 0) != 0;
1963 bool Verify(::flatbuffers::Verifier &verifier)
const {
1964 return VerifyTableStart(verifier) &&
1965 VerifyField<uint32_t>(verifier,
VT_TYPE, 4) &&
1966 VerifyOffset(verifier,
VT_IN) &&
1967 verifier.VerifyTable(in()) &&
1968 VerifyOffset(verifier,
VT_OUT) &&
1969 verifier.VerifyTable(out()) &&
1971 verifier.VerifyVector(dim_arg()) &&
1972 VerifyField<uint8_t>(verifier, VT_KEEP_DIM, 1) &&
1973 verifier.EndTable();
1979 ::flatbuffers::FlatBufferBuilder &
fbb_;
1984 void add_in(::flatbuffers::Offset<tt::target::TensorRef> in) {
1987 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
1990 void add_dim_arg(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> dim_arg) {
1994 fbb_.AddElement<uint8_t>(ReductionOp::VT_KEEP_DIM,
static_cast<uint8_t
>(keep_dim), 0);
2000 ::flatbuffers::Offset<ReductionOp>
Finish() {
2002 auto o = ::flatbuffers::Offset<ReductionOp>(end);
2008 ::flatbuffers::FlatBufferBuilder &_fbb,
2010 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
2011 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
2012 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> dim_arg = 0,
2013 bool keep_dim =
false) {
2020 return builder_.
Finish();
2029 ::flatbuffers::FlatBufferBuilder &_fbb,
2031 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
2032 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
2033 const std::vector<int32_t> *dim_arg =
nullptr,
2034 bool keep_dim =
false) {
2035 auto dim_arg__ = dim_arg ? _fbb.CreateVector<int32_t>(*dim_arg) : 0;
2048 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
2053 const tt::target::TensorRef *
input()
const {
2054 return GetPointer<const tt::target::TensorRef *>(
VT_INPUT);
2056 const tt::target::TensorRef *
weight()
const {
2057 return GetPointer<const tt::target::TensorRef *>(
VT_WEIGHT);
2059 const tt::target::TensorRef *
out()
const {
2060 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
2062 bool Verify(::flatbuffers::Verifier &verifier)
const {
2063 return VerifyTableStart(verifier) &&
2064 VerifyOffset(verifier,
VT_INPUT) &&
2065 verifier.VerifyTable(input()) &&
2067 verifier.VerifyTable(weight()) &&
2068 VerifyOffset(verifier,
VT_OUT) &&
2069 verifier.VerifyTable(out()) &&
2070 verifier.EndTable();
2076 ::flatbuffers::FlatBufferBuilder &
fbb_;
2078 void add_input(::flatbuffers::Offset<tt::target::TensorRef> input) {
2081 void add_weight(::flatbuffers::Offset<tt::target::TensorRef> weight) {
2084 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
2091 ::flatbuffers::Offset<EmbeddingOp>
Finish() {
2093 auto o = ::flatbuffers::Offset<EmbeddingOp>(end);
2099 ::flatbuffers::FlatBufferBuilder &_fbb,
2100 ::flatbuffers::Offset<tt::target::TensorRef> input = 0,
2101 ::flatbuffers::Offset<tt::target::TensorRef> weight = 0,
2102 ::flatbuffers::Offset<tt::target::TensorRef> out = 0) {
2107 return builder_.
Finish();
2118 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
2126 const tt::target::TensorRef *
input()
const {
2127 return GetPointer<const tt::target::TensorRef *>(
VT_INPUT);
2129 const tt::target::TensorRef *
weight()
const {
2130 return GetPointer<const tt::target::TensorRef *>(
VT_WEIGHT);
2133 return GetPointer<const tt::target::TensorRef *>(
VT_IN_GRAD);
2135 ::flatbuffers::Optional<tt::target::DataType>
dtype()
const {
2136 return GetOptional<uint16_t, tt::target::DataType>(
VT_DTYPE);
2138 const tt::target::MemoryConfigDesc *
memcfg()
const {
2139 return GetPointer<const tt::target::MemoryConfigDesc *>(
VT_MEMCFG);
2141 const tt::target::TensorRef *
out()
const {
2142 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
2144 bool Verify(::flatbuffers::Verifier &verifier)
const {
2145 return VerifyTableStart(verifier) &&
2146 VerifyOffset(verifier,
VT_INPUT) &&
2147 verifier.VerifyTable(input()) &&
2149 verifier.VerifyTable(weight()) &&
2151 verifier.VerifyTable(in_grad()) &&
2152 VerifyField<uint16_t>(verifier,
VT_DTYPE, 2) &&
2154 verifier.VerifyTable(memcfg()) &&
2155 VerifyOffset(verifier,
VT_OUT) &&
2156 verifier.VerifyTable(out()) &&
2157 verifier.EndTable();
2163 ::flatbuffers::FlatBufferBuilder &
fbb_;
2165 void add_input(::flatbuffers::Offset<tt::target::TensorRef> input) {
2168 void add_weight(::flatbuffers::Offset<tt::target::TensorRef> weight) {
2171 void add_in_grad(::flatbuffers::Offset<tt::target::TensorRef> in_grad) {
2177 void add_memcfg(::flatbuffers::Offset<tt::target::MemoryConfigDesc> memcfg) {
2180 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
2187 ::flatbuffers::Offset<EmbeddingBackwardOp>
Finish() {
2189 auto o = ::flatbuffers::Offset<EmbeddingBackwardOp>(end);
2195 ::flatbuffers::FlatBufferBuilder &_fbb,
2196 ::flatbuffers::Offset<tt::target::TensorRef> input = 0,
2197 ::flatbuffers::Offset<tt::target::TensorRef> weight = 0,
2198 ::flatbuffers::Offset<tt::target::TensorRef> in_grad = 0,
2199 ::flatbuffers::Optional<tt::target::DataType> dtype = ::flatbuffers::nullopt,
2200 ::flatbuffers::Offset<tt::target::MemoryConfigDesc> memcfg = 0,
2201 ::flatbuffers::Offset<tt::target::TensorRef> out = 0) {
2208 if(dtype) { builder_.
add_dtype(*dtype); }
2209 return builder_.
Finish();
2220 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
2225 const tt::target::TensorRef *
in()
const {
2226 return GetPointer<const tt::target::TensorRef *>(
VT_IN);
2228 const tt::target::TensorRef *
out()
const {
2229 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
2232 return GetField<int32_t>(VT_DIMENSION, 0);
2234 bool Verify(::flatbuffers::Verifier &verifier)
const {
2235 return VerifyTableStart(verifier) &&
2236 VerifyOffset(verifier,
VT_IN) &&
2237 verifier.VerifyTable(in()) &&
2238 VerifyOffset(verifier,
VT_OUT) &&
2239 verifier.VerifyTable(out()) &&
2240 VerifyField<int32_t>(verifier, VT_DIMENSION, 4) &&
2241 verifier.EndTable();
2247 ::flatbuffers::FlatBufferBuilder &
fbb_;
2249 void add_in(::flatbuffers::Offset<tt::target::TensorRef> in) {
2252 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
2256 fbb_.AddElement<int32_t>(SoftmaxOp::VT_DIMENSION, dimension, 0);
2264 auto o = ::flatbuffers::Offset<SoftmaxOp>(end);
2270 ::flatbuffers::FlatBufferBuilder &_fbb,
2271 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
2272 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
2273 int32_t dimension = 0) {
2278 return builder_.
Finish();
2289 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
2295 const tt::target::TensorRef *
in()
const {
2296 return GetPointer<const tt::target::TensorRef *>(
VT_IN);
2298 const tt::target::TensorRef *
out()
const {
2299 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
2302 return GetField<int32_t>(
VT_DIM0, 0);
2305 return GetField<int32_t>(VT_DIM1, 0);
2307 bool Verify(::flatbuffers::Verifier &verifier)
const {
2308 return VerifyTableStart(verifier) &&
2309 VerifyOffset(verifier,
VT_IN) &&
2310 verifier.VerifyTable(in()) &&
2311 VerifyOffset(verifier,
VT_OUT) &&
2312 verifier.VerifyTable(out()) &&
2313 VerifyField<int32_t>(verifier,
VT_DIM0, 4) &&
2314 VerifyField<int32_t>(verifier, VT_DIM1, 4) &&
2315 verifier.EndTable();
2321 ::flatbuffers::FlatBufferBuilder &
fbb_;
2323 void add_in(::flatbuffers::Offset<tt::target::TensorRef> in) {
2326 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
2333 fbb_.AddElement<int32_t>(TransposeOp::VT_DIM1, dim1, 0);
2339 ::flatbuffers::Offset<TransposeOp>
Finish() {
2341 auto o = ::flatbuffers::Offset<TransposeOp>(end);
2347 ::flatbuffers::FlatBufferBuilder &_fbb,
2348 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
2349 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
2357 return builder_.
Finish();
2368 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
2373 const ::flatbuffers::Vector<::flatbuffers::Offset<tt::target::TensorRef>> *
inputs()
const {
2374 return GetPointer<const ::flatbuffers::Vector<::flatbuffers::Offset<tt::target::TensorRef>> *>(
VT_INPUTS);
2376 const tt::target::TensorRef *
out()
const {
2377 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
2380 return GetField<int32_t>(
VT_DIM, 0);
2382 bool Verify(::flatbuffers::Verifier &verifier)
const {
2383 return VerifyTableStart(verifier) &&
2385 verifier.VerifyVector(inputs()) &&
2386 verifier.VerifyVectorOfTables(inputs()) &&
2387 VerifyOffset(verifier,
VT_OUT) &&
2388 verifier.VerifyTable(out()) &&
2389 VerifyField<int32_t>(verifier,
VT_DIM, 4) &&
2390 verifier.EndTable();
2396 ::flatbuffers::FlatBufferBuilder &
fbb_;
2398 void add_inputs(::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<tt::target::TensorRef>>> inputs) {
2401 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
2413 auto o = ::flatbuffers::Offset<ConcatOp>(end);
2419 ::flatbuffers::FlatBufferBuilder &_fbb,
2420 ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<tt::target::TensorRef>>> inputs = 0,
2421 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
2427 return builder_.
Finish();
2436 ::flatbuffers::FlatBufferBuilder &_fbb,
2437 const std::vector<::flatbuffers::Offset<tt::target::TensorRef>> *inputs =
nullptr,
2438 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
2440 auto inputs__ = inputs ? _fbb.CreateVector<::flatbuffers::Offset<tt::target::TensorRef>>(*inputs) : 0;
2451 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
2456 const tt::target::TensorRef *
in()
const {
2457 return GetPointer<const tt::target::TensorRef *>(
VT_IN);
2459 const tt::target::TensorRef *
out()
const {
2460 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
2462 const ::flatbuffers::Vector<int32_t> *
shape()
const {
2463 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_SHAPE);
2465 bool Verify(::flatbuffers::Verifier &verifier)
const {
2466 return VerifyTableStart(verifier) &&
2467 VerifyOffset(verifier,
VT_IN) &&
2468 verifier.VerifyTable(in()) &&
2469 VerifyOffset(verifier,
VT_OUT) &&
2470 verifier.VerifyTable(out()) &&
2471 VerifyOffset(verifier,
VT_SHAPE) &&
2472 verifier.VerifyVector(shape()) &&
2473 verifier.EndTable();
2479 ::flatbuffers::FlatBufferBuilder &
fbb_;
2481 void add_in(::flatbuffers::Offset<tt::target::TensorRef> in) {
2484 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
2487 void add_shape(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> shape) {
2496 auto o = ::flatbuffers::Offset<ReshapeOp>(end);
2502 ::flatbuffers::FlatBufferBuilder &_fbb,
2503 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
2504 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
2505 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> shape = 0) {
2510 return builder_.
Finish();
2519 ::flatbuffers::FlatBufferBuilder &_fbb,
2520 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
2521 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
2522 const std::vector<int32_t> *shape =
nullptr) {
2523 auto shape__ = shape ? _fbb.CreateVector<int32_t>(*shape) : 0;
2534 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
2539 const tt::target::TensorRef *
in()
const {
2540 return GetPointer<const tt::target::TensorRef *>(
VT_IN);
2542 const tt::target::TensorRef *
out()
const {
2543 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
2545 const ::flatbuffers::Vector<uint32_t> *
shape()
const {
2546 return GetPointer<const ::flatbuffers::Vector<uint32_t> *>(
VT_SHAPE);
2548 bool Verify(::flatbuffers::Verifier &verifier)
const {
2549 return VerifyTableStart(verifier) &&
2550 VerifyOffset(verifier,
VT_IN) &&
2551 verifier.VerifyTable(in()) &&
2552 VerifyOffset(verifier,
VT_OUT) &&
2553 verifier.VerifyTable(out()) &&
2554 VerifyOffset(verifier,
VT_SHAPE) &&
2555 verifier.VerifyVector(shape()) &&
2556 verifier.EndTable();
2562 ::flatbuffers::FlatBufferBuilder &
fbb_;
2564 void add_in(::flatbuffers::Offset<tt::target::TensorRef> in) {
2567 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
2570 void add_shape(::flatbuffers::Offset<::flatbuffers::Vector<uint32_t>> shape) {
2579 auto o = ::flatbuffers::Offset<RepeatOp>(end);
2585 ::flatbuffers::FlatBufferBuilder &_fbb,
2586 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
2587 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
2588 ::flatbuffers::Offset<::flatbuffers::Vector<uint32_t>> shape = 0) {
2593 return builder_.
Finish();
2602 ::flatbuffers::FlatBufferBuilder &_fbb,
2603 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
2604 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
2605 const std::vector<uint32_t> *shape =
nullptr) {
2606 auto shape__ = shape ? _fbb.CreateVector<uint32_t>(*shape) : 0;
2617 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
2624 const tt::target::TensorRef *
in()
const {
2625 return GetPointer<const tt::target::TensorRef *>(
VT_IN);
2627 const tt::target::TensorRef *
out()
const {
2628 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
2630 const ::flatbuffers::Vector<int64_t> *
begins()
const {
2631 return GetPointer<const ::flatbuffers::Vector<int64_t> *>(
VT_BEGINS);
2633 const ::flatbuffers::Vector<int64_t> *
ends()
const {
2634 return GetPointer<const ::flatbuffers::Vector<int64_t> *>(
VT_ENDS);
2636 const ::flatbuffers::Vector<int64_t> *
step()
const {
2637 return GetPointer<const ::flatbuffers::Vector<int64_t> *>(
VT_STEP);
2639 bool Verify(::flatbuffers::Verifier &verifier)
const {
2640 return VerifyTableStart(verifier) &&
2641 VerifyOffset(verifier,
VT_IN) &&
2642 verifier.VerifyTable(in()) &&
2643 VerifyOffset(verifier,
VT_OUT) &&
2644 verifier.VerifyTable(out()) &&
2646 verifier.VerifyVector(begins()) &&
2647 VerifyOffset(verifier,
VT_ENDS) &&
2648 verifier.VerifyVector(ends()) &&
2649 VerifyOffset(verifier,
VT_STEP) &&
2650 verifier.VerifyVector(step()) &&
2651 verifier.EndTable();
2657 ::flatbuffers::FlatBufferBuilder &
fbb_;
2659 void add_in(::flatbuffers::Offset<tt::target::TensorRef> in) {
2662 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
2665 void add_begins(::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> begins) {
2668 void add_ends(::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> ends) {
2671 void add_step(::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> step) {
2680 auto o = ::flatbuffers::Offset<SliceOp>(end);
2686 ::flatbuffers::FlatBufferBuilder &_fbb,
2687 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
2688 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
2689 ::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> begins = 0,
2690 ::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> ends = 0,
2691 ::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> step = 0) {
2698 return builder_.
Finish();
2707 ::flatbuffers::FlatBufferBuilder &_fbb,
2708 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
2709 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
2710 const std::vector<int64_t> *begins =
nullptr,
2711 const std::vector<int64_t> *ends =
nullptr,
2712 const std::vector<int64_t> *step =
nullptr) {
2713 auto begins__ = begins ? _fbb.CreateVector<int64_t>(*begins) : 0;
2714 auto ends__ = ends ? _fbb.CreateVector<int64_t>(*ends) : 0;
2715 auto step__ = step ? _fbb.CreateVector<int64_t>(*step) : 0;
2728 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
2734 const tt::target::TensorRef *
in0()
const {
2735 return GetPointer<const tt::target::TensorRef *>(
VT_IN0);
2737 const tt::target::TensorRef *
in1()
const {
2738 return GetPointer<const tt::target::TensorRef *>(
VT_IN1);
2740 const tt::target::TensorRef *
bias()
const {
2741 return GetPointer<const tt::target::TensorRef *>(
VT_BIAS);
2743 const tt::target::TensorRef *
out()
const {
2744 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
2746 bool Verify(::flatbuffers::Verifier &verifier)
const {
2747 return VerifyTableStart(verifier) &&
2748 VerifyOffset(verifier,
VT_IN0) &&
2749 verifier.VerifyTable(in0()) &&
2750 VerifyOffset(verifier,
VT_IN1) &&
2751 verifier.VerifyTable(in1()) &&
2752 VerifyOffset(verifier,
VT_BIAS) &&
2753 verifier.VerifyTable(bias()) &&
2754 VerifyOffset(verifier,
VT_OUT) &&
2755 verifier.VerifyTable(out()) &&
2756 verifier.EndTable();
2762 ::flatbuffers::FlatBufferBuilder &
fbb_;
2764 void add_in0(::flatbuffers::Offset<tt::target::TensorRef> in0) {
2767 void add_in1(::flatbuffers::Offset<tt::target::TensorRef> in1) {
2770 void add_bias(::flatbuffers::Offset<tt::target::TensorRef> bias) {
2773 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
2782 auto o = ::flatbuffers::Offset<LinearOp>(end);
2788 ::flatbuffers::FlatBufferBuilder &_fbb,
2789 ::flatbuffers::Offset<tt::target::TensorRef> in0 = 0,
2790 ::flatbuffers::Offset<tt::target::TensorRef> in1 = 0,
2791 ::flatbuffers::Offset<tt::target::TensorRef> bias = 0,
2792 ::flatbuffers::Offset<tt::target::TensorRef> out = 0) {
2798 return builder_.
Finish();
2809 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
2814 const tt::target::TensorRef *
in0()
const {
2815 return GetPointer<const tt::target::TensorRef *>(
VT_IN0);
2817 const tt::target::TensorRef *
in1()
const {
2818 return GetPointer<const tt::target::TensorRef *>(
VT_IN1);
2820 const tt::target::TensorRef *
out()
const {
2821 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
2823 bool Verify(::flatbuffers::Verifier &verifier)
const {
2824 return VerifyTableStart(verifier) &&
2825 VerifyOffset(verifier,
VT_IN0) &&
2826 verifier.VerifyTable(in0()) &&
2827 VerifyOffset(verifier,
VT_IN1) &&
2828 verifier.VerifyTable(in1()) &&
2829 VerifyOffset(verifier,
VT_OUT) &&
2830 verifier.VerifyTable(out()) &&
2831 verifier.EndTable();
2837 ::flatbuffers::FlatBufferBuilder &
fbb_;
2839 void add_in0(::flatbuffers::Offset<tt::target::TensorRef> in0) {
2842 void add_in1(::flatbuffers::Offset<tt::target::TensorRef> in1) {
2845 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
2854 auto o = ::flatbuffers::Offset<MatmulOp>(end);
2860 ::flatbuffers::FlatBufferBuilder &_fbb,
2861 ::flatbuffers::Offset<tt::target::TensorRef> in0 = 0,
2862 ::flatbuffers::Offset<tt::target::TensorRef> in1 = 0,
2863 ::flatbuffers::Offset<tt::target::TensorRef> out = 0) {
2868 return builder_.
Finish();
2879 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
2900 const tt::target::TensorRef *
input()
const {
2901 return GetPointer<const tt::target::TensorRef *>(
VT_INPUT);
2903 const tt::target::TensorRef *
weight()
const {
2904 return GetPointer<const tt::target::TensorRef *>(
VT_WEIGHT);
2906 const tt::target::TensorRef *
bias()
const {
2907 return GetPointer<const tt::target::TensorRef *>(
VT_BIAS);
2909 const tt::target::TensorRef *
out()
const {
2910 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
2912 const tt::target::DeviceRef *
device()
const {
2913 return GetPointer<const tt::target::DeviceRef *>(
VT_DEVICE);
2955 return GetField<uint32_t>(VT_GROUPS, 0);
2957 bool Verify(::flatbuffers::Verifier &verifier)
const {
2958 return VerifyTableStart(verifier) &&
2959 VerifyOffset(verifier,
VT_INPUT) &&
2960 verifier.VerifyTable(input()) &&
2962 verifier.VerifyTable(weight()) &&
2963 VerifyOffset(verifier,
VT_BIAS) &&
2964 verifier.VerifyTable(bias()) &&
2965 VerifyOffset(verifier,
VT_OUT) &&
2966 verifier.VerifyTable(out()) &&
2968 verifier.VerifyTable(device()) &&
2982 VerifyField<uint32_t>(verifier, VT_GROUPS, 4) &&
2983 verifier.EndTable();
2989 ::flatbuffers::FlatBufferBuilder &
fbb_;
2991 void add_input(::flatbuffers::Offset<tt::target::TensorRef> input) {
2994 void add_weight(::flatbuffers::Offset<tt::target::TensorRef> weight) {
2997 void add_bias(::flatbuffers::Offset<tt::target::TensorRef> bias) {
3000 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
3003 void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
3046 fbb_.AddElement<uint32_t>(Conv2dOp::VT_GROUPS, groups, 0);
3054 auto o = ::flatbuffers::Offset<Conv2dOp>(end);
3060 ::flatbuffers::FlatBufferBuilder &_fbb,
3061 ::flatbuffers::Offset<tt::target::TensorRef> input = 0,
3062 ::flatbuffers::Offset<tt::target::TensorRef> weight = 0,
3063 ::flatbuffers::Offset<tt::target::TensorRef> bias = 0,
3064 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
3065 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
3066 uint32_t in_channels = 0,
3067 uint32_t out_channels = 0,
3068 uint32_t batch_size = 0,
3069 uint32_t input_height = 0,
3070 uint32_t input_width = 0,
3071 uint32_t kernel_height = 0,
3072 uint32_t kernel_width = 0,
3073 uint32_t stride_height = 0,
3074 uint32_t stride_width = 0,
3075 uint32_t padding_height = 0,
3076 uint32_t padding_width = 0,
3077 uint32_t dilation_height = 0,
3078 uint32_t dilation_width = 0,
3079 uint32_t groups = 0) {
3100 return builder_.
Finish();
3111 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
3129 const tt::target::TensorRef *
input()
const {
3130 return GetPointer<const tt::target::TensorRef *>(
VT_INPUT);
3132 const tt::target::TensorRef *
weight()
const {
3133 return GetPointer<const tt::target::TensorRef *>(
VT_WEIGHT);
3135 const tt::target::TensorRef *
bias()
const {
3136 return GetPointer<const tt::target::TensorRef *>(
VT_BIAS);
3138 const tt::target::TensorRef *
out()
const {
3139 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
3141 const tt::target::DeviceRef *
device()
const {
3142 return GetPointer<const tt::target::DeviceRef *>(
VT_DEVICE);
3160 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_KERNEL_SIZE);
3162 const ::flatbuffers::Vector<int32_t> *
stride()
const {
3163 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_STRIDE);
3165 const ::flatbuffers::Vector<int32_t> *
padding()
const {
3166 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_PADDING);
3171 const ::flatbuffers::Vector<int32_t> *
dilation()
const {
3172 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_DILATION);
3175 return GetField<uint32_t>(VT_GROUPS, 0);
3177 bool Verify(::flatbuffers::Verifier &verifier)
const {
3178 return VerifyTableStart(verifier) &&
3179 VerifyOffset(verifier,
VT_INPUT) &&
3180 verifier.VerifyTable(input()) &&
3182 verifier.VerifyTable(weight()) &&
3183 VerifyOffset(verifier,
VT_BIAS) &&
3184 verifier.VerifyTable(bias()) &&
3185 VerifyOffset(verifier,
VT_OUT) &&
3186 verifier.VerifyTable(out()) &&
3188 verifier.VerifyTable(device()) &&
3195 verifier.VerifyVector(kernel_size()) &&
3197 verifier.VerifyVector(stride()) &&
3199 verifier.VerifyVector(padding()) &&
3201 verifier.VerifyVector(output_padding()) &&
3203 verifier.VerifyVector(dilation()) &&
3204 VerifyField<uint32_t>(verifier, VT_GROUPS, 4) &&
3205 verifier.EndTable();
3211 ::flatbuffers::FlatBufferBuilder &
fbb_;
3213 void add_input(::flatbuffers::Offset<tt::target::TensorRef> input) {
3216 void add_weight(::flatbuffers::Offset<tt::target::TensorRef> weight) {
3219 void add_bias(::flatbuffers::Offset<tt::target::TensorRef> bias) {
3222 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
3225 void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
3246 void add_stride(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> stride) {
3249 void add_padding(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> padding) {
3255 void add_dilation(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> dilation) {
3259 fbb_.AddElement<uint32_t>(ConvTranspose2dOp::VT_GROUPS, groups, 0);
3265 ::flatbuffers::Offset<ConvTranspose2dOp>
Finish() {
3267 auto o = ::flatbuffers::Offset<ConvTranspose2dOp>(end);
3273 ::flatbuffers::FlatBufferBuilder &_fbb,
3274 ::flatbuffers::Offset<tt::target::TensorRef> input = 0,
3275 ::flatbuffers::Offset<tt::target::TensorRef> weight = 0,
3276 ::flatbuffers::Offset<tt::target::TensorRef> bias = 0,
3277 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
3278 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
3279 uint32_t in_channels = 0,
3280 uint32_t out_channels = 0,
3281 uint32_t batch_size = 0,
3282 uint32_t input_height = 0,
3283 uint32_t input_width = 0,
3284 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> kernel_size = 0,
3285 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> stride = 0,
3286 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> padding = 0,
3287 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> output_padding = 0,
3288 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> dilation = 0,
3289 uint32_t groups = 0) {
3307 return builder_.
Finish();
3316 ::flatbuffers::FlatBufferBuilder &_fbb,
3317 ::flatbuffers::Offset<tt::target::TensorRef> input = 0,
3318 ::flatbuffers::Offset<tt::target::TensorRef> weight = 0,
3319 ::flatbuffers::Offset<tt::target::TensorRef> bias = 0,
3320 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
3321 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
3322 uint32_t in_channels = 0,
3323 uint32_t out_channels = 0,
3324 uint32_t batch_size = 0,
3325 uint32_t input_height = 0,
3326 uint32_t input_width = 0,
3327 const std::vector<int32_t> *kernel_size =
nullptr,
3328 const std::vector<int32_t> *stride =
nullptr,
3329 const std::vector<int32_t> *padding =
nullptr,
3330 const std::vector<int32_t> *output_padding =
nullptr,
3331 const std::vector<int32_t> *dilation =
nullptr,
3332 uint32_t groups = 0) {
3333 auto kernel_size__ = kernel_size ? _fbb.CreateVector<int32_t>(*kernel_size) : 0;
3334 auto stride__ = stride ? _fbb.CreateVector<int32_t>(*stride) : 0;
3335 auto padding__ = padding ? _fbb.CreateVector<int32_t>(*padding) : 0;
3336 auto output_padding__ = output_padding ? _fbb.CreateVector<int32_t>(*output_padding) : 0;
3337 auto dilation__ = dilation ? _fbb.CreateVector<int32_t>(*dilation) : 0;
3361 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
3379 const tt::target::TensorRef *
in()
const {
3380 return GetPointer<const tt::target::TensorRef *>(
VT_IN);
3382 const tt::target::TensorRef *
out()
const {
3383 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
3385 const tt::target::DeviceRef *
device()
const {
3386 return GetPointer<const tt::target::DeviceRef *>(
VT_DEVICE);
3427 bool Verify(::flatbuffers::Verifier &verifier)
const {
3428 return VerifyTableStart(verifier) &&
3429 VerifyOffset(verifier,
VT_IN) &&
3430 verifier.VerifyTable(in()) &&
3431 VerifyOffset(verifier,
VT_OUT) &&
3432 verifier.VerifyTable(out()) &&
3434 verifier.VerifyTable(device()) &&
3438 VerifyField<uint32_t>(verifier,
VT_CHANNELS, 4) &&
3448 verifier.EndTable();
3454 ::flatbuffers::FlatBufferBuilder &
fbb_;
3456 void add_in(::flatbuffers::Offset<tt::target::TensorRef> in) {
3459 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
3462 void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
3508 ::flatbuffers::Offset<MaxPool2dOp>
Finish() {
3510 auto o = ::flatbuffers::Offset<MaxPool2dOp>(end);
3516 ::flatbuffers::FlatBufferBuilder &_fbb,
3517 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
3518 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
3519 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
3520 uint32_t batch_size = 0,
3521 uint32_t input_height = 0,
3522 uint32_t input_width = 0,
3523 uint32_t channels = 0,
3524 uint32_t kernel_height = 0,
3525 uint32_t kernel_width = 0,
3526 uint32_t stride_height = 0,
3527 uint32_t stride_width = 0,
3528 uint32_t dilation_height = 0,
3529 uint32_t dilation_width = 0,
3530 bool ceil_mode =
false,
3531 uint32_t padding_height = 0,
3532 uint32_t padding_width = 0) {
3550 return builder_.
Finish();
3561 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
3565 const tt::target::TensorRef *
in()
const {
3566 return GetPointer<const tt::target::TensorRef *>(
VT_IN);
3569 return GetField<uint8_t>(VT_FORCE, 0) != 0;
3571 bool Verify(::flatbuffers::Verifier &verifier)
const {
3572 return VerifyTableStart(verifier) &&
3573 VerifyOffset(verifier,
VT_IN) &&
3574 verifier.VerifyTable(in()) &&
3575 VerifyField<uint8_t>(verifier, VT_FORCE, 1) &&
3576 verifier.EndTable();
3582 ::flatbuffers::FlatBufferBuilder &
fbb_;
3584 void add_in(::flatbuffers::Offset<tt::target::TensorRef> in) {
3588 fbb_.AddElement<uint8_t>(DeallocateOp::VT_FORCE,
static_cast<uint8_t
>(force), 0);
3594 ::flatbuffers::Offset<DeallocateOp>
Finish() {
3596 auto o = ::flatbuffers::Offset<DeallocateOp>(end);
3602 ::flatbuffers::FlatBufferBuilder &_fbb,
3603 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
3604 bool force =
false) {
3608 return builder_.
Finish();
3619 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
3626 const tt::target::TensorRef *
in()
const {
3627 return GetPointer<const tt::target::TensorRef *>(
VT_IN);
3629 const tt::target::TensorRef *
out()
const {
3630 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
3632 const tt::target::DeviceRef *
device()
const {
3633 return GetPointer<const tt::target::DeviceRef *>(
VT_DEVICE);
3636 return GetField<uint32_t>(
VT_DIM, 0);
3639 return GetField<uint32_t>(VT_NUM_LINKS, 0);
3641 bool Verify(::flatbuffers::Verifier &verifier)
const {
3642 return VerifyTableStart(verifier) &&
3643 VerifyOffset(verifier,
VT_IN) &&
3644 verifier.VerifyTable(in()) &&
3645 VerifyOffset(verifier,
VT_OUT) &&
3646 verifier.VerifyTable(out()) &&
3648 verifier.VerifyTable(device()) &&
3649 VerifyField<uint32_t>(verifier,
VT_DIM, 4) &&
3650 VerifyField<uint32_t>(verifier, VT_NUM_LINKS, 4) &&
3651 verifier.EndTable();
3657 ::flatbuffers::FlatBufferBuilder &
fbb_;
3659 void add_in(::flatbuffers::Offset<tt::target::TensorRef> in) {
3662 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
3665 void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
3672 fbb_.AddElement<uint32_t>(AllGatherOp::VT_NUM_LINKS, num_links, 0);
3678 ::flatbuffers::Offset<AllGatherOp>
Finish() {
3680 auto o = ::flatbuffers::Offset<AllGatherOp>(end);
3686 ::flatbuffers::FlatBufferBuilder &_fbb,
3687 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
3688 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
3689 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
3691 uint32_t num_links = 0) {
3698 return builder_.
Finish();
3709 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
3716 const tt::target::TensorRef *
in()
const {
3717 return GetPointer<const tt::target::TensorRef *>(
VT_IN);
3720 return GetPointer<const ::flatbuffers::Vector<int64_t> *>(
VT_PERMUTATION);
3728 const tt::target::TensorRef *
out()
const {
3729 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
3731 bool Verify(::flatbuffers::Verifier &verifier)
const {
3732 return VerifyTableStart(verifier) &&
3733 VerifyOffset(verifier,
VT_IN) &&
3734 verifier.VerifyTable(in()) &&
3736 verifier.VerifyVector(permutation()) &&
3738 verifier.VerifyTable(memory_config()) &&
3740 VerifyOffset(verifier,
VT_OUT) &&
3741 verifier.VerifyTable(out()) &&
3742 verifier.EndTable();
3748 ::flatbuffers::FlatBufferBuilder &
fbb_;
3750 void add_in(::flatbuffers::Offset<tt::target::TensorRef> in) {
3762 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
3771 auto o = ::flatbuffers::Offset<PermuteOp>(end);
3777 ::flatbuffers::FlatBufferBuilder &_fbb,
3778 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
3779 ::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> permutation = 0,
3780 ::flatbuffers::Offset<tt::target::MemoryConfigDesc> memory_config = 0,
3781 float pad_value = 0.0f,
3782 ::flatbuffers::Offset<tt::target::TensorRef> out = 0) {
3789 return builder_.
Finish();
3798 ::flatbuffers::FlatBufferBuilder &_fbb,
3799 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
3800 const std::vector<int64_t> *permutation =
nullptr,
3801 ::flatbuffers::Offset<tt::target::MemoryConfigDesc> memory_config = 0,
3802 float pad_value = 0.0f,
3803 ::flatbuffers::Offset<tt::target::TensorRef> out = 0) {
3804 auto permutation__ = permutation ? _fbb.CreateVector<int64_t>(*permutation) : 0;
3817 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
3825 const tt::target::TensorRef *
in()
const {
3826 return GetPointer<const tt::target::TensorRef *>(
VT_IN);
3828 const tt::target::TensorRef *
out()
const {
3829 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
3831 const tt::target::DeviceRef *
device()
const {
3832 return GetPointer<const tt::target::DeviceRef *>(
VT_DEVICE);
3841 return GetField<uint32_t>(VT_NUM_LINKS, 0);
3843 bool Verify(::flatbuffers::Verifier &verifier)
const {
3844 return VerifyTableStart(verifier) &&
3845 VerifyOffset(verifier,
VT_IN) &&
3846 verifier.VerifyTable(in()) &&
3847 VerifyOffset(verifier,
VT_OUT) &&
3848 verifier.VerifyTable(out()) &&
3850 verifier.VerifyTable(device()) &&
3852 VerifyField<uint32_t>(verifier,
VT_MATH_OP, 4) &&
3853 VerifyField<uint32_t>(verifier, VT_NUM_LINKS, 4) &&
3854 verifier.EndTable();
3860 ::flatbuffers::FlatBufferBuilder &
fbb_;
3862 void add_in(::flatbuffers::Offset<tt::target::TensorRef> in) {
3865 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
3868 void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
3878 fbb_.AddElement<uint32_t>(ReduceScatterOp::VT_NUM_LINKS, num_links, 0);
3884 ::flatbuffers::Offset<ReduceScatterOp>
Finish() {
3886 auto o = ::flatbuffers::Offset<ReduceScatterOp>(end);
3892 ::flatbuffers::FlatBufferBuilder &_fbb,
3893 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
3894 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
3895 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
3896 uint32_t scatter_split_dim = 0,
3897 uint32_t math_op = 0,
3898 uint32_t num_links = 0) {
3906 return builder_.
Finish();
3917 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
3925 const tt::target::TensorRef *
in()
const {
3926 return GetPointer<const tt::target::TensorRef *>(
VT_IN);
3928 const tt::target::TensorRef *
out()
const {
3929 return GetPointer<const tt::target::TensorRef *>(
VT_OUT);
3931 const tt::target::DeviceRef *
device()
const {
3932 return GetPointer<const tt::target::DeviceRef *>(
VT_DEVICE);
3941 return GetPointer<const ::flatbuffers::Vector<int64_t> *>(VT_SHARD_SHAPE);
3943 bool Verify(::flatbuffers::Verifier &verifier)
const {
3944 return VerifyTableStart(verifier) &&
3945 VerifyOffset(verifier,
VT_IN) &&
3946 verifier.VerifyTable(in()) &&
3947 VerifyOffset(verifier,
VT_OUT) &&
3948 verifier.VerifyTable(out()) &&
3950 verifier.VerifyTable(device()) &&
3953 VerifyOffset(verifier, VT_SHARD_SHAPE) &&
3954 verifier.VerifyVector(shard_shape()) &&
3955 verifier.EndTable();
3961 ::flatbuffers::FlatBufferBuilder &
fbb_;
3963 void add_in(::flatbuffers::Offset<tt::target::TensorRef> in) {
3966 void add_out(::flatbuffers::Offset<tt::target::TensorRef> out) {
3969 void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
3979 fbb_.AddOffset(MeshShardOp::VT_SHARD_SHAPE, shard_shape);
3985 ::flatbuffers::Offset<MeshShardOp>
Finish() {
3987 auto o = ::flatbuffers::Offset<MeshShardOp>(end);
3993 ::flatbuffers::FlatBufferBuilder &_fbb,
3994 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
3995 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
3996 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
3999 ::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> shard_shape = 0) {
4007 return builder_.
Finish();
4016 ::flatbuffers::FlatBufferBuilder &_fbb,
4017 ::flatbuffers::Offset<tt::target::TensorRef> in = 0,
4018 ::flatbuffers::Offset<tt::target::TensorRef> out = 0,
4019 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
4022 const std::vector<int64_t> *shard_shape =
nullptr) {
4023 auto shard_shape__ = shard_shape ? _fbb.CreateVector<int64_t>(*shard_shape) : 0;
4037 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
4047 return GetPointer<const void *>(
VT_TYPE);
4147 return GetPointer<const ::flatbuffers::String *>(
VT_DEBUG_INFO);
4150 return GetPointer<const ::flatbuffers::String *>(VT_LOC_INFO);
4152 bool Verify(::flatbuffers::Verifier &verifier)
const {
4153 return VerifyTableStart(verifier) &&
4155 VerifyOffset(verifier,
VT_TYPE) &&
4158 verifier.VerifyString(debug_info()) &&
4159 VerifyOffset(verifier, VT_LOC_INFO) &&
4160 verifier.VerifyString(loc_info()) &&
4161 verifier.EndTable();
4165 template<>
inline const tt::target::ttnn::GetDeviceOp *Operation::type_as<tt::target::ttnn::GetDeviceOp>()
const {
4166 return type_as_GetDeviceOp();
4169 template<>
inline const tt::target::ttnn::ToMemoryConfigOp *Operation::type_as<tt::target::ttnn::ToMemoryConfigOp>()
const {
4170 return type_as_ToMemoryConfigOp();
4173 template<>
inline const tt::target::ttnn::ToLayoutOp *Operation::type_as<tt::target::ttnn::ToLayoutOp>()
const {
4174 return type_as_ToLayoutOp();
4177 template<>
inline const tt::target::ttnn::TypecastOp *Operation::type_as<tt::target::ttnn::TypecastOp>()
const {
4178 return type_as_TypecastOp();
4181 template<>
inline const tt::target::ttnn::ToDeviceOp *Operation::type_as<tt::target::ttnn::ToDeviceOp>()
const {
4182 return type_as_ToDeviceOp();
4185 template<>
inline const tt::target::ttnn::FromDeviceOp *Operation::type_as<tt::target::ttnn::FromDeviceOp>()
const {
4186 return type_as_FromDeviceOp();
4189 template<>
inline const tt::target::ttnn::EmptyOp *Operation::type_as<tt::target::ttnn::EmptyOp>()
const {
4190 return type_as_EmptyOp();
4193 template<>
inline const tt::target::ttnn::OnesOp *Operation::type_as<tt::target::ttnn::OnesOp>()
const {
4194 return type_as_OnesOp();
4197 template<>
inline const tt::target::ttnn::FullOp *Operation::type_as<tt::target::ttnn::FullOp>()
const {
4198 return type_as_FullOp();
4201 template<>
inline const tt::target::ttnn::EltwiseOp *Operation::type_as<tt::target::ttnn::EltwiseOp>()
const {
4202 return type_as_EltwiseOp();
4205 template<>
inline const tt::target::ttnn::LinearOp *Operation::type_as<tt::target::ttnn::LinearOp>()
const {
4206 return type_as_LinearOp();
4209 template<>
inline const tt::target::ttnn::MatmulOp *Operation::type_as<tt::target::ttnn::MatmulOp>()
const {
4210 return type_as_MatmulOp();
4213 template<>
inline const tt::target::ttnn::ReductionOp *Operation::type_as<tt::target::ttnn::ReductionOp>()
const {
4214 return type_as_ReductionOp();
4217 template<>
inline const tt::target::ttnn::EmbeddingOp *Operation::type_as<tt::target::ttnn::EmbeddingOp>()
const {
4218 return type_as_EmbeddingOp();
4221 template<>
inline const tt::target::ttnn::EmbeddingBackwardOp *Operation::type_as<tt::target::ttnn::EmbeddingBackwardOp>()
const {
4222 return type_as_EmbeddingBackwardOp();
4225 template<>
inline const tt::target::ttnn::SoftmaxOp *Operation::type_as<tt::target::ttnn::SoftmaxOp>()
const {
4226 return type_as_SoftmaxOp();
4229 template<>
inline const tt::target::ttnn::TransposeOp *Operation::type_as<tt::target::ttnn::TransposeOp>()
const {
4230 return type_as_TransposeOp();
4233 template<>
inline const tt::target::ttnn::Conv2dOp *Operation::type_as<tt::target::ttnn::Conv2dOp>()
const {
4234 return type_as_Conv2dOp();
4237 template<>
inline const tt::target::ttnn::ConvTranspose2dOp *Operation::type_as<tt::target::ttnn::ConvTranspose2dOp>()
const {
4238 return type_as_ConvTranspose2dOp();
4241 template<>
inline const tt::target::ttnn::ConcatOp *Operation::type_as<tt::target::ttnn::ConcatOp>()
const {
4242 return type_as_ConcatOp();
4245 template<>
inline const tt::target::ttnn::ReshapeOp *Operation::type_as<tt::target::ttnn::ReshapeOp>()
const {
4246 return type_as_ReshapeOp();
4249 template<>
inline const tt::target::ttnn::SliceOp *Operation::type_as<tt::target::ttnn::SliceOp>()
const {
4250 return type_as_SliceOp();
4253 template<>
inline const tt::target::ttnn::MaxPool2dOp *Operation::type_as<tt::target::ttnn::MaxPool2dOp>()
const {
4254 return type_as_MaxPool2dOp();
4257 template<>
inline const tt::target::ttnn::DeallocateOp *Operation::type_as<tt::target::ttnn::DeallocateOp>()
const {
4258 return type_as_DeallocateOp();
4261 template<>
inline const tt::target::ttnn::AllGatherOp *Operation::type_as<tt::target::ttnn::AllGatherOp>()
const {
4262 return type_as_AllGatherOp();
4265 template<>
inline const tt::target::ttnn::ReduceScatterOp *Operation::type_as<tt::target::ttnn::ReduceScatterOp>()
const {
4266 return type_as_ReduceScatterOp();
4269 template<>
inline const tt::target::ttnn::MeshShardOp *Operation::type_as<tt::target::ttnn::MeshShardOp>()
const {
4270 return type_as_MeshShardOp();
4273 template<>
inline const tt::target::ttnn::ArangeOp *Operation::type_as<tt::target::ttnn::ArangeOp>()
const {
4274 return type_as_ArangeOp();
4277 template<>
inline const tt::target::ttnn::UpdateCacheOp *Operation::type_as<tt::target::ttnn::UpdateCacheOp>()
const {
4278 return type_as_UpdateCacheOp();
4281 template<>
inline const tt::target::ttnn::FillCacheOp *Operation::type_as<tt::target::ttnn::FillCacheOp>()
const {
4282 return type_as_FillCacheOp();
4285 template<>
inline const tt::target::ttnn::PermuteOp *Operation::type_as<tt::target::ttnn::PermuteOp>()
const {
4286 return type_as_PermuteOp();
4289 template<>
inline const tt::target::ttnn::RepeatOp *Operation::type_as<tt::target::ttnn::RepeatOp>()
const {
4290 return type_as_RepeatOp();
4295 ::flatbuffers::FlatBufferBuilder &
fbb_;
4307 fbb_.AddOffset(Operation::VT_LOC_INFO, loc_info);
4315 auto o = ::flatbuffers::Offset<Operation>(end);
4321 ::flatbuffers::FlatBufferBuilder &_fbb,
4323 ::flatbuffers::Offset<void> type = 0,
4324 ::flatbuffers::Offset<::flatbuffers::String> debug_info = 0,
4325 ::flatbuffers::Offset<::flatbuffers::String> loc_info = 0) {
4331 return builder_.
Finish();
4340 ::flatbuffers::FlatBufferBuilder &_fbb,
4342 ::flatbuffers::Offset<void> type = 0,
4343 const char *debug_info =
nullptr,
4344 const char *loc_info =
nullptr) {
4345 auto debug_info__ = debug_info ? _fbb.CreateString(debug_info) : 0;
4346 auto loc_info__ = loc_info ? _fbb.CreateString(loc_info) : 0;
4358 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
4365 const ::flatbuffers::String *
name()
const {
4366 return GetPointer<const ::flatbuffers::String *>(
VT_NAME);
4368 const ::flatbuffers::Vector<::flatbuffers::Offset<tt::target::TensorRef>> *
inputs()
const {
4369 return GetPointer<const ::flatbuffers::Vector<::flatbuffers::Offset<tt::target::TensorRef>> *>(
VT_INPUTS);
4371 const ::flatbuffers::Vector<::flatbuffers::Offset<tt::target::TensorRef>> *
outputs()
const {
4372 return GetPointer<const ::flatbuffers::Vector<::flatbuffers::Offset<tt::target::TensorRef>> *>(
VT_OUTPUTS);
4374 const ::flatbuffers::Vector<::flatbuffers::Offset<tt::target::ttnn::Operation>> *
operations()
const {
4375 return GetPointer<const ::flatbuffers::Vector<::flatbuffers::Offset<tt::target::ttnn::Operation>> *>(
VT_OPERATIONS);
4378 return GetPointer<const tt::target::DebugInfo *>(
VT_DEBUG_INFO);
4380 bool Verify(::flatbuffers::Verifier &verifier)
const {
4381 return VerifyTableStart(verifier) &&
4382 VerifyOffset(verifier,
VT_NAME) &&
4383 verifier.VerifyString(name()) &&
4385 verifier.VerifyVector(inputs()) &&
4386 verifier.VerifyVectorOfTables(inputs()) &&
4388 verifier.VerifyVector(outputs()) &&
4389 verifier.VerifyVectorOfTables(outputs()) &&
4391 verifier.VerifyVector(operations()) &&
4392 verifier.VerifyVectorOfTables(operations()) &&
4394 verifier.VerifyTable(debug_info()) &&
4395 verifier.EndTable();
4401 ::flatbuffers::FlatBufferBuilder &
fbb_;
4403 void add_name(::flatbuffers::Offset<::flatbuffers::String> name) {
4406 void add_inputs(::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<tt::target::TensorRef>>> inputs) {
4409 void add_outputs(::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<tt::target::TensorRef>>> outputs) {
4412 void add_operations(::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<tt::target::ttnn::Operation>>> operations) {
4424 auto o = ::flatbuffers::Offset<Program>(end);
4430 ::flatbuffers::FlatBufferBuilder &_fbb,
4431 ::flatbuffers::Offset<::flatbuffers::String> name = 0,
4432 ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<tt::target::TensorRef>>> inputs = 0,
4433 ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<tt::target::TensorRef>>> outputs = 0,
4434 ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<tt::target::ttnn::Operation>>> operations = 0,
4435 ::flatbuffers::Offset<tt::target::DebugInfo> debug_info = 0) {
4442 return builder_.
Finish();
4451 ::flatbuffers::FlatBufferBuilder &_fbb,
4452 const char *name =
nullptr,
4453 const std::vector<::flatbuffers::Offset<tt::target::TensorRef>> *inputs =
nullptr,
4454 const std::vector<::flatbuffers::Offset<tt::target::TensorRef>> *outputs =
nullptr,
4455 const std::vector<::flatbuffers::Offset<tt::target::ttnn::Operation>> *operations =
nullptr,
4456 ::flatbuffers::Offset<tt::target::DebugInfo> debug_info = 0) {
4457 auto name__ = name ? _fbb.CreateString(name) : 0;
4458 auto inputs__ = inputs ? _fbb.CreateVector<::flatbuffers::Offset<tt::target::TensorRef>>(*inputs) : 0;
4459 auto outputs__ = outputs ? _fbb.CreateVector<::flatbuffers::Offset<tt::target::TensorRef>>(*outputs) : 0;
4460 auto operations__ = operations ? _fbb.CreateVector<::flatbuffers::Offset<tt::target::ttnn::Operation>>(*operations) : 0;
4476 auto ptr =
reinterpret_cast<const tt::target::ttnn::ClampOpParams *
>(obj);
4477 return verifier.VerifyTable(ptr);
4480 auto ptr =
reinterpret_cast<const tt::target::ttnn::EltwiseOpWithFloatParams *
>(obj);
4481 return verifier.VerifyTable(ptr);
4483 default:
return true;
4487 inline bool VerifyEltwiseOpParamsVector(::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset<void>> *values, const ::flatbuffers::Vector<EltwiseOpParams> *types) {
4488 if (!values || !types)
return !values && !types;
4489 if (values->size() != types->size())
return false;
4490 for (::flatbuffers::uoffset_t i = 0; i < values->size(); ++i) {
4505 auto ptr =
reinterpret_cast<const tt::target::ttnn::GetDeviceOp *
>(obj);
4506 return verifier.VerifyTable(ptr);
4509 auto ptr =
reinterpret_cast<const tt::target::ttnn::ToMemoryConfigOp *
>(obj);
4510 return verifier.VerifyTable(ptr);
4513 auto ptr =
reinterpret_cast<const tt::target::ttnn::ToLayoutOp *
>(obj);
4514 return verifier.VerifyTable(ptr);
4517 auto ptr =
reinterpret_cast<const tt::target::ttnn::TypecastOp *
>(obj);
4518 return verifier.VerifyTable(ptr);
4521 auto ptr =
reinterpret_cast<const tt::target::ttnn::ToDeviceOp *
>(obj);
4522 return verifier.VerifyTable(ptr);
4525 auto ptr =
reinterpret_cast<const tt::target::ttnn::FromDeviceOp *
>(obj);
4526 return verifier.VerifyTable(ptr);
4529 auto ptr =
reinterpret_cast<const tt::target::ttnn::EmptyOp *
>(obj);
4530 return verifier.VerifyTable(ptr);
4533 auto ptr =
reinterpret_cast<const tt::target::ttnn::OnesOp *
>(obj);
4534 return verifier.VerifyTable(ptr);
4537 auto ptr =
reinterpret_cast<const tt::target::ttnn::FullOp *
>(obj);
4538 return verifier.VerifyTable(ptr);
4541 auto ptr =
reinterpret_cast<const tt::target::ttnn::EltwiseOp *
>(obj);
4542 return verifier.VerifyTable(ptr);
4545 auto ptr =
reinterpret_cast<const tt::target::ttnn::LinearOp *
>(obj);
4546 return verifier.VerifyTable(ptr);
4549 auto ptr =
reinterpret_cast<const tt::target::ttnn::MatmulOp *
>(obj);
4550 return verifier.VerifyTable(ptr);
4553 auto ptr =
reinterpret_cast<const tt::target::ttnn::ReductionOp *
>(obj);
4554 return verifier.VerifyTable(ptr);
4557 auto ptr =
reinterpret_cast<const tt::target::ttnn::EmbeddingOp *
>(obj);
4558 return verifier.VerifyTable(ptr);
4561 auto ptr =
reinterpret_cast<const tt::target::ttnn::EmbeddingBackwardOp *
>(obj);
4562 return verifier.VerifyTable(ptr);
4565 auto ptr =
reinterpret_cast<const tt::target::ttnn::SoftmaxOp *
>(obj);
4566 return verifier.VerifyTable(ptr);
4569 auto ptr =
reinterpret_cast<const tt::target::ttnn::TransposeOp *
>(obj);
4570 return verifier.VerifyTable(ptr);
4573 auto ptr =
reinterpret_cast<const tt::target::ttnn::Conv2dOp *
>(obj);
4574 return verifier.VerifyTable(ptr);
4577 auto ptr =
reinterpret_cast<const tt::target::ttnn::ConvTranspose2dOp *
>(obj);
4578 return verifier.VerifyTable(ptr);
4581 auto ptr =
reinterpret_cast<const tt::target::ttnn::ConcatOp *
>(obj);
4582 return verifier.VerifyTable(ptr);
4585 auto ptr =
reinterpret_cast<const tt::target::ttnn::ReshapeOp *
>(obj);
4586 return verifier.VerifyTable(ptr);
4589 auto ptr =
reinterpret_cast<const tt::target::ttnn::SliceOp *
>(obj);
4590 return verifier.VerifyTable(ptr);
4593 auto ptr =
reinterpret_cast<const tt::target::ttnn::MaxPool2dOp *
>(obj);
4594 return verifier.VerifyTable(ptr);
4597 auto ptr =
reinterpret_cast<const tt::target::ttnn::DeallocateOp *
>(obj);
4598 return verifier.VerifyTable(ptr);
4601 auto ptr =
reinterpret_cast<const tt::target::ttnn::AllGatherOp *
>(obj);
4602 return verifier.VerifyTable(ptr);
4605 auto ptr =
reinterpret_cast<const tt::target::ttnn::ReduceScatterOp *
>(obj);
4606 return verifier.VerifyTable(ptr);
4609 auto ptr =
reinterpret_cast<const tt::target::ttnn::MeshShardOp *
>(obj);
4610 return verifier.VerifyTable(ptr);
4613 auto ptr =
reinterpret_cast<const tt::target::ttnn::ArangeOp *
>(obj);
4614 return verifier.VerifyTable(ptr);
4617 auto ptr =
reinterpret_cast<const tt::target::ttnn::UpdateCacheOp *
>(obj);
4618 return verifier.VerifyTable(ptr);
4621 auto ptr =
reinterpret_cast<const tt::target::ttnn::FillCacheOp *
>(obj);
4622 return verifier.VerifyTable(ptr);
4625 auto ptr =
reinterpret_cast<const tt::target::ttnn::PermuteOp *
>(obj);
4626 return verifier.VerifyTable(ptr);
4629 auto ptr =
reinterpret_cast<const tt::target::ttnn::RepeatOp *
>(obj);
4630 return verifier.VerifyTable(ptr);
4632 default:
return true;
4636 inline bool VerifyOpTypeVector(::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset<void>> *values, const ::flatbuffers::Vector<OpType> *types) {
4637 if (!values || !types)
return !values && !types;
4638 if (values->size() != types->size())
return false;
4639 for (::flatbuffers::uoffset_t i = 0; i < values->size(); ++i) {
4641 verifier, values->Get(i), types->GetEnum<
OpType>(i))) {
VT_LAYOUT
Definition: program_generated.h:792
VT_IN0
Definition: program_generated.h:721
VT_KERNEL_WIDTH
Definition: program_generated.h:2891
VT_INPUT
Definition: program_generated.h:1043
VT_DTYPE
Definition: program_generated.h:793
VT_CACHE
Definition: program_generated.h:1042
VT_NUM_SHARDS
Definition: program_generated.h:1253
VT_SCATTER_SPLIT_DIM
Definition: program_generated.h:3821
VT_STRATEGY
Definition: program_generated.h:1256
VT_SHAPE
Definition: program_generated.h:1250
VT_BEGINS
Definition: program_generated.h:2620
VT_DILATION
Definition: program_generated.h:3126
VT_PARAMS_TYPE
Definition: program_generated.h:1822
VT_START
Definition: program_generated.h:1605
VT_IN_GRAD
Definition: program_generated.h:2121
VT_PADDING
Definition: program_generated.h:3124
VT_IN1
Definition: program_generated.h:2730
VT_STRIDE_HEIGHT
Definition: program_generated.h:2892
VT_WEIGHT
Definition: program_generated.h:2050
VT_DIM0
Definition: program_generated.h:2292
VT_STRIDE
Definition: program_generated.h:3123
VT_CHANNELS
Definition: program_generated.h:3368
VT_KERNEL_SIZE
Definition: program_generated.h:3122
VT_KERNEL_HEIGHT
Definition: program_generated.h:2890
VT_BATCH_SIZE
Definition: program_generated.h:2887
VT_DIM
Definition: program_generated.h:3623
VT_MESH
Definition: program_generated.h:639
VT_DIM_ARG
Definition: program_generated.h:1945
VT_PERMUTATION
Definition: program_generated.h:3711
VT_MATH_OP
Definition: program_generated.h:3822
VT_TYPE_TYPE
Definition: program_generated.h:4038
VT_MIN
Definition: program_generated.h:1715
VT_DEBUG_INFO
Definition: program_generated.h:4040
VT_PADDING_WIDTH
Definition: program_generated.h:2895
VT_STRIDE_WIDTH
Definition: program_generated.h:2893
VT_INPUTS
Definition: program_generated.h:2369
VT_CHIP_IDS
Definition: program_generated.h:640
VT_TYPE
Definition: program_generated.h:1819
VT_UPDATE_INDEX
Definition: program_generated.h:1044
VT_INPUT_HEIGHT
Definition: program_generated.h:2888
VT_IN_CHANNELS
Definition: program_generated.h:2885
VT_OUTPUTS
Definition: program_generated.h:4361
VT_IN
Definition: program_generated.h:791
VT_PAD_VALUE
Definition: program_generated.h:3713
VT_SHARD_DIRECTION
Definition: program_generated.h:3921
VT_DEVICE
Definition: program_generated.h:795
VT_DILATION_WIDTH
Definition: program_generated.h:2897
VT_END
Definition: program_generated.h:1606
VT_FILL_VALUE
Definition: program_generated.h:1516
VT_INPUT_WIDTH
Definition: program_generated.h:2889
VT_ENDS
Definition: program_generated.h:2621
VT_DILATION_HEIGHT
Definition: program_generated.h:2896
VT_NAME
Definition: program_generated.h:4359
VT_BIAS
Definition: program_generated.h:2731
VT_OUT
Definition: program_generated.h:1821
VT_OUTPUT_PADDING
Definition: program_generated.h:3125
VT_PADDING_HEIGHT
Definition: program_generated.h:2894
VT_OPERATIONS
Definition: program_generated.h:4362
VT_OUT_CHANNELS
Definition: program_generated.h:2886
VT_INS
Definition: program_generated.h:1820
VT_MEMCFG
Definition: program_generated.h:722
VT_STEP
Definition: program_generated.h:1607
VT_SHARD_TYPE
Definition: program_generated.h:3922
VT_CEIL_MODE
Definition: program_generated.h:3375
VT_MEMORY_CONFIG
Definition: program_generated.h:3712
EltwiseOpType
Definition: program_generated.h:131
inline ::flatbuffers::Offset< SliceOp > CreateSliceOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> begins=0, ::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> ends=0, ::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> step=0)
Definition: program_generated.h:2685
bool VerifyOpTypeVector(::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset< void >> *values, const ::flatbuffers::Vector< OpType > *types)
Definition: program_generated.h:4636
inline ::flatbuffers::Offset< Program > CreateProgram(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset<::flatbuffers::String > name=0, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> inputs=0, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> outputs=0, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::Operation >>> operations=0, ::flatbuffers::Offset< tt::target::DebugInfo > debug_info=0)
Definition: program_generated.h:4429
inline ::flatbuffers::Offset< SliceOp > CreateSliceOpDirect(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, const std::vector< int64_t > *begins=nullptr, const std::vector< int64_t > *ends=nullptr, const std::vector< int64_t > *step=nullptr)
Definition: program_generated.h:2706
inline ::flatbuffers::Offset< Program > CreateProgramDirect(::flatbuffers::FlatBufferBuilder &_fbb, const char *name=nullptr, const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> *inputs=nullptr, const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> *outputs=nullptr, const std::vector<::flatbuffers::Offset< tt::target::ttnn::Operation >> *operations=nullptr, ::flatbuffers::Offset< tt::target::DebugInfo > debug_info=0)
Definition: program_generated.h:4450
inline ::flatbuffers::Offset< EltwiseOp > CreateEltwiseOp(::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::EltwiseOpType type=tt::target::ttnn::EltwiseOpType::Add, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> ins=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, tt::target::ttnn::EltwiseOpParams params_type=tt::target::ttnn::EltwiseOpParams::NONE, ::flatbuffers::Offset< void > params=0)
Definition: program_generated.h:1900
EltwiseOpParams
Definition: program_generated.h:296
@ EltwiseOpWithFloatParams
inline ::flatbuffers::Offset< ReductionOp > CreateReductionOp(::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::ReductionOpType type=tt::target::ttnn::ReductionOpType::Sum, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> dim_arg=0, bool keep_dim=false)
Definition: program_generated.h:2007
inline ::flatbuffers::Offset< TransposeOp > CreateTransposeOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, int32_t dim0=0, int32_t dim1=0)
Definition: program_generated.h:2346
const char * EnumNameEltwiseOpType(EltwiseOpType e)
Definition: program_generated.h:290
inline ::flatbuffers::Offset< MatmulOp > CreateMatmulOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in0=0, ::flatbuffers::Offset< tt::target::TensorRef > in1=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
Definition: program_generated.h:2859
const EltwiseOpType(& EnumValuesEltwiseOpType())[47]
Definition: program_generated.h:183
ReductionOpType
Definition: program_generated.h:344
inline ::flatbuffers::Offset< ReduceScatterOp > CreateReduceScatterOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, uint32_t scatter_split_dim=0, uint32_t math_op=0, uint32_t num_links=0)
Definition: program_generated.h:3891
OpType
Definition: program_generated.h:377
inline ::flatbuffers::Offset< UpdateCacheOp > CreateUpdateCacheOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > cache=0, ::flatbuffers::Offset< tt::target::TensorRef > input=0, ::flatbuffers::Offset< tt::target::TensorRef > update_index=0, uint32_t batch_offset=0)
Definition: program_generated.h:1099
inline ::flatbuffers::Offset< GetDeviceOp > CreateGetDeviceOp(::flatbuffers::FlatBufferBuilder &_fbb, const tt::target::Dim2d *mesh=nullptr, ::flatbuffers::Offset<::flatbuffers::Vector< uint32_t >> chip_ids=0, ::flatbuffers::Offset< tt::target::DeviceRef > out=0)
Definition: program_generated.h:687
inline ::flatbuffers::Offset< AllGatherOp > CreateAllGatherOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, uint32_t dim=0, uint32_t num_links=0)
Definition: program_generated.h:3685
const char *const * EnumNamesEltwiseOpParams()
Definition: program_generated.h:313
inline ::flatbuffers::Offset< FillCacheOp > CreateFillCacheOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > cache=0, ::flatbuffers::Offset< tt::target::TensorRef > input=0, uint32_t batch_offset=0)
Definition: program_generated.h:1170
const char *const * EnumNamesReductionOpType()
Definition: program_generated.h:361
const char * EnumNameReductionOpType(ReductionOpType e)
Definition: program_generated.h:371
inline ::flatbuffers::Offset< RepeatOp > CreateRepeatOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset<::flatbuffers::Vector< uint32_t >> shape=0)
Definition: program_generated.h:2584
const char *const * EnumNamesEltwiseOpType()
Definition: program_generated.h:236
inline ::flatbuffers::Offset< ToLayoutOp > CreateToLayoutOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, tt::target::TensorLayout layout=tt::target::TensorLayout::RowMajor, ::flatbuffers::Optional< tt::target::DataType > dtype=::flatbuffers::nullopt, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
Definition: program_generated.h:865
inline ::flatbuffers::Offset< GetDeviceOp > CreateGetDeviceOpDirect(::flatbuffers::FlatBufferBuilder &_fbb, const tt::target::Dim2d *mesh=nullptr, const std::vector< uint32_t > *chip_ids=nullptr, ::flatbuffers::Offset< tt::target::DeviceRef > out=0)
Definition: program_generated.h:704
inline ::flatbuffers::Offset< ConvTranspose2dOp > CreateConvTranspose2dOpDirect(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > input=0, ::flatbuffers::Offset< tt::target::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::TensorRef > bias=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, uint32_t in_channels=0, uint32_t out_channels=0, uint32_t batch_size=0, uint32_t input_height=0, uint32_t input_width=0, const std::vector< int32_t > *kernel_size=nullptr, const std::vector< int32_t > *stride=nullptr, const std::vector< int32_t > *padding=nullptr, const std::vector< int32_t > *output_padding=nullptr, const std::vector< int32_t > *dilation=nullptr, uint32_t groups=0)
Definition: program_generated.h:3315
const ReductionOpType(& EnumValuesReductionOpType())[3]
Definition: program_generated.h:352
inline ::flatbuffers::Offset< ClampOpParams > CreateClampOpParams(::flatbuffers::FlatBufferBuilder &_fbb, float min=0.0f, float max=0.0f)
Definition: program_generated.h:1753
inline ::flatbuffers::Offset< FromDeviceOp > CreateFromDeviceOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
Definition: program_generated.h:1231
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::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::DistributionStrategy > strategy=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
Definition: program_generated.h:1341
inline ::flatbuffers::Offset< LinearOp > CreateLinearOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in0=0, ::flatbuffers::Offset< tt::target::TensorRef > in1=0, ::flatbuffers::Offset< tt::target::TensorRef > bias=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
Definition: program_generated.h:2787
inline ::flatbuffers::Offset< Conv2dOp > CreateConv2dOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > input=0, ::flatbuffers::Offset< tt::target::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::TensorRef > bias=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, uint32_t in_channels=0, uint32_t out_channels=0, uint32_t batch_size=0, uint32_t input_height=0, uint32_t input_width=0, uint32_t kernel_height=0, uint32_t kernel_width=0, uint32_t stride_height=0, uint32_t stride_width=0, uint32_t padding_height=0, uint32_t padding_width=0, uint32_t dilation_height=0, uint32_t dilation_width=0, uint32_t groups=0)
Definition: program_generated.h:3059
inline ::flatbuffers::Offset< MeshShardOp > CreateMeshShardOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, tt::target::MeshShardDirection shard_direction=tt::target::MeshShardDirection::FullToShardShape, tt::target::MeshShardType shard_type=tt::target::MeshShardType::Manual, ::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> shard_shape=0)
Definition: program_generated.h:3992
const EltwiseOpParams(& EnumValuesEltwiseOpParams())[3]
Definition: program_generated.h:304
const char *const * EnumNamesOpType()
Definition: program_generated.h:454
inline ::flatbuffers::Offset< ToMemoryConfigOp > CreateToMemoryConfigOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in0=0, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
Definition: program_generated.h:770
const char * EnumNameEltwiseOpParams(EltwiseOpParams e)
Definition: program_generated.h:323
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::DistributionStrategy > strategy=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
Definition: program_generated.h:1580
inline ::flatbuffers::Offset< PermuteOp > CreatePermuteOpDirect(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, const std::vector< int64_t > *permutation=nullptr, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memory_config=0, float pad_value=0.0f, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
Definition: program_generated.h:3797
inline ::flatbuffers::Offset< ConvTranspose2dOp > CreateConvTranspose2dOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > input=0, ::flatbuffers::Offset< tt::target::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::TensorRef > bias=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, uint32_t in_channels=0, uint32_t out_channels=0, uint32_t batch_size=0, uint32_t input_height=0, uint32_t input_width=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> kernel_size=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> stride=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> padding=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> output_padding=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> dilation=0, uint32_t groups=0)
Definition: program_generated.h:3272
const char * EnumNameOpType(OpType e)
Definition: program_generated.h:494
bool VerifyOpType(::flatbuffers::Verifier &verifier, const void *obj, OpType type)
Definition: program_generated.h:4499
inline ::flatbuffers::Offset< Operation > CreateOperation(::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::OpType type_type=tt::target::ttnn::OpType::NONE, ::flatbuffers::Offset< void > type=0, ::flatbuffers::Offset<::flatbuffers::String > debug_info=0, ::flatbuffers::Offset<::flatbuffers::String > loc_info=0)
Definition: program_generated.h:4320
inline ::flatbuffers::Offset< ConcatOp > CreateConcatOpDirect(::flatbuffers::FlatBufferBuilder &_fbb, const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> *inputs=nullptr, ::flatbuffers::Offset< tt::target::TensorRef > out=0, int32_t dim=0)
Definition: program_generated.h:2435
inline ::flatbuffers::Offset< ReshapeOp > CreateReshapeOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> shape=0)
Definition: program_generated.h:2501
inline ::flatbuffers::Offset< OnesOp > CreateOnesOpDirect(::flatbuffers::FlatBufferBuilder &_fbb, 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::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
Definition: program_generated.h:1492
inline ::flatbuffers::Offset< ReshapeOp > CreateReshapeOpDirect(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, const std::vector< int32_t > *shape=nullptr)
Definition: program_generated.h:2518
inline ::flatbuffers::Offset< ConcatOp > CreateConcatOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> inputs=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, int32_t dim=0)
Definition: program_generated.h:2418
inline ::flatbuffers::Offset< TypecastOp > CreateTypecastOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, tt::target::DataType dtype=tt::target::DataType::Float32, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
Definition: program_generated.h:940
bool VerifyEltwiseOpParams(::flatbuffers::Verifier &verifier, const void *obj, EltwiseOpParams type)
Definition: program_generated.h:4470
inline ::flatbuffers::Offset< OnesOp > CreateOnesOp(::flatbuffers::FlatBufferBuilder &_fbb, ::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::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
Definition: program_generated.h:1469
inline ::flatbuffers::Offset< RepeatOp > CreateRepeatOpDirect(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, const std::vector< uint32_t > *shape=nullptr)
Definition: program_generated.h:2601
inline ::flatbuffers::Offset< ReductionOp > CreateReductionOpDirect(::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::ReductionOpType type=tt::target::ttnn::ReductionOpType::Sum, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, const std::vector< int32_t > *dim_arg=nullptr, bool keep_dim=false)
Definition: program_generated.h:2028
inline ::flatbuffers::Offset< MaxPool2dOp > CreateMaxPool2dOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, uint32_t batch_size=0, uint32_t input_height=0, uint32_t input_width=0, uint32_t channels=0, uint32_t kernel_height=0, uint32_t kernel_width=0, uint32_t stride_height=0, uint32_t stride_width=0, uint32_t dilation_height=0, uint32_t dilation_width=0, bool ceil_mode=false, uint32_t padding_height=0, uint32_t padding_width=0)
Definition: program_generated.h:3515
inline ::flatbuffers::Offset< EltwiseOpWithFloatParams > CreateEltwiseOpWithFloatParams(::flatbuffers::FlatBufferBuilder &_fbb, float parameter=0.0f)
Definition: program_generated.h:1802
inline ::flatbuffers::Offset< EmbeddingOp > CreateEmbeddingOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > input=0, ::flatbuffers::Offset< tt::target::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
Definition: program_generated.h:2098
bool VerifyEltwiseOpParamsVector(::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset< void >> *values, const ::flatbuffers::Vector< EltwiseOpParams > *types)
Definition: program_generated.h:4487
inline ::flatbuffers::Offset< ToDeviceOp > CreateToDeviceOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
Definition: program_generated.h:1019
inline ::flatbuffers::Offset< EmbeddingBackwardOp > CreateEmbeddingBackwardOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > input=0, ::flatbuffers::Offset< tt::target::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::TensorRef > in_grad=0, ::flatbuffers::Optional< tt::target::DataType > dtype=::flatbuffers::nullopt, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
Definition: program_generated.h:2194
inline ::flatbuffers::Offset< MeshShardOp > CreateMeshShardOpDirect(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, tt::target::MeshShardDirection shard_direction=tt::target::MeshShardDirection::FullToShardShape, tt::target::MeshShardType shard_type=tt::target::MeshShardType::Manual, const std::vector< int64_t > *shard_shape=nullptr)
Definition: program_generated.h:4015
inline ::flatbuffers::Offset< PermuteOp > CreatePermuteOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> permutation=0, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memory_config=0, float pad_value=0.0f, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
Definition: program_generated.h:3776
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::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
Definition: program_generated.h:1686
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::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::DistributionStrategy > strategy=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0)
Definition: program_generated.h:1368
inline ::flatbuffers::Offset< DeallocateOp > CreateDeallocateOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, bool force=false)
Definition: program_generated.h:3601
inline ::flatbuffers::Offset< SoftmaxOp > CreateSoftmaxOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, int32_t dimension=0)
Definition: program_generated.h:2269
inline ::flatbuffers::Offset< EltwiseOp > CreateEltwiseOpDirect(::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::EltwiseOpType type=tt::target::ttnn::EltwiseOpType::Add, const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> *ins=nullptr, ::flatbuffers::Offset< tt::target::TensorRef > out=0, tt::target::ttnn::EltwiseOpParams params_type=tt::target::ttnn::EltwiseOpParams::NONE, ::flatbuffers::Offset< void > params=0)
Definition: program_generated.h:1921
const OpType(& EnumValuesOpType())[33]
Definition: program_generated.h:415
inline ::flatbuffers::Offset< Operation > CreateOperationDirect(::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::OpType type_type=tt::target::ttnn::OpType::NONE, ::flatbuffers::Offset< void > type=0, const char *debug_info=nullptr, const char *loc_info=nullptr)
Definition: program_generated.h:4339
TensorLayout
Definition: types_generated.h:325
MeshShardType
Definition: types_generated.h:427
MeshShardDirection
Definition: types_generated.h:397
DataType
Definition: types_generated.h:117
Definition: debug_info_generated.h:18
Definition: debug_info_generated.h:36
Definition: program_generated.h:3655
::flatbuffers::Offset< AllGatherOp > Finish()
Definition: program_generated.h:3678
void add_num_links(uint32_t num_links)
Definition: program_generated.h:3671
AllGatherOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:3674
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: program_generated.h:3665
void add_dim(uint32_t dim)
Definition: program_generated.h:3668
AllGatherOp Table
Definition: program_generated.h:3656
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:3662
::flatbuffers::uoffset_t start_
Definition: program_generated.h:3658
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:3657
void add_in(::flatbuffers::Offset< tt::target::TensorRef > in)
Definition: program_generated.h:3659
Definition: program_generated.h:3701
static constexpr auto Create
Definition: program_generated.h:3703
AllGatherOp type
Definition: program_generated.h:3702
Definition: program_generated.h:1650
ArangeOp Table
Definition: program_generated.h:1651
void add_start(float start)
Definition: program_generated.h:1654
void add_dtype(tt::target::DataType dtype)
Definition: program_generated.h:1663
void add_step(float step)
Definition: program_generated.h:1660
::flatbuffers::Offset< ArangeOp > Finish()
Definition: program_generated.h:1679
ArangeOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:1675
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:1652
void add_memcfg(::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg)
Definition: program_generated.h:1669
::flatbuffers::uoffset_t start_
Definition: program_generated.h:1653
void add_end(float end)
Definition: program_generated.h:1657
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: program_generated.h:1666
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:1672
Definition: program_generated.h:1706
ArangeOp type
Definition: program_generated.h:1707
static constexpr auto Create
Definition: program_generated.h:1708
Definition: program_generated.h:1732
void add_max(float max)
Definition: program_generated.h:1739
ClampOpParams Table
Definition: program_generated.h:1733
void add_min(float min)
Definition: program_generated.h:1736
::flatbuffers::uoffset_t start_
Definition: program_generated.h:1735
::flatbuffers::Offset< ClampOpParams > Finish()
Definition: program_generated.h:1746
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:1734
ClampOpParamsBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:1742
Definition: program_generated.h:1763
ClampOpParams type
Definition: program_generated.h:1764
static constexpr auto Create
Definition: program_generated.h:1765
Definition: program_generated.h:2394
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:2396
void add_dim(int32_t dim)
Definition: program_generated.h:2404
::flatbuffers::Offset< ConcatOp > Finish()
Definition: program_generated.h:2411
ConcatOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:2407
void add_inputs(::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> inputs)
Definition: program_generated.h:2398
::flatbuffers::uoffset_t start_
Definition: program_generated.h:2397
ConcatOp Table
Definition: program_generated.h:2395
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:2401
Definition: program_generated.h:2430
ConcatOp type
Definition: program_generated.h:2431
static constexpr auto Create
Definition: program_generated.h:2432
Definition: program_generated.h:2987
void add_in_channels(uint32_t in_channels)
Definition: program_generated.h:3006
void add_input_width(uint32_t input_width)
Definition: program_generated.h:3018
void add_input(::flatbuffers::Offset< tt::target::TensorRef > input)
Definition: program_generated.h:2991
Conv2dOp Table
Definition: program_generated.h:2988
void add_out_channels(uint32_t out_channels)
Definition: program_generated.h:3009
::flatbuffers::Offset< Conv2dOp > Finish()
Definition: program_generated.h:3052
void add_stride_width(uint32_t stride_width)
Definition: program_generated.h:3030
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: program_generated.h:3003
Conv2dOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:3048
void add_groups(uint32_t groups)
Definition: program_generated.h:3045
void add_padding_width(uint32_t padding_width)
Definition: program_generated.h:3036
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:3000
void add_dilation_width(uint32_t dilation_width)
Definition: program_generated.h:3042
void add_kernel_width(uint32_t kernel_width)
Definition: program_generated.h:3024
void add_kernel_height(uint32_t kernel_height)
Definition: program_generated.h:3021
::flatbuffers::uoffset_t start_
Definition: program_generated.h:2990
void add_padding_height(uint32_t padding_height)
Definition: program_generated.h:3033
void add_batch_size(uint32_t batch_size)
Definition: program_generated.h:3012
void add_bias(::flatbuffers::Offset< tt::target::TensorRef > bias)
Definition: program_generated.h:2997
void add_input_height(uint32_t input_height)
Definition: program_generated.h:3015
void add_stride_height(uint32_t stride_height)
Definition: program_generated.h:3027
void add_dilation_height(uint32_t dilation_height)
Definition: program_generated.h:3039
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:2989
void add_weight(::flatbuffers::Offset< tt::target::TensorRef > weight)
Definition: program_generated.h:2994
Definition: program_generated.h:3103
static constexpr auto Create
Definition: program_generated.h:3105
Conv2dOp type
Definition: program_generated.h:3104
Definition: program_generated.h:3209
void add_input(::flatbuffers::Offset< tt::target::TensorRef > input)
Definition: program_generated.h:3213
void add_groups(uint32_t groups)
Definition: program_generated.h:3258
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: program_generated.h:3225
ConvTranspose2dOp Table
Definition: program_generated.h:3210
void add_input_height(uint32_t input_height)
Definition: program_generated.h:3237
::flatbuffers::Offset< ConvTranspose2dOp > Finish()
Definition: program_generated.h:3265
ConvTranspose2dOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:3261
::flatbuffers::uoffset_t start_
Definition: program_generated.h:3212
void add_in_channels(uint32_t in_channels)
Definition: program_generated.h:3228
void add_stride(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> stride)
Definition: program_generated.h:3246
void add_batch_size(uint32_t batch_size)
Definition: program_generated.h:3234
void add_out_channels(uint32_t out_channels)
Definition: program_generated.h:3231
void add_kernel_size(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> kernel_size)
Definition: program_generated.h:3243
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:3222
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:3211
void add_padding(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> padding)
Definition: program_generated.h:3249
void add_weight(::flatbuffers::Offset< tt::target::TensorRef > weight)
Definition: program_generated.h:3216
void add_dilation(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> dilation)
Definition: program_generated.h:3255
void add_bias(::flatbuffers::Offset< tt::target::TensorRef > bias)
Definition: program_generated.h:3219
void add_output_padding(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> output_padding)
Definition: program_generated.h:3252
void add_input_width(uint32_t input_width)
Definition: program_generated.h:3240
Definition: program_generated.h:3310
ConvTranspose2dOp type
Definition: program_generated.h:3311
static constexpr auto Create
Definition: program_generated.h:3312
Definition: program_generated.h:3580
void add_in(::flatbuffers::Offset< tt::target::TensorRef > in)
Definition: program_generated.h:3584
DeallocateOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:3590
::flatbuffers::uoffset_t start_
Definition: program_generated.h:3583
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:3582
DeallocateOp Table
Definition: program_generated.h:3581
::flatbuffers::Offset< DeallocateOp > Finish()
Definition: program_generated.h:3594
void add_force(bool force)
Definition: program_generated.h:3587
Definition: program_generated.h:3611
DeallocateOp type
Definition: program_generated.h:3612
static constexpr auto Create
Definition: program_generated.h:3613
Definition: program_generated.h:1870
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:1880
::flatbuffers::Offset< EltwiseOp > Finish()
Definition: program_generated.h:1893
void add_params(::flatbuffers::Offset< void > params)
Definition: program_generated.h:1886
void add_params_type(tt::target::ttnn::EltwiseOpParams params_type)
Definition: program_generated.h:1883
EltwiseOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:1889
void add_ins(::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> ins)
Definition: program_generated.h:1877
::flatbuffers::uoffset_t start_
Definition: program_generated.h:1873
void add_type(tt::target::ttnn::EltwiseOpType type)
Definition: program_generated.h:1874
EltwiseOp Table
Definition: program_generated.h:1871
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:1872
Definition: program_generated.h:329
static const EltwiseOpParams enum_value
Definition: program_generated.h:330
Definition: program_generated.h:1784
::flatbuffers::Offset< EltwiseOpWithFloatParams > Finish()
Definition: program_generated.h:1795
::flatbuffers::uoffset_t start_
Definition: program_generated.h:1787
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:1786
EltwiseOpWithFloatParams Table
Definition: program_generated.h:1785
EltwiseOpWithFloatParamsBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:1791
void add_parameter(float parameter)
Definition: program_generated.h:1788
Definition: program_generated.h:1810
static constexpr auto Create
Definition: program_generated.h:1812
EltwiseOpWithFloatParams type
Definition: program_generated.h:1811
Definition: program_generated.h:1916
static constexpr auto Create
Definition: program_generated.h:1918
EltwiseOp type
Definition: program_generated.h:1917
Definition: program_generated.h:2161
::flatbuffers::Offset< EmbeddingBackwardOp > Finish()
Definition: program_generated.h:2187
void add_dtype(tt::target::DataType dtype)
Definition: program_generated.h:2174
void add_in_grad(::flatbuffers::Offset< tt::target::TensorRef > in_grad)
Definition: program_generated.h:2171
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:2180
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:2163
EmbeddingBackwardOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:2183
void add_input(::flatbuffers::Offset< tt::target::TensorRef > input)
Definition: program_generated.h:2165
void add_weight(::flatbuffers::Offset< tt::target::TensorRef > weight)
Definition: program_generated.h:2168
void add_memcfg(::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg)
Definition: program_generated.h:2177
EmbeddingBackwardOp Table
Definition: program_generated.h:2162
::flatbuffers::uoffset_t start_
Definition: program_generated.h:2164
Definition: program_generated.h:2212
static constexpr auto Create
Definition: program_generated.h:2214
EmbeddingBackwardOp type
Definition: program_generated.h:2213
Definition: program_generated.h:2074
::flatbuffers::uoffset_t start_
Definition: program_generated.h:2077
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:2084
EmbeddingOp Table
Definition: program_generated.h:2075
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:2076
::flatbuffers::Offset< EmbeddingOp > Finish()
Definition: program_generated.h:2091
EmbeddingOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:2087
void add_input(::flatbuffers::Offset< tt::target::TensorRef > input)
Definition: program_generated.h:2078
void add_weight(::flatbuffers::Offset< tt::target::TensorRef > weight)
Definition: program_generated.h:2081
Definition: program_generated.h:2110
EmbeddingOp type
Definition: program_generated.h:2111
static constexpr auto Create
Definition: program_generated.h:2112
Definition: program_generated.h:1302
EmptyOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:1330
void add_layout(tt::target::TensorLayout layout)
Definition: program_generated.h:1312
void add_shape(::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> shape)
Definition: program_generated.h:1306
::flatbuffers::Offset< EmptyOp > Finish()
Definition: program_generated.h:1334
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: program_generated.h:1318
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:1304
void add_memcfg(::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg)
Definition: program_generated.h:1321
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:1327
void add_strategy(::flatbuffers::Offset< tt::target::DistributionStrategy > strategy)
Definition: program_generated.h:1324
void add_dtype(tt::target::DataType dtype)
Definition: program_generated.h:1309
EmptyOp Table
Definition: program_generated.h:1303
void add_num_shards(uint32_t num_shards)
Definition: program_generated.h:1315
::flatbuffers::uoffset_t start_
Definition: program_generated.h:1305
Definition: program_generated.h:1363
static constexpr auto Create
Definition: program_generated.h:1365
EmptyOp type
Definition: program_generated.h:1364
Definition: binary_generated.h:30
const ::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef > > * outputs() const
Definition: program_generated.h:4371
tt::target::DataType dtype() const
Definition: program_generated.h:899
uint32_t padding_width() const
Definition: program_generated.h:2945
uint32_t in_channels() const
Definition: program_generated.h:2915
const tt::target::ttnn::ReshapeOp * type_as_ReshapeOp() const
Definition: program_generated.h:4110
UpdateCacheOpBuilder Builder
Definition: program_generated.h:1039
const ::flatbuffers::Vector< int64_t > * begins() const
Definition: program_generated.h:2630
const ::flatbuffers::Vector< uint32_t > * chip_ids() const
Definition: program_generated.h:646
const ::flatbuffers::String * name() const
Definition: program_generated.h:4365
uint32_t batch_size() const
Definition: program_generated.h:2921
DeallocateOpBuilder Builder
Definition: program_generated.h:3559
const tt::target::ttnn::SoftmaxOp * type_as_SoftmaxOp() const
Definition: program_generated.h:4095
ConvTranspose2dOpBuilder Builder
Definition: program_generated.h:3109
const tt::target::MemoryConfigDesc * memory_config() const
Definition: program_generated.h:3722
ReshapeOpBuilder Builder
Definition: program_generated.h:2449
const tt::target::TensorRef * in() const
Definition: program_generated.h:798
uint32_t input_width() const
Definition: program_generated.h:2927
const tt::target::ttnn::MaxPool2dOp * type_as_MaxPool2dOp() const
Definition: program_generated.h:4116
const ::flatbuffers::Vector< int64_t > * ends() const
Definition: program_generated.h:2633
int32_t dim() const
Definition: program_generated.h:2379
const tt::target::ttnn::ReductionOp * type_as_ReductionOp() const
Definition: program_generated.h:4086
const tt::target::TensorRef * in1() const
Definition: program_generated.h:2737
uint32_t num_links() const
Definition: program_generated.h:3638
ClampOpParamsBuilder Builder
Definition: program_generated.h:1712
::flatbuffers::Optional< tt::target::TensorLayout > layout() const
Definition: program_generated.h:1408
const tt::target::ttnn::EmbeddingOp * type_as_EmbeddingOp() const
Definition: program_generated.h:4089
ReduceScatterOpBuilder Builder
Definition: program_generated.h:3815
const tt::target::ttnn::ConcatOp * type_as_ConcatOp() const
Definition: program_generated.h:4107
const tt::target::TensorRef * weight() const
Definition: program_generated.h:2056
const tt::target::TensorRef * in0() const
Definition: program_generated.h:725
const void * type() const
Definition: program_generated.h:4046
ProgramBuilder Builder
Definition: program_generated.h:4356
uint32_t dim() const
Definition: program_generated.h:3635
const tt::target::TensorRef * bias() const
Definition: program_generated.h:2740
bool ceil_mode() const
Definition: program_generated.h:3418
EltwiseOpBuilder Builder
Definition: program_generated.h:1816
uint32_t dilation_height() const
Definition: program_generated.h:2948
float start() const
Definition: program_generated.h:1613
const ::flatbuffers::Vector< int64_t > * step() const
Definition: program_generated.h:2636
uint32_t batch_offset() const
Definition: program_generated.h:1056
tt::target::MeshShardType shard_type() const
Definition: program_generated.h:3937
const tt::target::TensorRef * update_index() const
Definition: program_generated.h:1053
const tt::target::ttnn::EmptyOp * type_as_EmptyOp() const
Definition: program_generated.h:4068
uint32_t channels() const
Definition: program_generated.h:3397
const ::flatbuffers::String * debug_info() const
Definition: program_generated.h:4146
int32_t dim1() const
Definition: program_generated.h:2304
const ::flatbuffers::Vector< int32_t > * shape() const
Definition: program_generated.h:2462
uint32_t kernel_height() const
Definition: program_generated.h:2930
const tt::target::ttnn::ClampOpParams * params_as_ClampOpParams() const
Definition: program_generated.h:1841
const tt::target::ttnn::SliceOp * type_as_SliceOp() const
Definition: program_generated.h:4113
const tt::target::ttnn::DeallocateOp * type_as_DeallocateOp() const
Definition: program_generated.h:4119
uint32_t kernel_width() const
Definition: program_generated.h:2933
uint32_t stride_width() const
Definition: program_generated.h:2939
int32_t dim0() const
Definition: program_generated.h:2301
const T * type_as() const
uint32_t num_shards() const
Definition: program_generated.h:1268
const tt::target::Dim2d * mesh() const
Definition: program_generated.h:643
const tt::target::ttnn::RepeatOp * type_as_RepeatOp() const
Definition: program_generated.h:4143
const tt::target::ttnn::MeshShardOp * type_as_MeshShardOp() const
Definition: program_generated.h:4128
uint32_t padding_height() const
Definition: program_generated.h:2942
const tt::target::ttnn::ToMemoryConfigOp * type_as_ToMemoryConfigOp() const
Definition: program_generated.h:4053
uint32_t dilation_width() const
Definition: program_generated.h:2951
const tt::target::ttnn::EmbeddingBackwardOp * type_as_EmbeddingBackwardOp() const
Definition: program_generated.h:4092
const tt::target::ttnn::ToLayoutOp * type_as_ToLayoutOp() const
Definition: program_generated.h:4056
EmbeddingOpBuilder Builder
Definition: program_generated.h:2046
const tt::target::ttnn::ArangeOp * type_as_ArangeOp() const
Definition: program_generated.h:4131
const tt::target::ttnn::FromDeviceOp * type_as_FromDeviceOp() const
Definition: program_generated.h:4065
const ::flatbuffers::Vector< int32_t > * dilation() const
Definition: program_generated.h:3171
const ::flatbuffers::Vector< int64_t > * shape() const
Definition: program_generated.h:1259
SliceOpBuilder Builder
Definition: program_generated.h:2615
GetDeviceOpBuilder Builder
Definition: program_generated.h:636
tt::target::TensorLayout layout() const
Definition: program_generated.h:801
const tt::target::ttnn::EltwiseOp * type_as_EltwiseOp() const
Definition: program_generated.h:4077
EmptyOpBuilder Builder
Definition: program_generated.h:1247
const tt::target::ttnn::EltwiseOpWithFloatParams * params_as_EltwiseOpWithFloatParams() const
Definition: program_generated.h:1844
ToDeviceOpBuilder Builder
Definition: program_generated.h:958
const ::flatbuffers::String * loc_info() const
Definition: program_generated.h:4149
uint32_t math_op() const
Definition: program_generated.h:3837
const tt::target::ttnn::TransposeOp * type_as_TransposeOp() const
Definition: program_generated.h:4098
AllGatherOpBuilder Builder
Definition: program_generated.h:3617
bool keep_dim() const
Definition: program_generated.h:1960
uint32_t input_height() const
Definition: program_generated.h:2924
float fill_value() const
Definition: program_generated.h:1524
const tt::target::ttnn::FullOp * type_as_FullOp() const
Definition: program_generated.h:4074
const ::flatbuffers::Vector< int32_t > * stride() const
Definition: program_generated.h:3162
::flatbuffers::Optional< tt::target::DataType > dtype() const
Definition: program_generated.h:804
ArangeOpBuilder Builder
Definition: program_generated.h:1602
ConcatOpBuilder Builder
Definition: program_generated.h:2366
uint32_t stride_height() const
Definition: program_generated.h:2936
FullOpBuilder Builder
Definition: program_generated.h:1512
tt::target::MeshShardDirection shard_direction() const
Definition: program_generated.h:3934
const ::flatbuffers::Vector< int32_t > * padding() const
Definition: program_generated.h:3165
const tt::target::ttnn::LinearOp * type_as_LinearOp() const
Definition: program_generated.h:4080
const tt::target::ttnn::TypecastOp * type_as_TypecastOp() const
Definition: program_generated.h:4059
int32_t dimension() const
Definition: program_generated.h:2231
const ::flatbuffers::Vector< int32_t > * output_padding() const
Definition: program_generated.h:3168
const tt::target::MemoryConfigDesc * memcfg() const
Definition: program_generated.h:728
ToMemoryConfigOpBuilder Builder
Definition: program_generated.h:718
const tt::target::TensorRef * input() const
Definition: program_generated.h:1050
tt::target::ttnn::OpType type_type() const
Definition: program_generated.h:4043
Conv2dOpBuilder Builder
Definition: program_generated.h:2877
float parameter() const
Definition: program_generated.h:1774
const tt::target::ttnn::MatmulOp * type_as_MatmulOp() const
Definition: program_generated.h:4083
const ::flatbuffers::Vector< int32_t > * kernel_size() const
Definition: program_generated.h:3159
const ::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::Operation > > * operations() const
Definition: program_generated.h:4374
bool Verify(::flatbuffers::Verifier &verifier) const
Definition: program_generated.h:652
const tt::target::ttnn::AllGatherOp * type_as_AllGatherOp() const
Definition: program_generated.h:4122
const tt::target::TensorRef * out() const
Definition: program_generated.h:731
const tt::target::ttnn::Conv2dOp * type_as_Conv2dOp() const
Definition: program_generated.h:4101
const ::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef > > * inputs() const
Definition: program_generated.h:2373
tt::target::ttnn::EltwiseOpType type() const
Definition: program_generated.h:1825
SoftmaxOpBuilder Builder
Definition: program_generated.h:2218
const tt::target::ttnn::PermuteOp * type_as_PermuteOp() const
Definition: program_generated.h:4140
float max() const
Definition: program_generated.h:1721
MaxPool2dOpBuilder Builder
Definition: program_generated.h:3359
const ::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef > > * ins() const
Definition: program_generated.h:1828
ReductionOpBuilder Builder
Definition: program_generated.h:1939
const ::flatbuffers::Vector< int64_t > * shard_shape() const
Definition: program_generated.h:3940
uint32_t groups() const
Definition: program_generated.h:2954
OperationBuilder Builder
Definition: program_generated.h:4035
float pad_value() const
Definition: program_generated.h:3725
const tt::target::ttnn::FillCacheOp * type_as_FillCacheOp() const
Definition: program_generated.h:4137
const tt::target::ttnn::OnesOp * type_as_OnesOp() const
Definition: program_generated.h:4071
FillCacheOpBuilder Builder
Definition: program_generated.h:1119
TransposeOpBuilder Builder
Definition: program_generated.h:2287
tt::target::ttnn::EltwiseOpParams params_type() const
Definition: program_generated.h:1834
const T * params_as() const
tt::target::ttnn::ReductionOpType type() const
Definition: program_generated.h:1948
uint32_t out_channels() const
Definition: program_generated.h:2918
ToLayoutOpBuilder Builder
Definition: program_generated.h:788
const ::flatbuffers::Vector< uint32_t > * shape() const
Definition: program_generated.h:2545
float end() const
Definition: program_generated.h:1616
const tt::target::DeviceRef * out() const
Definition: program_generated.h:649
bool force() const
Definition: program_generated.h:3568
EmbeddingBackwardOpBuilder Builder
Definition: program_generated.h:2116
const void * params() const
Definition: program_generated.h:1837
const tt::target::DebugInfo * debug_info() const
Definition: program_generated.h:4377
PermuteOpBuilder Builder
Definition: program_generated.h:3707
const ::flatbuffers::Vector< int32_t > * dim_arg() const
Definition: program_generated.h:1957
const tt::target::ttnn::ConvTranspose2dOp * type_as_ConvTranspose2dOp() const
Definition: program_generated.h:4104
EltwiseOpWithFloatParamsBuilder Builder
Definition: program_generated.h:1769
float step() const
Definition: program_generated.h:1619
const tt::target::DistributionStrategy * strategy() const
Definition: program_generated.h:1277
OnesOpBuilder Builder
Definition: program_generated.h:1392
const tt::target::DeviceRef * device() const
Definition: program_generated.h:810
const ::flatbuffers::Vector< int64_t > * permutation() const
Definition: program_generated.h:3719
LinearOpBuilder Builder
Definition: program_generated.h:2726
FromDeviceOpBuilder Builder
Definition: program_generated.h:1188
const tt::target::TensorRef * cache() const
Definition: program_generated.h:1047
float min() const
Definition: program_generated.h:1718
MatmulOpBuilder Builder
Definition: program_generated.h:2807
RepeatOpBuilder Builder
Definition: program_generated.h:2532
const tt::target::ttnn::ToDeviceOp * type_as_ToDeviceOp() const
Definition: program_generated.h:4062
uint32_t scatter_split_dim() const
Definition: program_generated.h:3834
const tt::target::ttnn::ReduceScatterOp * type_as_ReduceScatterOp() const
Definition: program_generated.h:4125
const tt::target::ttnn::UpdateCacheOp * type_as_UpdateCacheOp() const
Definition: program_generated.h:4134
const tt::target::ttnn::GetDeviceOp * type_as_GetDeviceOp() const
Definition: program_generated.h:4050
const tt::target::TensorRef * in_grad() const
Definition: program_generated.h:2132
MeshShardOpBuilder Builder
Definition: program_generated.h:3915
TypecastOpBuilder Builder
Definition: program_generated.h:889
Definition: program_generated.h:1146
::flatbuffers::Offset< FillCacheOp > Finish()
Definition: program_generated.h:1163
void add_cache(::flatbuffers::Offset< tt::target::TensorRef > cache)
Definition: program_generated.h:1150
::flatbuffers::uoffset_t start_
Definition: program_generated.h:1149
void add_batch_offset(uint32_t batch_offset)
Definition: program_generated.h:1156
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:1148
void add_input(::flatbuffers::Offset< tt::target::TensorRef > input)
Definition: program_generated.h:1153
FillCacheOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:1159
FillCacheOp Table
Definition: program_generated.h:1147
Definition: program_generated.h:1182
FillCacheOp type
Definition: program_generated.h:1183
static constexpr auto Create
Definition: program_generated.h:1184
Definition: program_generated.h:1210
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:1212
void add_in(::flatbuffers::Offset< tt::target::TensorRef > in)
Definition: program_generated.h:1214
::flatbuffers::uoffset_t start_
Definition: program_generated.h:1213
FromDeviceOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:1220
FromDeviceOp Table
Definition: program_generated.h:1211
::flatbuffers::Offset< FromDeviceOp > Finish()
Definition: program_generated.h:1224
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:1217
Definition: program_generated.h:1241
static constexpr auto Create
Definition: program_generated.h:1243
FromDeviceOp type
Definition: program_generated.h:1242
Definition: program_generated.h:1550
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: program_generated.h:1554
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:1552
void add_num_shards(uint32_t num_shards)
Definition: program_generated.h:1560
::flatbuffers::Offset< FullOp > Finish()
Definition: program_generated.h:1573
void add_strategy(::flatbuffers::Offset< tt::target::DistributionStrategy > strategy)
Definition: program_generated.h:1563
void add_fill_value(float fill_value)
Definition: program_generated.h:1557
FullOp Table
Definition: program_generated.h:1551
FullOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:1569
::flatbuffers::uoffset_t start_
Definition: program_generated.h:1553
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:1566
Definition: program_generated.h:1596
FullOp type
Definition: program_generated.h:1597
static constexpr auto Create
Definition: program_generated.h:1598
Definition: program_generated.h:663
GetDeviceOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:676
::flatbuffers::uoffset_t start_
Definition: program_generated.h:666
::flatbuffers::Offset< GetDeviceOp > Finish()
Definition: program_generated.h:680
void add_chip_ids(::flatbuffers::Offset<::flatbuffers::Vector< uint32_t >> chip_ids)
Definition: program_generated.h:670
void add_mesh(const tt::target::Dim2d *mesh)
Definition: program_generated.h:667
GetDeviceOp Table
Definition: program_generated.h:664
void add_out(::flatbuffers::Offset< tt::target::DeviceRef > out)
Definition: program_generated.h:673
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:665
Definition: program_generated.h:699
GetDeviceOp type
Definition: program_generated.h:700
static constexpr auto Create
Definition: program_generated.h:701
Definition: program_generated.h:2760
LinearOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:2776
void add_in0(::flatbuffers::Offset< tt::target::TensorRef > in0)
Definition: program_generated.h:2764
::flatbuffers::Offset< LinearOp > Finish()
Definition: program_generated.h:2780
::flatbuffers::uoffset_t start_
Definition: program_generated.h:2763
void add_in1(::flatbuffers::Offset< tt::target::TensorRef > in1)
Definition: program_generated.h:2767
void add_bias(::flatbuffers::Offset< tt::target::TensorRef > bias)
Definition: program_generated.h:2770
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:2773
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:2762
LinearOp Table
Definition: program_generated.h:2761
Definition: program_generated.h:2801
static constexpr auto Create
Definition: program_generated.h:2803
LinearOp type
Definition: program_generated.h:2802
Definition: program_generated.h:2835
::flatbuffers::uoffset_t start_
Definition: program_generated.h:2838
::flatbuffers::Offset< MatmulOp > Finish()
Definition: program_generated.h:2852
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:2845
MatmulOp Table
Definition: program_generated.h:2836
void add_in1(::flatbuffers::Offset< tt::target::TensorRef > in1)
Definition: program_generated.h:2842
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:2837
void add_in0(::flatbuffers::Offset< tt::target::TensorRef > in0)
Definition: program_generated.h:2839
MatmulOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:2848
Definition: program_generated.h:2871
static constexpr auto Create
Definition: program_generated.h:2873
MatmulOp type
Definition: program_generated.h:2872
Definition: program_generated.h:3452
void add_input_width(uint32_t input_width)
Definition: program_generated.h:3471
::flatbuffers::uoffset_t start_
Definition: program_generated.h:3455
void add_input_height(uint32_t input_height)
Definition: program_generated.h:3468
MaxPool2dOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:3504
void add_stride_height(uint32_t stride_height)
Definition: program_generated.h:3483
void add_ceil_mode(bool ceil_mode)
Definition: program_generated.h:3495
void add_dilation_width(uint32_t dilation_width)
Definition: program_generated.h:3492
void add_padding_height(uint32_t padding_height)
Definition: program_generated.h:3498
void add_kernel_height(uint32_t kernel_height)
Definition: program_generated.h:3477
MaxPool2dOp Table
Definition: program_generated.h:3453
void add_dilation_height(uint32_t dilation_height)
Definition: program_generated.h:3489
void add_in(::flatbuffers::Offset< tt::target::TensorRef > in)
Definition: program_generated.h:3456
void add_channels(uint32_t channels)
Definition: program_generated.h:3474
void add_padding_width(uint32_t padding_width)
Definition: program_generated.h:3501
::flatbuffers::Offset< MaxPool2dOp > Finish()
Definition: program_generated.h:3508
void add_kernel_width(uint32_t kernel_width)
Definition: program_generated.h:3480
void add_stride_width(uint32_t stride_width)
Definition: program_generated.h:3486
void add_batch_size(uint32_t batch_size)
Definition: program_generated.h:3465
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:3459
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:3454
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: program_generated.h:3462
Definition: program_generated.h:3553
MaxPool2dOp type
Definition: program_generated.h:3554
static constexpr auto Create
Definition: program_generated.h:3555
Definition: program_generated.h:3959
MeshShardOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:3981
void add_in(::flatbuffers::Offset< tt::target::TensorRef > in)
Definition: program_generated.h:3963
MeshShardOp Table
Definition: program_generated.h:3960
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: program_generated.h:3969
::flatbuffers::uoffset_t start_
Definition: program_generated.h:3962
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:3966
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:3961
void add_shard_type(tt::target::MeshShardType shard_type)
Definition: program_generated.h:3975
void add_shard_direction(tt::target::MeshShardDirection shard_direction)
Definition: program_generated.h:3972
void add_shard_shape(::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> shard_shape)
Definition: program_generated.h:3978
::flatbuffers::Offset< MeshShardOp > Finish()
Definition: program_generated.h:3985
Definition: program_generated.h:4010
MeshShardOp type
Definition: program_generated.h:4011
static constexpr auto Create
Definition: program_generated.h:4012
Definition: program_generated.h:1436
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:1455
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: program_generated.h:1449
void add_layout(tt::target::TensorLayout layout)
Definition: program_generated.h:1446
::flatbuffers::uoffset_t start_
Definition: program_generated.h:1439
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:1438
void add_shape(::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> shape)
Definition: program_generated.h:1440
void add_dtype(tt::target::DataType dtype)
Definition: program_generated.h:1443
::flatbuffers::Offset< OnesOp > Finish()
Definition: program_generated.h:1462
OnesOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:1458
void add_memcfg(::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg)
Definition: program_generated.h:1452
OnesOp Table
Definition: program_generated.h:1437
Definition: program_generated.h:1487
OnesOp type
Definition: program_generated.h:1488
static constexpr auto Create
Definition: program_generated.h:1489
Definition: program_generated.h:500
static const OpType enum_value
Definition: program_generated.h:501
Definition: program_generated.h:4293
void add_type(::flatbuffers::Offset< void > type)
Definition: program_generated.h:4300
OperationBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:4309
::flatbuffers::Offset< Operation > Finish()
Definition: program_generated.h:4313
Operation Table
Definition: program_generated.h:4294
void add_debug_info(::flatbuffers::Offset<::flatbuffers::String > debug_info)
Definition: program_generated.h:4303
void add_loc_info(::flatbuffers::Offset<::flatbuffers::String > loc_info)
Definition: program_generated.h:4306
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:4295
::flatbuffers::uoffset_t start_
Definition: program_generated.h:4296
void add_type_type(tt::target::ttnn::OpType type_type)
Definition: program_generated.h:4297
Definition: program_generated.h:4334
static constexpr auto Create
Definition: program_generated.h:4336
Operation type
Definition: program_generated.h:4335
Definition: program_generated.h:3746
::flatbuffers::Offset< PermuteOp > Finish()
Definition: program_generated.h:3769
void add_in(::flatbuffers::Offset< tt::target::TensorRef > in)
Definition: program_generated.h:3750
::flatbuffers::uoffset_t start_
Definition: program_generated.h:3749
void add_permutation(::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> permutation)
Definition: program_generated.h:3753
PermuteOp Table
Definition: program_generated.h:3747
void add_pad_value(float pad_value)
Definition: program_generated.h:3759
PermuteOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:3765
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:3748
void add_memory_config(::flatbuffers::Offset< tt::target::MemoryConfigDesc > memory_config)
Definition: program_generated.h:3756
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:3762
Definition: program_generated.h:3792
static constexpr auto Create
Definition: program_generated.h:3794
PermuteOp type
Definition: program_generated.h:3793
Definition: program_generated.h:4399
Program Table
Definition: program_generated.h:4400
::flatbuffers::uoffset_t start_
Definition: program_generated.h:4402
void add_name(::flatbuffers::Offset<::flatbuffers::String > name)
Definition: program_generated.h:4403
void add_debug_info(::flatbuffers::Offset< tt::target::DebugInfo > debug_info)
Definition: program_generated.h:4415
void add_inputs(::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> inputs)
Definition: program_generated.h:4406
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:4401
void add_outputs(::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> outputs)
Definition: program_generated.h:4409
::flatbuffers::Offset< Program > Finish()
Definition: program_generated.h:4422
ProgramBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:4418
void add_operations(::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::Operation >>> operations)
Definition: program_generated.h:4412
Definition: program_generated.h:4445
static constexpr auto Create
Definition: program_generated.h:4447
Program type
Definition: program_generated.h:4446
Definition: program_generated.h:3858
void add_num_links(uint32_t num_links)
Definition: program_generated.h:3877
ReduceScatterOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:3880
void add_scatter_split_dim(uint32_t scatter_split_dim)
Definition: program_generated.h:3871
ReduceScatterOp Table
Definition: program_generated.h:3859
void add_math_op(uint32_t math_op)
Definition: program_generated.h:3874
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:3860
void add_in(::flatbuffers::Offset< tt::target::TensorRef > in)
Definition: program_generated.h:3862
::flatbuffers::uoffset_t start_
Definition: program_generated.h:3861
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:3865
::flatbuffers::Offset< ReduceScatterOp > Finish()
Definition: program_generated.h:3884
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: program_generated.h:3868
Definition: program_generated.h:3909
static constexpr auto Create
Definition: program_generated.h:3911
ReduceScatterOp type
Definition: program_generated.h:3910
Definition: program_generated.h:1977
ReductionOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:1996
void add_in(::flatbuffers::Offset< tt::target::TensorRef > in)
Definition: program_generated.h:1984
void add_dim_arg(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> dim_arg)
Definition: program_generated.h:1990
::flatbuffers::Offset< ReductionOp > Finish()
Definition: program_generated.h:2000
void add_type(tt::target::ttnn::ReductionOpType type)
Definition: program_generated.h:1981
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:1979
::flatbuffers::uoffset_t start_
Definition: program_generated.h:1980
ReductionOp Table
Definition: program_generated.h:1978
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:1987
void add_keep_dim(bool keep_dim)
Definition: program_generated.h:1993
Definition: program_generated.h:2023
ReductionOp type
Definition: program_generated.h:2024
static constexpr auto Create
Definition: program_generated.h:2025
Definition: program_generated.h:2560
void add_shape(::flatbuffers::Offset<::flatbuffers::Vector< uint32_t >> shape)
Definition: program_generated.h:2570
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:2567
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:2562
void add_in(::flatbuffers::Offset< tt::target::TensorRef > in)
Definition: program_generated.h:2564
::flatbuffers::uoffset_t start_
Definition: program_generated.h:2563
RepeatOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:2573
::flatbuffers::Offset< RepeatOp > Finish()
Definition: program_generated.h:2577
RepeatOp Table
Definition: program_generated.h:2561
Definition: program_generated.h:2596
static constexpr auto Create
Definition: program_generated.h:2598
RepeatOp type
Definition: program_generated.h:2597
Definition: program_generated.h:2477
void add_shape(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> shape)
Definition: program_generated.h:2487
::flatbuffers::uoffset_t start_
Definition: program_generated.h:2480
ReshapeOp Table
Definition: program_generated.h:2478
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:2479
void add_in(::flatbuffers::Offset< tt::target::TensorRef > in)
Definition: program_generated.h:2481
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:2484
::flatbuffers::Offset< ReshapeOp > Finish()
Definition: program_generated.h:2494
ReshapeOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:2490
Definition: program_generated.h:2513
static constexpr auto Create
Definition: program_generated.h:2515
ReshapeOp type
Definition: program_generated.h:2514
Definition: program_generated.h:2655
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:2657
SliceOp Table
Definition: program_generated.h:2656
void add_step(::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> step)
Definition: program_generated.h:2671
void add_begins(::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> begins)
Definition: program_generated.h:2665
::flatbuffers::Offset< SliceOp > Finish()
Definition: program_generated.h:2678
void add_in(::flatbuffers::Offset< tt::target::TensorRef > in)
Definition: program_generated.h:2659
::flatbuffers::uoffset_t start_
Definition: program_generated.h:2658
void add_ends(::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> ends)
Definition: program_generated.h:2668
SliceOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:2674
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:2662
Definition: program_generated.h:2701
SliceOp type
Definition: program_generated.h:2702
static constexpr auto Create
Definition: program_generated.h:2703
Definition: program_generated.h:2245
void add_dimension(int32_t dimension)
Definition: program_generated.h:2255
::flatbuffers::uoffset_t start_
Definition: program_generated.h:2248
void add_in(::flatbuffers::Offset< tt::target::TensorRef > in)
Definition: program_generated.h:2249
SoftmaxOp Table
Definition: program_generated.h:2246
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:2247
::flatbuffers::Offset< SoftmaxOp > Finish()
Definition: program_generated.h:2262
SoftmaxOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:2258
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:2252
Definition: program_generated.h:2281
static constexpr auto Create
Definition: program_generated.h:2283
SoftmaxOp type
Definition: program_generated.h:2282
Definition: program_generated.h:992
ToDeviceOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:1008
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:1005
void add_in(::flatbuffers::Offset< tt::target::TensorRef > in)
Definition: program_generated.h:996
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: program_generated.h:999
::flatbuffers::Offset< ToDeviceOp > Finish()
Definition: program_generated.h:1012
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:994
ToDeviceOp Table
Definition: program_generated.h:993
::flatbuffers::uoffset_t start_
Definition: program_generated.h:995
void add_memcfg(::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg)
Definition: program_generated.h:1002
Definition: program_generated.h:1033
ToDeviceOp type
Definition: program_generated.h:1034
static constexpr auto Create
Definition: program_generated.h:1035
Definition: program_generated.h:832
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:834
ToLayoutOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:854
ToLayoutOp Table
Definition: program_generated.h:833
::flatbuffers::Offset< ToLayoutOp > Finish()
Definition: program_generated.h:858
void add_layout(tt::target::TensorLayout layout)
Definition: program_generated.h:839
void add_dtype(tt::target::DataType dtype)
Definition: program_generated.h:842
void add_in(::flatbuffers::Offset< tt::target::TensorRef > in)
Definition: program_generated.h:836
void add_memcfg(::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg)
Definition: program_generated.h:845
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: program_generated.h:848
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:851
::flatbuffers::uoffset_t start_
Definition: program_generated.h:835
Definition: program_generated.h:883
static constexpr auto Create
Definition: program_generated.h:885
ToLayoutOp type
Definition: program_generated.h:884
Definition: program_generated.h:746
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:756
void add_memcfg(::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg)
Definition: program_generated.h:753
::flatbuffers::uoffset_t start_
Definition: program_generated.h:749
ToMemoryConfigOp Table
Definition: program_generated.h:747
void add_in0(::flatbuffers::Offset< tt::target::TensorRef > in0)
Definition: program_generated.h:750
ToMemoryConfigOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:759
::flatbuffers::Offset< ToMemoryConfigOp > Finish()
Definition: program_generated.h:763
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:748
Definition: program_generated.h:782
ToMemoryConfigOp type
Definition: program_generated.h:783
static constexpr auto Create
Definition: program_generated.h:784
Definition: program_generated.h:2319
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:2321
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:2326
void add_dim0(int32_t dim0)
Definition: program_generated.h:2329
::flatbuffers::Offset< TransposeOp > Finish()
Definition: program_generated.h:2339
TransposeOp Table
Definition: program_generated.h:2320
TransposeOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:2335
::flatbuffers::uoffset_t start_
Definition: program_generated.h:2322
void add_dim1(int32_t dim1)
Definition: program_generated.h:2332
void add_in(::flatbuffers::Offset< tt::target::TensorRef > in)
Definition: program_generated.h:2323
Definition: program_generated.h:2360
TransposeOp type
Definition: program_generated.h:2361
static constexpr auto Create
Definition: program_generated.h:2362
Definition: program_generated.h:916
::flatbuffers::Offset< TypecastOp > Finish()
Definition: program_generated.h:933
::flatbuffers::uoffset_t start_
Definition: program_generated.h:919
void add_dtype(tt::target::DataType dtype)
Definition: program_generated.h:923
TypecastOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:929
TypecastOp Table
Definition: program_generated.h:917
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:918
void add_out(::flatbuffers::Offset< tt::target::TensorRef > out)
Definition: program_generated.h:926
void add_in(::flatbuffers::Offset< tt::target::TensorRef > in)
Definition: program_generated.h:920
Definition: program_generated.h:952
TypecastOp type
Definition: program_generated.h:953
static constexpr auto Create
Definition: program_generated.h:954
Definition: program_generated.h:1072
void add_input(::flatbuffers::Offset< tt::target::TensorRef > input)
Definition: program_generated.h:1079
void add_batch_offset(uint32_t batch_offset)
Definition: program_generated.h:1085
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:1074
::flatbuffers::uoffset_t start_
Definition: program_generated.h:1075
::flatbuffers::Offset< UpdateCacheOp > Finish()
Definition: program_generated.h:1092
void add_cache(::flatbuffers::Offset< tt::target::TensorRef > cache)
Definition: program_generated.h:1076
UpdateCacheOp Table
Definition: program_generated.h:1073
void add_update_index(::flatbuffers::Offset< tt::target::TensorRef > update_index)
Definition: program_generated.h:1082
UpdateCacheOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:1088
Definition: program_generated.h:1113
static constexpr auto Create
Definition: program_generated.h:1115
UpdateCacheOp type
Definition: program_generated.h:1114