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");
43 struct OperationBuilder;
46 struct ProgramBuilder;
106 static const OpType values[] = {
164 static const char *
const names[53] = {
167 "CollectivePermuteOp",
171 "PrepareConv2dWeightsOp",
183 "RepeatInterleaveOp",
190 "EltwiseBinaryCompositeOp",
191 "EltwiseTernaryWhereOp",
192 "EltwiseQuantizationOp",
194 "EltwiseUnaryCompositeOp",
195 "EmbeddingBackwardOp",
224 const size_t index =
static_cast<size_t>(e);
437 bool VerifyOpTypeVector(::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset<void>> *values, const ::flatbuffers::Vector<OpType> *types);
443 return "tt.target.ttnn.Operation";
445 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
455 return GetPointer<const void *>(
VT_TYPE);
457 template<
typename T>
const T *
type_as()
const;
612 return GetPointer<const ::flatbuffers::String *>(
VT_DEBUG_INFO);
615 return GetPointer<const ::flatbuffers::String *>(VT_LOC_INFO);
617 bool Verify(::flatbuffers::Verifier &verifier)
const {
618 return VerifyTableStart(verifier) &&
620 VerifyOffset(verifier,
VT_TYPE) &&
623 verifier.VerifyString(debug_info()) &&
624 VerifyOffset(verifier, VT_LOC_INFO) &&
625 verifier.VerifyString(loc_info()) &&
630 template<>
inline const tt::target::ttnn::AllGatherOp *Operation::type_as<tt::target::ttnn::AllGatherOp>()
const {
631 return type_as_AllGatherOp();
634 template<>
inline const tt::target::ttnn::CollectivePermuteOp *Operation::type_as<tt::target::ttnn::CollectivePermuteOp>()
const {
635 return type_as_CollectivePermuteOp();
638 template<>
inline const tt::target::ttnn::MeshShardOp *Operation::type_as<tt::target::ttnn::MeshShardOp>()
const {
639 return type_as_MeshShardOp();
642 template<>
inline const tt::target::ttnn::ReduceScatterOp *Operation::type_as<tt::target::ttnn::ReduceScatterOp>()
const {
643 return type_as_ReduceScatterOp();
646 template<>
inline const tt::target::ttnn::GetDeviceOp *Operation::type_as<tt::target::ttnn::GetDeviceOp>()
const {
647 return type_as_GetDeviceOp();
650 template<>
inline const tt::target::ttnn::PrepareConv2dWeightsOp *Operation::type_as<tt::target::ttnn::PrepareConv2dWeightsOp>()
const {
651 return type_as_PrepareConv2dWeightsOp();
654 template<>
inline const tt::target::ttnn::Conv2dOp *Operation::type_as<tt::target::ttnn::Conv2dOp>()
const {
655 return type_as_Conv2dOp();
658 template<>
inline const tt::target::ttnn::ConvTranspose2dOp *Operation::type_as<tt::target::ttnn::ConvTranspose2dOp>()
const {
659 return type_as_ConvTranspose2dOp();
662 template<>
inline const tt::target::ttnn::CpuOp *Operation::type_as<tt::target::ttnn::CpuOp>()
const {
663 return type_as_CpuOp();
666 template<>
inline const tt::target::ttnn::ArangeOp *Operation::type_as<tt::target::ttnn::ArangeOp>()
const {
667 return type_as_ArangeOp();
670 template<>
inline const tt::target::ttnn::ConstantOp *Operation::type_as<tt::target::ttnn::ConstantOp>()
const {
671 return type_as_ConstantOp();
674 template<>
inline const tt::target::ttnn::EmptyOp *Operation::type_as<tt::target::ttnn::EmptyOp>()
const {
675 return type_as_EmptyOp();
678 template<>
inline const tt::target::ttnn::FullOp *Operation::type_as<tt::target::ttnn::FullOp>()
const {
679 return type_as_FullOp();
682 template<>
inline const tt::target::ttnn::NamedFullOp *Operation::type_as<tt::target::ttnn::NamedFullOp>()
const {
683 return type_as_NamedFullOp();
686 template<>
inline const tt::target::ttnn::ConcatOp *Operation::type_as<tt::target::ttnn::ConcatOp>()
const {
687 return type_as_ConcatOp();
690 template<>
inline const tt::target::ttnn::PadOp *Operation::type_as<tt::target::ttnn::PadOp>()
const {
691 return type_as_PadOp();
694 template<>
inline const tt::target::ttnn::PermuteOp *Operation::type_as<tt::target::ttnn::PermuteOp>()
const {
695 return type_as_PermuteOp();
698 template<>
inline const tt::target::ttnn::RepeatInterleaveOp *Operation::type_as<tt::target::ttnn::RepeatInterleaveOp>()
const {
699 return type_as_RepeatInterleaveOp();
702 template<>
inline const tt::target::ttnn::RepeatOp *Operation::type_as<tt::target::ttnn::RepeatOp>()
const {
703 return type_as_RepeatOp();
706 template<>
inline const tt::target::ttnn::ReshapeOp *Operation::type_as<tt::target::ttnn::ReshapeOp>()
const {
707 return type_as_ReshapeOp();
710 template<>
inline const tt::target::ttnn::SliceOp *Operation::type_as<tt::target::ttnn::SliceOp>()
const {
711 return type_as_SliceOp();
714 template<>
inline const tt::target::ttnn::TransposeOp *Operation::type_as<tt::target::ttnn::TransposeOp>()
const {
715 return type_as_TransposeOp();
718 template<>
inline const tt::target::ttnn::DeallocateOp *Operation::type_as<tt::target::ttnn::DeallocateOp>()
const {
719 return type_as_DeallocateOp();
722 template<>
inline const tt::target::ttnn::EltwiseBinaryOp *Operation::type_as<tt::target::ttnn::EltwiseBinaryOp>()
const {
723 return type_as_EltwiseBinaryOp();
726 template<>
inline const tt::target::ttnn::EltwiseBinaryCompositeOp *Operation::type_as<tt::target::ttnn::EltwiseBinaryCompositeOp>()
const {
727 return type_as_EltwiseBinaryCompositeOp();
730 template<>
inline const tt::target::ttnn::EltwiseTernaryWhereOp *Operation::type_as<tt::target::ttnn::EltwiseTernaryWhereOp>()
const {
731 return type_as_EltwiseTernaryWhereOp();
734 template<>
inline const tt::target::ttnn::EltwiseQuantizationOp *Operation::type_as<tt::target::ttnn::EltwiseQuantizationOp>()
const {
735 return type_as_EltwiseQuantizationOp();
738 template<>
inline const tt::target::ttnn::EltwiseUnaryOp *Operation::type_as<tt::target::ttnn::EltwiseUnaryOp>()
const {
739 return type_as_EltwiseUnaryOp();
742 template<>
inline const tt::target::ttnn::EltwiseUnaryCompositeOp *Operation::type_as<tt::target::ttnn::EltwiseUnaryCompositeOp>()
const {
743 return type_as_EltwiseUnaryCompositeOp();
746 template<>
inline const tt::target::ttnn::EmbeddingBackwardOp *Operation::type_as<tt::target::ttnn::EmbeddingBackwardOp>()
const {
747 return type_as_EmbeddingBackwardOp();
750 template<>
inline const tt::target::ttnn::EmbeddingOp *Operation::type_as<tt::target::ttnn::EmbeddingOp>()
const {
751 return type_as_EmbeddingOp();
754 template<>
inline const tt::target::ttnn::FillCacheOp *Operation::type_as<tt::target::ttnn::FillCacheOp>()
const {
755 return type_as_FillCacheOp();
758 template<>
inline const tt::target::ttnn::UpdateCacheOp *Operation::type_as<tt::target::ttnn::UpdateCacheOp>()
const {
759 return type_as_UpdateCacheOp();
762 template<>
inline const tt::target::ttnn::FromDeviceOp *Operation::type_as<tt::target::ttnn::FromDeviceOp>()
const {
763 return type_as_FromDeviceOp();
766 template<>
inline const tt::target::ttnn::ToDeviceOp *Operation::type_as<tt::target::ttnn::ToDeviceOp>()
const {
767 return type_as_ToDeviceOp();
770 template<>
inline const tt::target::ttnn::ToDTypeOp *Operation::type_as<tt::target::ttnn::ToDTypeOp>()
const {
771 return type_as_ToDTypeOp();
774 template<>
inline const tt::target::ttnn::ToLayoutOp *Operation::type_as<tt::target::ttnn::ToLayoutOp>()
const {
775 return type_as_ToLayoutOp();
778 template<>
inline const tt::target::ttnn::ToMemoryConfigOp *Operation::type_as<tt::target::ttnn::ToMemoryConfigOp>()
const {
779 return type_as_ToMemoryConfigOp();
782 template<>
inline const tt::target::ttnn::TypecastOp *Operation::type_as<tt::target::ttnn::TypecastOp>()
const {
783 return type_as_TypecastOp();
786 template<>
inline const tt::target::ttnn::LinearOp *Operation::type_as<tt::target::ttnn::LinearOp>()
const {
787 return type_as_LinearOp();
790 template<>
inline const tt::target::ttnn::MatmulOp *Operation::type_as<tt::target::ttnn::MatmulOp>()
const {
791 return type_as_MatmulOp();
794 template<>
inline const tt::target::ttnn::MorehCumSumOp *Operation::type_as<tt::target::ttnn::MorehCumSumOp>()
const {
795 return type_as_MorehCumSumOp();
798 template<>
inline const tt::target::ttnn::SoftmaxOp *Operation::type_as<tt::target::ttnn::SoftmaxOp>()
const {
799 return type_as_SoftmaxOp();
802 template<>
inline const tt::target::ttnn::Pool2dOp *Operation::type_as<tt::target::ttnn::Pool2dOp>()
const {
803 return type_as_Pool2dOp();
806 template<>
inline const tt::target::ttnn::UpsampleOp *Operation::type_as<tt::target::ttnn::UpsampleOp>()
const {
807 return type_as_UpsampleOp();
810 template<>
inline const tt::target::ttnn::ReductionArgMaxOp *Operation::type_as<tt::target::ttnn::ReductionArgMaxOp>()
const {
811 return type_as_ReductionArgMaxOp();
814 template<>
inline const tt::target::ttnn::ReductionOp *Operation::type_as<tt::target::ttnn::ReductionOp>()
const {
815 return type_as_ReductionOp();
818 template<>
inline const tt::target::ttnn::ReductionProdOp *Operation::type_as<tt::target::ttnn::ReductionProdOp>()
const {
819 return type_as_ReductionProdOp();
822 template<>
inline const tt::target::ttnn::LoadCachedOp *Operation::type_as<tt::target::ttnn::LoadCachedOp>()
const {
823 return type_as_LoadCachedOp();
826 template<>
inline const tt::target::ttnn::BatchNormOp *Operation::type_as<tt::target::ttnn::BatchNormOp>()
const {
827 return type_as_BatchNormOp();
830 template<>
inline const tt::target::ttnn::TraceOp *Operation::type_as<tt::target::ttnn::TraceOp>()
const {
831 return type_as_TraceOp();
836 ::flatbuffers::FlatBufferBuilder &
fbb_;
847 void add_loc_info(::flatbuffers::Offset<::flatbuffers::String> loc_info) {
848 fbb_.AddOffset(Operation::VT_LOC_INFO, loc_info);
854 ::flatbuffers::Offset<Operation>
Finish() {
856 auto o = ::flatbuffers::Offset<Operation>(end);
862 ::flatbuffers::FlatBufferBuilder &_fbb,
864 ::flatbuffers::Offset<void> type = 0,
865 ::flatbuffers::Offset<::flatbuffers::String> debug_info = 0,
866 ::flatbuffers::Offset<::flatbuffers::String> loc_info = 0) {
881 ::flatbuffers::FlatBufferBuilder &_fbb,
883 ::flatbuffers::Offset<void> type = 0,
884 const char *debug_info =
nullptr,
885 const char *loc_info =
nullptr) {
886 auto debug_info__ = debug_info ? _fbb.CreateString(debug_info) : 0;
887 auto loc_info__ = loc_info ? _fbb.CreateString(loc_info) : 0;
900 return "tt.target.ttnn.Program";
902 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
912 const ::flatbuffers::String *
name()
const {
913 return GetPointer<const ::flatbuffers::String *>(
VT_NAME);
915 const ::flatbuffers::Vector<::flatbuffers::Offset<tt::target::ttnn::TensorRef>> *
inputs()
const {
916 return GetPointer<const ::flatbuffers::Vector<::flatbuffers::Offset<tt::target::ttnn::TensorRef>> *>(
VT_INPUTS);
918 const ::flatbuffers::Vector<::flatbuffers::Offset<tt::target::ttnn::TensorRef>> *
outputs()
const {
919 return GetPointer<const ::flatbuffers::Vector<::flatbuffers::Offset<tt::target::ttnn::TensorRef>> *>(
VT_OUTPUTS);
921 const ::flatbuffers::Vector<::flatbuffers::Offset<tt::target::ttnn::Operation>> *
operations()
const {
922 return GetPointer<const ::flatbuffers::Vector<::flatbuffers::Offset<tt::target::ttnn::Operation>> *>(
VT_OPERATIONS);
924 const ::flatbuffers::Vector<::flatbuffers::Offset<tt::target::DynamicLib>> *
dylibs()
const {
925 return GetPointer<const ::flatbuffers::Vector<::flatbuffers::Offset<tt::target::DynamicLib>> *>(
VT_DYLIBS);
928 return GetPointer<const tt::target::DebugInfo *>(
VT_DEBUG_INFO);
936 bool Verify(::flatbuffers::Verifier &verifier)
const {
937 return VerifyTableStart(verifier) &&
938 VerifyOffset(verifier,
VT_NAME) &&
939 verifier.VerifyString(name()) &&
941 verifier.VerifyVector(inputs()) &&
942 verifier.VerifyVectorOfTables(inputs()) &&
944 verifier.VerifyVector(outputs()) &&
945 verifier.VerifyVectorOfTables(outputs()) &&
947 verifier.VerifyVector(operations()) &&
948 verifier.VerifyVectorOfTables(operations()) &&
950 verifier.VerifyVector(dylibs()) &&
951 verifier.VerifyVectorOfTables(dylibs()) &&
953 verifier.VerifyTable(debug_info()) &&
955 VerifyField<tt::target::Dim2d>(verifier,
VT_MESH_SHAPE, 4) &&
962 ::flatbuffers::FlatBufferBuilder &
fbb_;
964 void add_name(::flatbuffers::Offset<::flatbuffers::String> name) {
967 void add_inputs(::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<tt::target::ttnn::TensorRef>>> inputs) {
970 void add_outputs(::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<tt::target::ttnn::TensorRef>>> outputs) {
973 void add_operations(::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<tt::target::ttnn::Operation>>> operations) {
976 void add_dylibs(::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<tt::target::DynamicLib>>> dylibs) {
992 ::flatbuffers::Offset<Program>
Finish() {
994 auto o = ::flatbuffers::Offset<Program>(end);
1000 ::flatbuffers::FlatBufferBuilder &_fbb,
1001 ::flatbuffers::Offset<::flatbuffers::String> name = 0,
1002 ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<tt::target::ttnn::TensorRef>>> inputs = 0,
1003 ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<tt::target::ttnn::TensorRef>>> outputs = 0,
1004 ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<tt::target::ttnn::Operation>>> operations = 0,
1005 ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset<tt::target::DynamicLib>>> dylibs = 0,
1006 ::flatbuffers::Offset<tt::target::DebugInfo> debug_info = 0,
1007 bool private_ =
false,
1008 const tt::target::Dim2d *mesh_shape =
nullptr) {
1018 return builder_.
Finish();
1027 ::flatbuffers::FlatBufferBuilder &_fbb,
1028 const char *name =
nullptr,
1029 const std::vector<::flatbuffers::Offset<tt::target::ttnn::TensorRef>> *inputs =
nullptr,
1030 const std::vector<::flatbuffers::Offset<tt::target::ttnn::TensorRef>> *outputs =
nullptr,
1031 const std::vector<::flatbuffers::Offset<tt::target::ttnn::Operation>> *operations =
nullptr,
1032 const std::vector<::flatbuffers::Offset<tt::target::DynamicLib>> *dylibs =
nullptr,
1033 ::flatbuffers::Offset<tt::target::DebugInfo> debug_info = 0,
1034 bool private_ =
false,
1035 const tt::target::Dim2d *mesh_shape =
nullptr) {
1036 auto name__ = name ? _fbb.CreateString(name) : 0;
1037 auto inputs__ = inputs ? _fbb.CreateVector<::flatbuffers::Offset<tt::target::ttnn::TensorRef>>(*inputs) : 0;
1038 auto outputs__ = outputs ? _fbb.CreateVector<::flatbuffers::Offset<tt::target::ttnn::TensorRef>>(*outputs) : 0;
1039 auto operations__ = operations ? _fbb.CreateVector<::flatbuffers::Offset<tt::target::ttnn::Operation>>(*operations) : 0;
1040 auto dylibs__ = dylibs ? _fbb.CreateVector<::flatbuffers::Offset<tt::target::DynamicLib>>(*dylibs) : 0;
1059 auto ptr =
reinterpret_cast<const tt::target::ttnn::AllGatherOp *
>(obj);
1060 return verifier.VerifyTable(ptr);
1063 auto ptr =
reinterpret_cast<const tt::target::ttnn::CollectivePermuteOp *
>(obj);
1064 return verifier.VerifyTable(ptr);
1067 auto ptr =
reinterpret_cast<const tt::target::ttnn::MeshShardOp *
>(obj);
1068 return verifier.VerifyTable(ptr);
1071 auto ptr =
reinterpret_cast<const tt::target::ttnn::ReduceScatterOp *
>(obj);
1072 return verifier.VerifyTable(ptr);
1075 auto ptr =
reinterpret_cast<const tt::target::ttnn::GetDeviceOp *
>(obj);
1076 return verifier.VerifyTable(ptr);
1079 auto ptr =
reinterpret_cast<const tt::target::ttnn::PrepareConv2dWeightsOp *
>(obj);
1080 return verifier.VerifyTable(ptr);
1083 auto ptr =
reinterpret_cast<const tt::target::ttnn::Conv2dOp *
>(obj);
1084 return verifier.VerifyTable(ptr);
1087 auto ptr =
reinterpret_cast<const tt::target::ttnn::ConvTranspose2dOp *
>(obj);
1088 return verifier.VerifyTable(ptr);
1091 auto ptr =
reinterpret_cast<const tt::target::ttnn::CpuOp *
>(obj);
1092 return verifier.VerifyTable(ptr);
1095 auto ptr =
reinterpret_cast<const tt::target::ttnn::ArangeOp *
>(obj);
1096 return verifier.VerifyTable(ptr);
1099 auto ptr =
reinterpret_cast<const tt::target::ttnn::ConstantOp *
>(obj);
1100 return verifier.VerifyTable(ptr);
1103 auto ptr =
reinterpret_cast<const tt::target::ttnn::EmptyOp *
>(obj);
1104 return verifier.VerifyTable(ptr);
1107 auto ptr =
reinterpret_cast<const tt::target::ttnn::FullOp *
>(obj);
1108 return verifier.VerifyTable(ptr);
1111 auto ptr =
reinterpret_cast<const tt::target::ttnn::NamedFullOp *
>(obj);
1112 return verifier.VerifyTable(ptr);
1115 auto ptr =
reinterpret_cast<const tt::target::ttnn::ConcatOp *
>(obj);
1116 return verifier.VerifyTable(ptr);
1119 auto ptr =
reinterpret_cast<const tt::target::ttnn::PadOp *
>(obj);
1120 return verifier.VerifyTable(ptr);
1123 auto ptr =
reinterpret_cast<const tt::target::ttnn::PermuteOp *
>(obj);
1124 return verifier.VerifyTable(ptr);
1127 auto ptr =
reinterpret_cast<const tt::target::ttnn::RepeatInterleaveOp *
>(obj);
1128 return verifier.VerifyTable(ptr);
1131 auto ptr =
reinterpret_cast<const tt::target::ttnn::RepeatOp *
>(obj);
1132 return verifier.VerifyTable(ptr);
1135 auto ptr =
reinterpret_cast<const tt::target::ttnn::ReshapeOp *
>(obj);
1136 return verifier.VerifyTable(ptr);
1139 auto ptr =
reinterpret_cast<const tt::target::ttnn::SliceOp *
>(obj);
1140 return verifier.VerifyTable(ptr);
1143 auto ptr =
reinterpret_cast<const tt::target::ttnn::TransposeOp *
>(obj);
1144 return verifier.VerifyTable(ptr);
1147 auto ptr =
reinterpret_cast<const tt::target::ttnn::DeallocateOp *
>(obj);
1148 return verifier.VerifyTable(ptr);
1151 auto ptr =
reinterpret_cast<const tt::target::ttnn::EltwiseBinaryOp *
>(obj);
1152 return verifier.VerifyTable(ptr);
1155 auto ptr =
reinterpret_cast<const tt::target::ttnn::EltwiseBinaryCompositeOp *
>(obj);
1156 return verifier.VerifyTable(ptr);
1159 auto ptr =
reinterpret_cast<const tt::target::ttnn::EltwiseTernaryWhereOp *
>(obj);
1160 return verifier.VerifyTable(ptr);
1163 auto ptr =
reinterpret_cast<const tt::target::ttnn::EltwiseQuantizationOp *
>(obj);
1164 return verifier.VerifyTable(ptr);
1167 auto ptr =
reinterpret_cast<const tt::target::ttnn::EltwiseUnaryOp *
>(obj);
1168 return verifier.VerifyTable(ptr);
1171 auto ptr =
reinterpret_cast<const tt::target::ttnn::EltwiseUnaryCompositeOp *
>(obj);
1172 return verifier.VerifyTable(ptr);
1175 auto ptr =
reinterpret_cast<const tt::target::ttnn::EmbeddingBackwardOp *
>(obj);
1176 return verifier.VerifyTable(ptr);
1179 auto ptr =
reinterpret_cast<const tt::target::ttnn::EmbeddingOp *
>(obj);
1180 return verifier.VerifyTable(ptr);
1183 auto ptr =
reinterpret_cast<const tt::target::ttnn::FillCacheOp *
>(obj);
1184 return verifier.VerifyTable(ptr);
1187 auto ptr =
reinterpret_cast<const tt::target::ttnn::UpdateCacheOp *
>(obj);
1188 return verifier.VerifyTable(ptr);
1191 auto ptr =
reinterpret_cast<const tt::target::ttnn::FromDeviceOp *
>(obj);
1192 return verifier.VerifyTable(ptr);
1195 auto ptr =
reinterpret_cast<const tt::target::ttnn::ToDeviceOp *
>(obj);
1196 return verifier.VerifyTable(ptr);
1199 auto ptr =
reinterpret_cast<const tt::target::ttnn::ToDTypeOp *
>(obj);
1200 return verifier.VerifyTable(ptr);
1203 auto ptr =
reinterpret_cast<const tt::target::ttnn::ToLayoutOp *
>(obj);
1204 return verifier.VerifyTable(ptr);
1207 auto ptr =
reinterpret_cast<const tt::target::ttnn::ToMemoryConfigOp *
>(obj);
1208 return verifier.VerifyTable(ptr);
1211 auto ptr =
reinterpret_cast<const tt::target::ttnn::TypecastOp *
>(obj);
1212 return verifier.VerifyTable(ptr);
1215 auto ptr =
reinterpret_cast<const tt::target::ttnn::LinearOp *
>(obj);
1216 return verifier.VerifyTable(ptr);
1219 auto ptr =
reinterpret_cast<const tt::target::ttnn::MatmulOp *
>(obj);
1220 return verifier.VerifyTable(ptr);
1223 auto ptr =
reinterpret_cast<const tt::target::ttnn::MorehCumSumOp *
>(obj);
1224 return verifier.VerifyTable(ptr);
1227 auto ptr =
reinterpret_cast<const tt::target::ttnn::SoftmaxOp *
>(obj);
1228 return verifier.VerifyTable(ptr);
1231 auto ptr =
reinterpret_cast<const tt::target::ttnn::Pool2dOp *
>(obj);
1232 return verifier.VerifyTable(ptr);
1235 auto ptr =
reinterpret_cast<const tt::target::ttnn::UpsampleOp *
>(obj);
1236 return verifier.VerifyTable(ptr);
1239 auto ptr =
reinterpret_cast<const tt::target::ttnn::ReductionArgMaxOp *
>(obj);
1240 return verifier.VerifyTable(ptr);
1243 auto ptr =
reinterpret_cast<const tt::target::ttnn::ReductionOp *
>(obj);
1244 return verifier.VerifyTable(ptr);
1247 auto ptr =
reinterpret_cast<const tt::target::ttnn::ReductionProdOp *
>(obj);
1248 return verifier.VerifyTable(ptr);
1251 auto ptr =
reinterpret_cast<const tt::target::ttnn::LoadCachedOp *
>(obj);
1252 return verifier.VerifyTable(ptr);
1255 auto ptr =
reinterpret_cast<const tt::target::ttnn::BatchNormOp *
>(obj);
1256 return verifier.VerifyTable(ptr);
1259 auto ptr =
reinterpret_cast<const tt::target::ttnn::TraceOp *
>(obj);
1260 return verifier.VerifyTable(ptr);
1262 default:
return true;
1266 inline bool VerifyOpTypeVector(::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset<void>> *values, const ::flatbuffers::Vector<OpType> *types) {
1267 if (!values || !types)
return !values && !types;
1268 if (values->size() != types->size())
return false;
1269 for (::flatbuffers::uoffset_t i = 0; i < values->size(); ++i) {
1271 verifier, values->Get(i), types->GetEnum<
OpType>(i))) {
VT_PRIVATE_
Definition: program_generated.h:909
VT_DYLIBS
Definition: program_generated.h:907
VT_DEBUG_INFO
Definition: program_generated.h:448
VT_INPUTS
Definition: program_generated.h:904
VT_OUTPUTS
Definition: program_generated.h:905
VT_NAME
Definition: program_generated.h:903
VT_OPERATIONS
Definition: program_generated.h:906
bool VerifyOpTypeVector(::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset< void >> *values, const ::flatbuffers::Vector< OpType > *types)
Definition: program_generated.h:1266
OpType
Definition: program_generated.h:48
@ EltwiseUnaryCompositeOp
@ EltwiseBinaryCompositeOp
inline ::flatbuffers::Offset< Program > CreateProgram(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset<::flatbuffers::String > name=0, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::TensorRef >>> inputs=0, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::TensorRef >>> outputs=0, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::Operation >>> operations=0, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::DynamicLib >>> dylibs=0, ::flatbuffers::Offset< tt::target::DebugInfo > debug_info=0, bool private_=false, const tt::target::Dim2d *mesh_shape=nullptr)
Definition: program_generated.h:999
const OpType(& EnumValuesOpType())[52]
Definition: program_generated.h:105
const char *const * EnumNamesOpType()
Definition: program_generated.h:163
const char * EnumNameOpType(OpType e)
Definition: program_generated.h:222
bool VerifyOpType(::flatbuffers::Verifier &verifier, const void *obj, OpType type)
Definition: program_generated.h:1053
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:861
inline ::flatbuffers::Offset< Program > CreateProgramDirect(::flatbuffers::FlatBufferBuilder &_fbb, const char *name=nullptr, const std::vector<::flatbuffers::Offset< tt::target::ttnn::TensorRef >> *inputs=nullptr, const std::vector<::flatbuffers::Offset< tt::target::ttnn::TensorRef >> *outputs=nullptr, const std::vector<::flatbuffers::Offset< tt::target::ttnn::Operation >> *operations=nullptr, const std::vector<::flatbuffers::Offset< tt::target::DynamicLib >> *dylibs=nullptr, ::flatbuffers::Offset< tt::target::DebugInfo > debug_info=0, bool private_=false, const tt::target::Dim2d *mesh_shape=nullptr)
Definition: program_generated.h:1026
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:880
Definition: debug_info_bfbs_hash_generated.h:4
Definition: debug_info_generated.h:36
Definition: binary_generated.h:31
const tt::target::ttnn::ReshapeOp * type_as_ReshapeOp() const
Definition: program_generated.h:515
const ::flatbuffers::String * name() const
Definition: program_generated.h:912
const tt::target::ttnn::SoftmaxOp * type_as_SoftmaxOp() const
Definition: program_generated.h:584
const tt::target::ttnn::CpuOp * type_as_CpuOp() const
Definition: program_generated.h:482
const tt::target::ttnn::ReductionOp * type_as_ReductionOp() const
Definition: program_generated.h:596
const tt::target::ttnn::EmbeddingOp * type_as_EmbeddingOp() const
Definition: program_generated.h:548
const tt::target::ttnn::ConcatOp * type_as_ConcatOp() const
Definition: program_generated.h:500
const void * type() const
Definition: program_generated.h:454
ProgramBuilder Builder
Definition: program_generated.h:897
const ::flatbuffers::Vector<::flatbuffers::Offset< tt::target::DynamicLib > > * dylibs() const
Definition: program_generated.h:924
const tt::target::ttnn::EmptyOp * type_as_EmptyOp() const
Definition: program_generated.h:491
const tt::target::ttnn::PrepareConv2dWeightsOp * type_as_PrepareConv2dWeightsOp() const
Definition: program_generated.h:473
const ::flatbuffers::String * debug_info() const
Definition: program_generated.h:611
const tt::target::ttnn::TraceOp * type_as_TraceOp() const
Definition: program_generated.h:608
const tt::target::ttnn::SliceOp * type_as_SliceOp() const
Definition: program_generated.h:518
const tt::target::ttnn::DeallocateOp * type_as_DeallocateOp() const
Definition: program_generated.h:524
const tt::target::ttnn::Pool2dOp * type_as_Pool2dOp() const
Definition: program_generated.h:587
const T * type_as() const
const tt::target::ttnn::RepeatOp * type_as_RepeatOp() const
Definition: program_generated.h:512
bool private_() const
Definition: program_generated.h:930
const tt::target::ttnn::PadOp * type_as_PadOp() const
Definition: program_generated.h:503
const tt::target::ttnn::MeshShardOp * type_as_MeshShardOp() const
Definition: program_generated.h:464
const tt::target::ttnn::ToMemoryConfigOp * type_as_ToMemoryConfigOp() const
Definition: program_generated.h:569
const tt::target::ttnn::CollectivePermuteOp * type_as_CollectivePermuteOp() const
Definition: program_generated.h:461
const tt::target::ttnn::EmbeddingBackwardOp * type_as_EmbeddingBackwardOp() const
Definition: program_generated.h:545
const tt::target::ttnn::ToLayoutOp * type_as_ToLayoutOp() const
Definition: program_generated.h:566
const tt::target::ttnn::ArangeOp * type_as_ArangeOp() const
Definition: program_generated.h:485
const tt::target::ttnn::UpsampleOp * type_as_UpsampleOp() const
Definition: program_generated.h:590
const tt::target::ttnn::FromDeviceOp * type_as_FromDeviceOp() const
Definition: program_generated.h:557
const ::flatbuffers::String * loc_info() const
Definition: program_generated.h:614
const tt::target::ttnn::TransposeOp * type_as_TransposeOp() const
Definition: program_generated.h:521
const tt::target::ttnn::FullOp * type_as_FullOp() const
Definition: program_generated.h:494
const tt::target::ttnn::RepeatInterleaveOp * type_as_RepeatInterleaveOp() const
Definition: program_generated.h:509
const tt::target::ttnn::EltwiseUnaryCompositeOp * type_as_EltwiseUnaryCompositeOp() const
Definition: program_generated.h:542
const tt::target::ttnn::LinearOp * type_as_LinearOp() const
Definition: program_generated.h:575
const tt::target::ttnn::TypecastOp * type_as_TypecastOp() const
Definition: program_generated.h:572
tt::target::ttnn::OpType type_type() const
Definition: program_generated.h:451
const tt::target::ttnn::MatmulOp * type_as_MatmulOp() const
Definition: program_generated.h:578
const ::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::Operation > > * operations() const
Definition: program_generated.h:921
bool Verify(::flatbuffers::Verifier &verifier) const
Definition: program_generated.h:617
const tt::target::ttnn::LoadCachedOp * type_as_LoadCachedOp() const
Definition: program_generated.h:602
const tt::target::ttnn::AllGatherOp * type_as_AllGatherOp() const
Definition: program_generated.h:458
const tt::target::ttnn::Conv2dOp * type_as_Conv2dOp() const
Definition: program_generated.h:476
const ::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::TensorRef > > * outputs() const
Definition: program_generated.h:918
const tt::target::ttnn::PermuteOp * type_as_PermuteOp() const
Definition: program_generated.h:506
const tt::target::ttnn::ToDTypeOp * type_as_ToDTypeOp() const
Definition: program_generated.h:563
OperationBuilder Builder
Definition: program_generated.h:440
const tt::target::ttnn::FillCacheOp * type_as_FillCacheOp() const
Definition: program_generated.h:551
const tt::target::ttnn::NamedFullOp * type_as_NamedFullOp() const
Definition: program_generated.h:497
const tt::target::ttnn::EltwiseBinaryCompositeOp * type_as_EltwiseBinaryCompositeOp() const
Definition: program_generated.h:530
const tt::target::ttnn::EltwiseQuantizationOp * type_as_EltwiseQuantizationOp() const
Definition: program_generated.h:536
const tt::target::Dim2d * mesh_shape() const
Definition: program_generated.h:933
const tt::target::ttnn::EltwiseTernaryWhereOp * type_as_EltwiseTernaryWhereOp() const
Definition: program_generated.h:533
const ::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::TensorRef > > * inputs() const
Definition: program_generated.h:915
const tt::target::ttnn::MorehCumSumOp * type_as_MorehCumSumOp() const
Definition: program_generated.h:581
const tt::target::DebugInfo * debug_info() const
Definition: program_generated.h:927
const tt::target::ttnn::BatchNormOp * type_as_BatchNormOp() const
Definition: program_generated.h:605
const tt::target::ttnn::ReductionProdOp * type_as_ReductionProdOp() const
Definition: program_generated.h:599
const tt::target::ttnn::EltwiseBinaryOp * type_as_EltwiseBinaryOp() const
Definition: program_generated.h:527
const tt::target::ttnn::ReductionArgMaxOp * type_as_ReductionArgMaxOp() const
Definition: program_generated.h:593
const tt::target::ttnn::ConvTranspose2dOp * type_as_ConvTranspose2dOp() const
Definition: program_generated.h:479
static FLATBUFFERS_CONSTEXPR_CPP11 const char * GetFullyQualifiedName()
Definition: program_generated.h:442
const tt::target::ttnn::EltwiseUnaryOp * type_as_EltwiseUnaryOp() const
Definition: program_generated.h:539
const tt::target::ttnn::ConstantOp * type_as_ConstantOp() const
Definition: program_generated.h:488
const tt::target::ttnn::ToDeviceOp * type_as_ToDeviceOp() const
Definition: program_generated.h:560
const tt::target::ttnn::ReduceScatterOp * type_as_ReduceScatterOp() const
Definition: program_generated.h:467
const tt::target::ttnn::UpdateCacheOp * type_as_UpdateCacheOp() const
Definition: program_generated.h:554
const tt::target::ttnn::GetDeviceOp * type_as_GetDeviceOp() const
Definition: program_generated.h:470
Definition: program_generated.h:228
static const OpType enum_value
Definition: program_generated.h:229
Definition: program_generated.h:834
void add_type(::flatbuffers::Offset< void > type)
Definition: program_generated.h:841
OperationBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:850
::flatbuffers::Offset< Operation > Finish()
Definition: program_generated.h:854
Operation Table
Definition: program_generated.h:835
void add_debug_info(::flatbuffers::Offset<::flatbuffers::String > debug_info)
Definition: program_generated.h:844
void add_loc_info(::flatbuffers::Offset<::flatbuffers::String > loc_info)
Definition: program_generated.h:847
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:836
::flatbuffers::uoffset_t start_
Definition: program_generated.h:837
void add_type_type(tt::target::ttnn::OpType type_type)
Definition: program_generated.h:838
Definition: program_generated.h:875
static constexpr auto Create
Definition: program_generated.h:877
Operation type
Definition: program_generated.h:876
Definition: program_generated.h:960
Program Table
Definition: program_generated.h:961
::flatbuffers::uoffset_t start_
Definition: program_generated.h:963
void add_name(::flatbuffers::Offset<::flatbuffers::String > name)
Definition: program_generated.h:964
void add_debug_info(::flatbuffers::Offset< tt::target::DebugInfo > debug_info)
Definition: program_generated.h:979
void add_dylibs(::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::DynamicLib >>> dylibs)
Definition: program_generated.h:976
void add_outputs(::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::TensorRef >>> outputs)
Definition: program_generated.h:970
::flatbuffers::FlatBufferBuilder & fbb_
Definition: program_generated.h:962
void add_inputs(::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::TensorRef >>> inputs)
Definition: program_generated.h:967
void add_private_(bool private_)
Definition: program_generated.h:982
::flatbuffers::Offset< Program > Finish()
Definition: program_generated.h:992
ProgramBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: program_generated.h:988
void add_operations(::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::Operation >>> operations)
Definition: program_generated.h:973
void add_mesh_shape(const tt::target::Dim2d *mesh_shape)
Definition: program_generated.h:985
Definition: program_generated.h:1021
static constexpr auto Create
Definition: program_generated.h:1023
Program type
Definition: program_generated.h:1022