4 #ifndef FLATBUFFERS_GENERATED_CONV_TT_TARGET_TTNN_H_
5 #define FLATBUFFERS_GENERATED_CONV_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 Conv2dConfigBuilder;
27 struct PrepareConv2dWeightsOpBuilder;
30 struct Conv2dOpBuilder;
33 struct ConvTranspose2dOpBuilder;
35 struct Conv2dConfig FLATBUFFERS_FINAL_CLASS :
private ::flatbuffers::Table {
39 return "tt.target.ttnn.Conv2dConfig";
41 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
60 VT_ENABLE_SUBBLOCK_PADDING = 40
62 ::flatbuffers::Optional<tt::target::DataType>
dtype()
const {
63 return GetOptional<uint16_t, tt::target::DataType>(
VT_DTYPE);
69 return GetPointer<const ::flatbuffers::String *>(
VT_ACTIVATION);
89 ::flatbuffers::Optional<tt::target::ttnn::TensorMemoryLayout>
shard_layout()
const {
90 return GetOptional<uint16_t, tt::target::ttnn::TensorMemoryLayout>(
VT_SHARD_LAYOUT);
92 const tt::target::ttnn::CoreRangeSet *
core_grid()
const {
93 return GetPointer<const tt::target::ttnn::CoreRangeSet *>(
VT_CORE_GRID);
98 ::flatbuffers::Optional<tt::target::TensorLayout>
output_layout()
const {
117 return GetOptional<uint8_t, bool>(VT_ENABLE_SUBBLOCK_PADDING);
119 bool Verify(::flatbuffers::Verifier &verifier)
const {
120 return VerifyTableStart(verifier) &&
121 VerifyField<uint16_t>(verifier,
VT_DTYPE, 2) &&
124 verifier.VerifyString(activation()) &&
133 verifier.VerifyTable(core_grid()) &&
141 VerifyField<uint8_t>(verifier, VT_ENABLE_SUBBLOCK_PADDING, 1) &&
148 ::flatbuffers::FlatBufferBuilder &
fbb_;
180 void add_core_grid(::flatbuffers::Offset<tt::target::ttnn::CoreRangeSet> core_grid) {
205 fbb_.AddElement<uint8_t>(Conv2dConfig::VT_ENABLE_SUBBLOCK_PADDING,
static_cast<uint8_t
>(enable_subblock_padding));
211 ::flatbuffers::Offset<Conv2dConfig>
Finish() {
213 auto o = ::flatbuffers::Offset<Conv2dConfig>(end);
219 ::flatbuffers::FlatBufferBuilder &_fbb,
220 ::flatbuffers::Optional<tt::target::DataType> dtype = ::flatbuffers::nullopt,
221 ::flatbuffers::Optional<tt::target::DataType> weights_dtype = ::flatbuffers::nullopt,
222 ::flatbuffers::Offset<::flatbuffers::String> activation = 0,
223 ::flatbuffers::Optional<bool> deallocate_activation = ::flatbuffers::nullopt,
224 ::flatbuffers::Optional<bool> reallocate_halo_output = ::flatbuffers::nullopt,
225 ::flatbuffers::Optional<uint32_t> act_block_h_override = ::flatbuffers::nullopt,
226 ::flatbuffers::Optional<uint32_t> act_block_w_div = ::flatbuffers::nullopt,
227 ::flatbuffers::Optional<bool> reshard_if_not_optimal = ::flatbuffers::nullopt,
228 ::flatbuffers::Optional<bool> override_sharding_config = ::flatbuffers::nullopt,
229 ::flatbuffers::Optional<tt::target::ttnn::TensorMemoryLayout> shard_layout = ::flatbuffers::nullopt,
230 ::flatbuffers::Offset<tt::target::ttnn::CoreRangeSet> core_grid = 0,
231 ::flatbuffers::Optional<bool> transpose_shards = ::flatbuffers::nullopt,
232 ::flatbuffers::Optional<tt::target::TensorLayout> output_layout = ::flatbuffers::nullopt,
233 ::flatbuffers::Optional<bool> preprocess_weights_on_device = ::flatbuffers::nullopt,
234 ::flatbuffers::Optional<bool> always_preprocess_weights = ::flatbuffers::nullopt,
235 ::flatbuffers::Optional<bool> enable_act_double_buffer = ::flatbuffers::nullopt,
236 ::flatbuffers::Optional<bool> enable_weights_double_buffer = ::flatbuffers::nullopt,
237 ::flatbuffers::Optional<bool> enable_split_reader = ::flatbuffers::nullopt,
238 ::flatbuffers::Optional<bool> enable_subblock_padding = ::flatbuffers::nullopt) {
247 if(dtype) { builder_.
add_dtype(*dtype); }
268 ::flatbuffers::FlatBufferBuilder &_fbb,
269 ::flatbuffers::Optional<tt::target::DataType> dtype = ::flatbuffers::nullopt,
270 ::flatbuffers::Optional<tt::target::DataType> weights_dtype = ::flatbuffers::nullopt,
271 const char *activation =
nullptr,
272 ::flatbuffers::Optional<bool> deallocate_activation = ::flatbuffers::nullopt,
273 ::flatbuffers::Optional<bool> reallocate_halo_output = ::flatbuffers::nullopt,
274 ::flatbuffers::Optional<uint32_t> act_block_h_override = ::flatbuffers::nullopt,
275 ::flatbuffers::Optional<uint32_t> act_block_w_div = ::flatbuffers::nullopt,
276 ::flatbuffers::Optional<bool> reshard_if_not_optimal = ::flatbuffers::nullopt,
277 ::flatbuffers::Optional<bool> override_sharding_config = ::flatbuffers::nullopt,
278 ::flatbuffers::Optional<tt::target::ttnn::TensorMemoryLayout> shard_layout = ::flatbuffers::nullopt,
279 ::flatbuffers::Offset<tt::target::ttnn::CoreRangeSet> core_grid = 0,
280 ::flatbuffers::Optional<bool> transpose_shards = ::flatbuffers::nullopt,
281 ::flatbuffers::Optional<tt::target::TensorLayout> output_layout = ::flatbuffers::nullopt,
282 ::flatbuffers::Optional<bool> preprocess_weights_on_device = ::flatbuffers::nullopt,
283 ::flatbuffers::Optional<bool> always_preprocess_weights = ::flatbuffers::nullopt,
284 ::flatbuffers::Optional<bool> enable_act_double_buffer = ::flatbuffers::nullopt,
285 ::flatbuffers::Optional<bool> enable_weights_double_buffer = ::flatbuffers::nullopt,
286 ::flatbuffers::Optional<bool> enable_split_reader = ::flatbuffers::nullopt,
287 ::flatbuffers::Optional<bool> enable_subblock_padding = ::flatbuffers::nullopt) {
288 auto activation__ = activation ? _fbb.CreateString(activation) : 0;
294 deallocate_activation,
295 reallocate_halo_output,
296 act_block_h_override,
298 reshard_if_not_optimal,
299 override_sharding_config,
304 preprocess_weights_on_device,
305 always_preprocess_weights,
306 enable_act_double_buffer,
307 enable_weights_double_buffer,
309 enable_subblock_padding);
316 return "tt.target.ttnn.PrepareConv2dWeightsOp";
318 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
336 VT_CONV2D_CONFIG = 38
341 const tt::target::ttnn::TensorRef *
out()
const {
342 return GetPointer<const tt::target::ttnn::TensorRef *>(
VT_OUT);
369 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_KERNEL_SIZE);
371 const ::flatbuffers::Vector<int32_t> *
stride()
const {
372 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_STRIDE);
374 const ::flatbuffers::Vector<int32_t> *
padding()
const {
375 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_PADDING);
377 const ::flatbuffers::Vector<int32_t> *
dilation()
const {
378 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_DILATION);
386 const tt::target::DeviceRef *
device()
const {
387 return GetPointer<const tt::target::DeviceRef *>(
VT_DEVICE);
390 return GetPointer<const tt::target::ttnn::Conv2dConfig *>(VT_CONV2D_CONFIG);
392 bool Verify(::flatbuffers::Verifier &verifier)
const {
393 return VerifyTableStart(verifier) &&
395 verifier.VerifyTable(weight_tensor()) &&
396 VerifyOffset(verifier,
VT_OUT) &&
397 verifier.VerifyTable(out()) &&
399 verifier.VerifyTable(input_memory_config()) &&
402 verifier.VerifyString(weights_format()) &&
409 verifier.VerifyVector(kernel_size()) &&
411 verifier.VerifyVector(stride()) &&
413 verifier.VerifyVector(padding()) &&
415 verifier.VerifyVector(dilation()) &&
417 VerifyField<uint32_t>(verifier,
VT_GROUPS, 4) &&
419 verifier.VerifyTable(device()) &&
420 VerifyOffset(verifier, VT_CONV2D_CONFIG) &&
421 verifier.VerifyTable(conv2d_config()) &&
428 ::flatbuffers::FlatBufferBuilder &
fbb_;
433 void add_out(::flatbuffers::Offset<tt::target::ttnn::TensorRef> out) {
460 void add_kernel_size(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> kernel_size) {
463 void add_stride(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> stride) {
466 void add_padding(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> padding) {
469 void add_dilation(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> dilation) {
478 void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
482 fbb_.AddOffset(PrepareConv2dWeightsOp::VT_CONV2D_CONFIG, conv2d_config);
488 ::flatbuffers::Offset<PrepareConv2dWeightsOp>
Finish() {
490 auto o = ::flatbuffers::Offset<PrepareConv2dWeightsOp>(end);
496 ::flatbuffers::FlatBufferBuilder &_fbb,
497 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> weight_tensor = 0,
498 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
499 ::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> input_memory_config = 0,
501 ::flatbuffers::Offset<::flatbuffers::String> weights_format = 0,
502 uint32_t in_channels = 0,
503 uint32_t out_channels = 0,
504 uint32_t batch_size = 0,
505 uint32_t input_height = 0,
506 uint32_t input_width = 0,
507 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> kernel_size = 0,
508 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> stride = 0,
509 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> padding = 0,
510 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> dilation = 0,
511 bool has_bias =
false,
513 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
514 ::flatbuffers::Offset<tt::target::ttnn::Conv2dConfig> conv2d_config = 0) {
543 ::flatbuffers::FlatBufferBuilder &_fbb,
544 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> weight_tensor = 0,
545 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
546 ::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> input_memory_config = 0,
548 const char *weights_format =
nullptr,
549 uint32_t in_channels = 0,
550 uint32_t out_channels = 0,
551 uint32_t batch_size = 0,
552 uint32_t input_height = 0,
553 uint32_t input_width = 0,
554 const std::vector<int32_t> *kernel_size =
nullptr,
555 const std::vector<int32_t> *stride =
nullptr,
556 const std::vector<int32_t> *padding =
nullptr,
557 const std::vector<int32_t> *dilation =
nullptr,
558 bool has_bias =
false,
560 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
561 ::flatbuffers::Offset<tt::target::ttnn::Conv2dConfig> conv2d_config = 0) {
562 auto weights_format__ = weights_format ? _fbb.CreateString(weights_format) : 0;
563 auto kernel_size__ = kernel_size ? _fbb.CreateVector<int32_t>(*kernel_size) : 0;
564 auto stride__ = stride ? _fbb.CreateVector<int32_t>(*stride) : 0;
565 auto padding__ = padding ? _fbb.CreateVector<int32_t>(*padding) : 0;
566 auto dilation__ = dilation ? _fbb.CreateVector<int32_t>(*dilation) : 0;
593 return "tt.target.ttnn.Conv2dOp";
595 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
611 VT_CONV2D_CONFIG = 34
613 const tt::target::ttnn::TensorRef *
input()
const {
614 return GetPointer<const tt::target::ttnn::TensorRef *>(
VT_INPUT);
616 const tt::target::ttnn::TensorRef *
weight()
const {
617 return GetPointer<const tt::target::ttnn::TensorRef *>(
VT_WEIGHT);
619 const tt::target::ttnn::TensorRef *
bias()
const {
620 return GetPointer<const tt::target::ttnn::TensorRef *>(
VT_BIAS);
622 const tt::target::ttnn::TensorRef *
out()
const {
623 return GetPointer<const tt::target::ttnn::TensorRef *>(
VT_OUT);
625 const tt::target::DeviceRef *
device()
const {
626 return GetPointer<const tt::target::DeviceRef *>(
VT_DEVICE);
644 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_KERNEL_SIZE);
646 const ::flatbuffers::Vector<int32_t> *
stride()
const {
647 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_STRIDE);
649 const ::flatbuffers::Vector<int32_t> *
padding()
const {
650 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_PADDING);
652 const ::flatbuffers::Vector<int32_t> *
dilation()
const {
653 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_DILATION);
659 return GetPointer<const tt::target::ttnn::Conv2dConfig *>(VT_CONV2D_CONFIG);
661 bool Verify(::flatbuffers::Verifier &verifier)
const {
662 return VerifyTableStart(verifier) &&
664 verifier.VerifyTable(input()) &&
666 verifier.VerifyTable(weight()) &&
667 VerifyOffset(verifier,
VT_BIAS) &&
668 verifier.VerifyTable(bias()) &&
669 VerifyOffset(verifier,
VT_OUT) &&
670 verifier.VerifyTable(out()) &&
672 verifier.VerifyTable(device()) &&
679 verifier.VerifyVector(kernel_size()) &&
681 verifier.VerifyVector(stride()) &&
683 verifier.VerifyVector(padding()) &&
685 verifier.VerifyVector(dilation()) &&
686 VerifyField<uint32_t>(verifier,
VT_GROUPS, 4) &&
687 VerifyOffset(verifier, VT_CONV2D_CONFIG) &&
688 verifier.VerifyTable(conv2d_config()) &&
695 ::flatbuffers::FlatBufferBuilder &
fbb_;
697 void add_input(::flatbuffers::Offset<tt::target::ttnn::TensorRef> input) {
700 void add_weight(::flatbuffers::Offset<tt::target::ttnn::TensorRef> weight) {
703 void add_bias(::flatbuffers::Offset<tt::target::ttnn::TensorRef> bias) {
706 void add_out(::flatbuffers::Offset<tt::target::ttnn::TensorRef> out) {
709 void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
727 void add_kernel_size(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> kernel_size) {
730 void add_stride(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> stride) {
733 void add_padding(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> padding) {
736 void add_dilation(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> dilation) {
743 fbb_.AddOffset(Conv2dOp::VT_CONV2D_CONFIG, conv2d_config);
749 ::flatbuffers::Offset<Conv2dOp>
Finish() {
751 auto o = ::flatbuffers::Offset<Conv2dOp>(end);
757 ::flatbuffers::FlatBufferBuilder &_fbb,
758 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> input = 0,
759 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> weight = 0,
760 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> bias = 0,
761 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
762 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
763 uint32_t in_channels = 0,
764 uint32_t out_channels = 0,
765 uint32_t batch_size = 0,
766 uint32_t input_height = 0,
767 uint32_t input_width = 0,
768 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> kernel_size = 0,
769 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> stride = 0,
770 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> padding = 0,
771 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> dilation = 0,
773 ::flatbuffers::Offset<tt::target::ttnn::Conv2dConfig> conv2d_config = 0) {
800 ::flatbuffers::FlatBufferBuilder &_fbb,
801 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> input = 0,
802 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> weight = 0,
803 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> bias = 0,
804 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
805 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
806 uint32_t in_channels = 0,
807 uint32_t out_channels = 0,
808 uint32_t batch_size = 0,
809 uint32_t input_height = 0,
810 uint32_t input_width = 0,
811 const std::vector<int32_t> *kernel_size =
nullptr,
812 const std::vector<int32_t> *stride =
nullptr,
813 const std::vector<int32_t> *padding =
nullptr,
814 const std::vector<int32_t> *dilation =
nullptr,
816 ::flatbuffers::Offset<tt::target::ttnn::Conv2dConfig> conv2d_config = 0) {
817 auto kernel_size__ = kernel_size ? _fbb.CreateVector<int32_t>(*kernel_size) : 0;
818 auto stride__ = stride ? _fbb.CreateVector<int32_t>(*stride) : 0;
819 auto padding__ = padding ? _fbb.CreateVector<int32_t>(*padding) : 0;
820 auto dilation__ = dilation ? _fbb.CreateVector<int32_t>(*dilation) : 0;
845 return "tt.target.ttnn.ConvTranspose2dOp";
847 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
866 const tt::target::ttnn::TensorRef *
input()
const {
867 return GetPointer<const tt::target::ttnn::TensorRef *>(
VT_INPUT);
869 const tt::target::ttnn::TensorRef *
weight()
const {
870 return GetPointer<const tt::target::ttnn::TensorRef *>(
VT_WEIGHT);
872 const tt::target::ttnn::TensorRef *
bias()
const {
873 return GetPointer<const tt::target::ttnn::TensorRef *>(
VT_BIAS);
875 const tt::target::ttnn::TensorRef *
out()
const {
876 return GetPointer<const tt::target::ttnn::TensorRef *>(
VT_OUT);
878 const tt::target::DeviceRef *
device()
const {
879 return GetPointer<const tt::target::DeviceRef *>(
VT_DEVICE);
897 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_KERNEL_SIZE);
899 const ::flatbuffers::Vector<int32_t> *
stride()
const {
900 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_STRIDE);
902 const ::flatbuffers::Vector<int32_t> *
padding()
const {
903 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_PADDING);
908 const ::flatbuffers::Vector<int32_t> *
dilation()
const {
909 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_DILATION);
915 return GetPointer<const tt::target::ttnn::MemoryConfig *>(
VT_MEMORY_CONFIG);
917 bool Verify(::flatbuffers::Verifier &verifier)
const {
918 return VerifyTableStart(verifier) &&
920 verifier.VerifyTable(input()) &&
922 verifier.VerifyTable(weight()) &&
923 VerifyOffset(verifier,
VT_BIAS) &&
924 verifier.VerifyTable(bias()) &&
925 VerifyOffset(verifier,
VT_OUT) &&
926 verifier.VerifyTable(out()) &&
928 verifier.VerifyTable(device()) &&
935 verifier.VerifyVector(kernel_size()) &&
937 verifier.VerifyVector(stride()) &&
939 verifier.VerifyVector(padding()) &&
941 verifier.VerifyVector(output_padding()) &&
943 verifier.VerifyVector(dilation()) &&
944 VerifyField<uint32_t>(verifier,
VT_GROUPS, 4) &&
946 verifier.VerifyTable(memory_config()) &&
953 ::flatbuffers::FlatBufferBuilder &
fbb_;
955 void add_input(::flatbuffers::Offset<tt::target::ttnn::TensorRef> input) {
958 void add_weight(::flatbuffers::Offset<tt::target::ttnn::TensorRef> weight) {
961 void add_bias(::flatbuffers::Offset<tt::target::ttnn::TensorRef> bias) {
964 void add_out(::flatbuffers::Offset<tt::target::ttnn::TensorRef> out) {
967 void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
985 void add_kernel_size(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> kernel_size) {
988 void add_stride(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> stride) {
991 void add_padding(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> padding) {
997 void add_dilation(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> dilation) {
1010 ::flatbuffers::Offset<ConvTranspose2dOp>
Finish() {
1012 auto o = ::flatbuffers::Offset<ConvTranspose2dOp>(end);
1018 ::flatbuffers::FlatBufferBuilder &_fbb,
1019 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> input = 0,
1020 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> weight = 0,
1021 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> bias = 0,
1022 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
1023 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
1024 uint32_t in_channels = 0,
1025 uint32_t out_channels = 0,
1026 uint32_t batch_size = 0,
1027 uint32_t input_height = 0,
1028 uint32_t input_width = 0,
1029 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> kernel_size = 0,
1030 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> stride = 0,
1031 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> padding = 0,
1032 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> output_padding = 0,
1033 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> dilation = 0,
1034 uint32_t groups = 0,
1035 ::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> memory_config = 0) {
1054 return builder_.
Finish();
1063 ::flatbuffers::FlatBufferBuilder &_fbb,
1064 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> input = 0,
1065 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> weight = 0,
1066 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> bias = 0,
1067 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
1068 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
1069 uint32_t in_channels = 0,
1070 uint32_t out_channels = 0,
1071 uint32_t batch_size = 0,
1072 uint32_t input_height = 0,
1073 uint32_t input_width = 0,
1074 const std::vector<int32_t> *kernel_size =
nullptr,
1075 const std::vector<int32_t> *stride =
nullptr,
1076 const std::vector<int32_t> *padding =
nullptr,
1077 const std::vector<int32_t> *output_padding =
nullptr,
1078 const std::vector<int32_t> *dilation =
nullptr,
1079 uint32_t groups = 0,
1080 ::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> memory_config = 0) {
1081 auto kernel_size__ = kernel_size ? _fbb.CreateVector<int32_t>(*kernel_size) : 0;
1082 auto stride__ = stride ? _fbb.CreateVector<int32_t>(*stride) : 0;
1083 auto padding__ = padding ? _fbb.CreateVector<int32_t>(*padding) : 0;
1084 auto output_padding__ = output_padding ? _fbb.CreateVector<int32_t>(*output_padding) : 0;
1085 auto dilation__ = dilation ? _fbb.CreateVector<int32_t>(*dilation) : 0;
VT_PREPROCESS_WEIGHTS_ON_DEVICE
Definition: conv_generated.h:55
VT_INPUT
Definition: conv_generated.h:596
VT_DTYPE
Definition: conv_generated.h:42
VT_REALLOCATE_HALO_OUTPUT
Definition: conv_generated.h:46
VT_DILATION
Definition: conv_generated.h:332
VT_WEIGHT_TENSOR
Definition: conv_generated.h:319
VT_ACT_BLOCK_W_DIV
Definition: conv_generated.h:48
VT_PADDING
Definition: conv_generated.h:331
VT_WEIGHT
Definition: conv_generated.h:597
VT_ENABLE_WEIGHTS_DOUBLE_BUFFER
Definition: conv_generated.h:58
VT_OVERRIDE_SHARDING_CONFIG
Definition: conv_generated.h:50
VT_STRIDE
Definition: conv_generated.h:330
VT_KERNEL_SIZE
Definition: conv_generated.h:329
VT_BATCH_SIZE
Definition: conv_generated.h:326
VT_CORE_GRID
Definition: conv_generated.h:52
VT_GROUPS
Definition: conv_generated.h:334
VT_WEIGHTS_FORMAT
Definition: conv_generated.h:323
VT_ENABLE_SPLIT_READER
Definition: conv_generated.h:59
VT_DEALLOCATE_ACTIVATION
Definition: conv_generated.h:45
VT_TRANSPOSE_SHARDS
Definition: conv_generated.h:53
VT_SHARD_LAYOUT
Definition: conv_generated.h:51
VT_OUTPUT_LAYOUT
Definition: conv_generated.h:54
VT_WEIGHTS_DTYPE
Definition: conv_generated.h:43
VT_INPUT_HEIGHT
Definition: conv_generated.h:327
VT_INPUT_MEMORY_CONFIG
Definition: conv_generated.h:321
VT_ENABLE_ACT_DOUBLE_BUFFER
Definition: conv_generated.h:57
VT_IN_CHANNELS
Definition: conv_generated.h:324
VT_DEVICE
Definition: conv_generated.h:335
VT_INPUT_WIDTH
Definition: conv_generated.h:328
VT_ACT_BLOCK_H_OVERRIDE
Definition: conv_generated.h:47
VT_BIAS
Definition: conv_generated.h:598
VT_OUT
Definition: conv_generated.h:320
VT_RESHARD_IF_NOT_OPTIMAL
Definition: conv_generated.h:49
VT_OUTPUT_PADDING
Definition: conv_generated.h:861
VT_ACTIVATION
Definition: conv_generated.h:44
VT_OUT_CHANNELS
Definition: conv_generated.h:325
VT_ALWAYS_PREPROCESS_WEIGHTS
Definition: conv_generated.h:56
VT_INPUT_TENSOR_LAYOUT
Definition: conv_generated.h:322
VT_HAS_BIAS
Definition: conv_generated.h:333
VT_MEMORY_CONFIG
Definition: data_movement_generated.h:278
inline ::flatbuffers::Offset< Conv2dConfig > CreateConv2dConfig(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Optional< tt::target::DataType > dtype=::flatbuffers::nullopt, ::flatbuffers::Optional< tt::target::DataType > weights_dtype=::flatbuffers::nullopt, ::flatbuffers::Offset<::flatbuffers::String > activation=0, ::flatbuffers::Optional< bool > deallocate_activation=::flatbuffers::nullopt, ::flatbuffers::Optional< bool > reallocate_halo_output=::flatbuffers::nullopt, ::flatbuffers::Optional< uint32_t > act_block_h_override=::flatbuffers::nullopt, ::flatbuffers::Optional< uint32_t > act_block_w_div=::flatbuffers::nullopt, ::flatbuffers::Optional< bool > reshard_if_not_optimal=::flatbuffers::nullopt, ::flatbuffers::Optional< bool > override_sharding_config=::flatbuffers::nullopt, ::flatbuffers::Optional< tt::target::ttnn::TensorMemoryLayout > shard_layout=::flatbuffers::nullopt, ::flatbuffers::Offset< tt::target::ttnn::CoreRangeSet > core_grid=0, ::flatbuffers::Optional< bool > transpose_shards=::flatbuffers::nullopt, ::flatbuffers::Optional< tt::target::TensorLayout > output_layout=::flatbuffers::nullopt, ::flatbuffers::Optional< bool > preprocess_weights_on_device=::flatbuffers::nullopt, ::flatbuffers::Optional< bool > always_preprocess_weights=::flatbuffers::nullopt, ::flatbuffers::Optional< bool > enable_act_double_buffer=::flatbuffers::nullopt, ::flatbuffers::Optional< bool > enable_weights_double_buffer=::flatbuffers::nullopt, ::flatbuffers::Optional< bool > enable_split_reader=::flatbuffers::nullopt, ::flatbuffers::Optional< bool > enable_subblock_padding=::flatbuffers::nullopt)
Definition: conv_generated.h:218
inline ::flatbuffers::Offset< Conv2dOp > CreateConv2dOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > input=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > bias=0, ::flatbuffers::Offset< tt::target::ttnn::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 >> dilation=0, uint32_t groups=0, ::flatbuffers::Offset< tt::target::ttnn::Conv2dConfig > conv2d_config=0)
Definition: conv_generated.h:756
inline ::flatbuffers::Offset< Conv2dOp > CreateConv2dOpDirect(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > input=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > bias=0, ::flatbuffers::Offset< tt::target::ttnn::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 > *dilation=nullptr, uint32_t groups=0, ::flatbuffers::Offset< tt::target::ttnn::Conv2dConfig > conv2d_config=0)
Definition: conv_generated.h:799
inline ::flatbuffers::Offset< ConvTranspose2dOp > CreateConvTranspose2dOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > input=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > bias=0, ::flatbuffers::Offset< tt::target::ttnn::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, ::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memory_config=0)
Definition: conv_generated.h:1017
inline ::flatbuffers::Offset< PrepareConv2dWeightsOp > CreatePrepareConv2dWeightsOp(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight_tensor=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0, ::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > input_memory_config=0, tt::target::TensorLayout input_tensor_layout=tt::target::TensorLayout::RowMajor, ::flatbuffers::Offset<::flatbuffers::String > weights_format=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 >> dilation=0, bool has_bias=false, uint32_t groups=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::ttnn::Conv2dConfig > conv2d_config=0)
Definition: conv_generated.h:495
inline ::flatbuffers::Offset< ConvTranspose2dOp > CreateConvTranspose2dOpDirect(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > input=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > bias=0, ::flatbuffers::Offset< tt::target::ttnn::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, ::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memory_config=0)
Definition: conv_generated.h:1062
inline ::flatbuffers::Offset< PrepareConv2dWeightsOp > CreatePrepareConv2dWeightsOpDirect(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight_tensor=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0, ::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > input_memory_config=0, tt::target::TensorLayout input_tensor_layout=tt::target::TensorLayout::RowMajor, const char *weights_format=nullptr, 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 > *dilation=nullptr, bool has_bias=false, uint32_t groups=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::ttnn::Conv2dConfig > conv2d_config=0)
Definition: conv_generated.h:542
TensorMemoryLayout
Definition: types_generated.h:62
inline ::flatbuffers::Offset< Conv2dConfig > CreateConv2dConfigDirect(::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Optional< tt::target::DataType > dtype=::flatbuffers::nullopt, ::flatbuffers::Optional< tt::target::DataType > weights_dtype=::flatbuffers::nullopt, const char *activation=nullptr, ::flatbuffers::Optional< bool > deallocate_activation=::flatbuffers::nullopt, ::flatbuffers::Optional< bool > reallocate_halo_output=::flatbuffers::nullopt, ::flatbuffers::Optional< uint32_t > act_block_h_override=::flatbuffers::nullopt, ::flatbuffers::Optional< uint32_t > act_block_w_div=::flatbuffers::nullopt, ::flatbuffers::Optional< bool > reshard_if_not_optimal=::flatbuffers::nullopt, ::flatbuffers::Optional< bool > override_sharding_config=::flatbuffers::nullopt, ::flatbuffers::Optional< tt::target::ttnn::TensorMemoryLayout > shard_layout=::flatbuffers::nullopt, ::flatbuffers::Offset< tt::target::ttnn::CoreRangeSet > core_grid=0, ::flatbuffers::Optional< bool > transpose_shards=::flatbuffers::nullopt, ::flatbuffers::Optional< tt::target::TensorLayout > output_layout=::flatbuffers::nullopt, ::flatbuffers::Optional< bool > preprocess_weights_on_device=::flatbuffers::nullopt, ::flatbuffers::Optional< bool > always_preprocess_weights=::flatbuffers::nullopt, ::flatbuffers::Optional< bool > enable_act_double_buffer=::flatbuffers::nullopt, ::flatbuffers::Optional< bool > enable_weights_double_buffer=::flatbuffers::nullopt, ::flatbuffers::Optional< bool > enable_split_reader=::flatbuffers::nullopt, ::flatbuffers::Optional< bool > enable_subblock_padding=::flatbuffers::nullopt)
Definition: conv_generated.h:267
TensorLayout
Definition: types_generated.h:250
DataType
Definition: types_generated.h:81
Definition: debug_info_generated.h:18
Definition: debug_info_generated.h:36
Definition: conv_generated.h:146
void add_enable_subblock_padding(bool enable_subblock_padding)
Definition: conv_generated.h:204
::flatbuffers::FlatBufferBuilder & fbb_
Definition: conv_generated.h:148
void add_activation(::flatbuffers::Offset<::flatbuffers::String > activation)
Definition: conv_generated.h:156
void add_enable_split_reader(bool enable_split_reader)
Definition: conv_generated.h:201
void add_reallocate_halo_output(bool reallocate_halo_output)
Definition: conv_generated.h:162
::flatbuffers::uoffset_t start_
Definition: conv_generated.h:149
void add_weights_dtype(tt::target::DataType weights_dtype)
Definition: conv_generated.h:153
void add_enable_act_double_buffer(bool enable_act_double_buffer)
Definition: conv_generated.h:195
void add_shard_layout(tt::target::ttnn::TensorMemoryLayout shard_layout)
Definition: conv_generated.h:177
void add_core_grid(::flatbuffers::Offset< tt::target::ttnn::CoreRangeSet > core_grid)
Definition: conv_generated.h:180
void add_transpose_shards(bool transpose_shards)
Definition: conv_generated.h:183
void add_override_sharding_config(bool override_sharding_config)
Definition: conv_generated.h:174
void add_preprocess_weights_on_device(bool preprocess_weights_on_device)
Definition: conv_generated.h:189
void add_enable_weights_double_buffer(bool enable_weights_double_buffer)
Definition: conv_generated.h:198
::flatbuffers::Offset< Conv2dConfig > Finish()
Definition: conv_generated.h:211
void add_reshard_if_not_optimal(bool reshard_if_not_optimal)
Definition: conv_generated.h:171
void add_dtype(tt::target::DataType dtype)
Definition: conv_generated.h:150
void add_act_block_h_override(uint32_t act_block_h_override)
Definition: conv_generated.h:165
void add_deallocate_activation(bool deallocate_activation)
Definition: conv_generated.h:159
Conv2dConfigBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: conv_generated.h:207
void add_always_preprocess_weights(bool always_preprocess_weights)
Definition: conv_generated.h:192
Conv2dConfig Table
Definition: conv_generated.h:147
void add_act_block_w_div(uint32_t act_block_w_div)
Definition: conv_generated.h:168
void add_output_layout(tt::target::TensorLayout output_layout)
Definition: conv_generated.h:186
Definition: conv_generated.h:262
static constexpr auto Create
Definition: conv_generated.h:264
Conv2dConfig type
Definition: conv_generated.h:263
Definition: conv_generated.h:693
void add_in_channels(uint32_t in_channels)
Definition: conv_generated.h:712
void add_conv2d_config(::flatbuffers::Offset< tt::target::ttnn::Conv2dConfig > conv2d_config)
Definition: conv_generated.h:742
void add_input_width(uint32_t input_width)
Definition: conv_generated.h:724
void add_dilation(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> dilation)
Definition: conv_generated.h:736
Conv2dOp Table
Definition: conv_generated.h:694
void add_out_channels(uint32_t out_channels)
Definition: conv_generated.h:715
void add_padding(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> padding)
Definition: conv_generated.h:733
void add_stride(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> stride)
Definition: conv_generated.h:730
::flatbuffers::Offset< Conv2dOp > Finish()
Definition: conv_generated.h:749
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: conv_generated.h:709
Conv2dOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: conv_generated.h:745
void add_groups(uint32_t groups)
Definition: conv_generated.h:739
void add_out(::flatbuffers::Offset< tt::target::ttnn::TensorRef > out)
Definition: conv_generated.h:706
void add_weight(::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight)
Definition: conv_generated.h:700
void add_kernel_size(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> kernel_size)
Definition: conv_generated.h:727
void add_input(::flatbuffers::Offset< tt::target::ttnn::TensorRef > input)
Definition: conv_generated.h:697
::flatbuffers::uoffset_t start_
Definition: conv_generated.h:696
void add_batch_size(uint32_t batch_size)
Definition: conv_generated.h:718
void add_input_height(uint32_t input_height)
Definition: conv_generated.h:721
void add_bias(::flatbuffers::Offset< tt::target::ttnn::TensorRef > bias)
Definition: conv_generated.h:703
::flatbuffers::FlatBufferBuilder & fbb_
Definition: conv_generated.h:695
Definition: conv_generated.h:794
static constexpr auto Create
Definition: conv_generated.h:796
Conv2dOp type
Definition: conv_generated.h:795
Definition: conv_generated.h:951
void add_input(::flatbuffers::Offset< tt::target::ttnn::TensorRef > input)
Definition: conv_generated.h:955
void add_groups(uint32_t groups)
Definition: conv_generated.h:1000
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: conv_generated.h:967
void add_out(::flatbuffers::Offset< tt::target::ttnn::TensorRef > out)
Definition: conv_generated.h:964
ConvTranspose2dOp Table
Definition: conv_generated.h:952
void add_input_height(uint32_t input_height)
Definition: conv_generated.h:979
void add_weight(::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight)
Definition: conv_generated.h:958
::flatbuffers::Offset< ConvTranspose2dOp > Finish()
Definition: conv_generated.h:1010
ConvTranspose2dOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: conv_generated.h:1006
::flatbuffers::uoffset_t start_
Definition: conv_generated.h:954
void add_in_channels(uint32_t in_channels)
Definition: conv_generated.h:970
void add_memory_config(::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memory_config)
Definition: conv_generated.h:1003
void add_stride(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> stride)
Definition: conv_generated.h:988
void add_batch_size(uint32_t batch_size)
Definition: conv_generated.h:976
void add_out_channels(uint32_t out_channels)
Definition: conv_generated.h:973
void add_bias(::flatbuffers::Offset< tt::target::ttnn::TensorRef > bias)
Definition: conv_generated.h:961
void add_kernel_size(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> kernel_size)
Definition: conv_generated.h:985
::flatbuffers::FlatBufferBuilder & fbb_
Definition: conv_generated.h:953
void add_padding(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> padding)
Definition: conv_generated.h:991
void add_dilation(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> dilation)
Definition: conv_generated.h:997
void add_output_padding(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> output_padding)
Definition: conv_generated.h:994
void add_input_width(uint32_t input_width)
Definition: conv_generated.h:982
Definition: conv_generated.h:1057
ConvTranspose2dOp type
Definition: conv_generated.h:1058
static constexpr auto Create
Definition: conv_generated.h:1059
const tt::target::ttnn::TensorRef * bias() const
Definition: conv_generated.h:619
const tt::target::ttnn::TensorRef * weight_tensor() const
Definition: conv_generated.h:338
uint32_t in_channels() const
Definition: conv_generated.h:353
::flatbuffers::Optional< bool > preprocess_weights_on_device() const
Definition: conv_generated.h:101
uint32_t batch_size() const
Definition: conv_generated.h:359
const tt::target::ttnn::MemoryConfig * input_memory_config() const
Definition: conv_generated.h:344
ConvTranspose2dOpBuilder Builder
Definition: conv_generated.h:842
::flatbuffers::Optional< bool > enable_split_reader() const
Definition: conv_generated.h:113
uint32_t input_width() const
Definition: conv_generated.h:365
::flatbuffers::Optional< bool > enable_act_double_buffer() const
Definition: conv_generated.h:107
::flatbuffers::Optional< tt::target::ttnn::TensorMemoryLayout > shard_layout() const
Definition: conv_generated.h:89
const tt::target::ttnn::TensorRef * out() const
Definition: conv_generated.h:341
const tt::target::ttnn::MemoryConfig * memory_config() const
Definition: conv_generated.h:914
const tt::target::ttnn::Conv2dConfig * conv2d_config() const
Definition: conv_generated.h:389
::flatbuffers::Optional< tt::target::TensorLayout > output_layout() const
Definition: conv_generated.h:98
bool has_bias() const
Definition: conv_generated.h:380
const ::flatbuffers::Vector< int32_t > * dilation() const
Definition: conv_generated.h:377
const tt::target::ttnn::TensorRef * input() const
Definition: conv_generated.h:613
const ::flatbuffers::String * activation() const
Definition: conv_generated.h:68
uint32_t input_height() const
Definition: conv_generated.h:362
const ::flatbuffers::Vector< int32_t > * stride() const
Definition: conv_generated.h:371
::flatbuffers::Optional< tt::target::DataType > dtype() const
Definition: conv_generated.h:62
const ::flatbuffers::Vector< int32_t > * padding() const
Definition: conv_generated.h:374
const tt::target::ttnn::TensorRef * weight() const
Definition: conv_generated.h:616
Conv2dConfigBuilder Builder
Definition: conv_generated.h:36
const ::flatbuffers::Vector< int32_t > * output_padding() const
Definition: conv_generated.h:905
::flatbuffers::Optional< bool > override_sharding_config() const
Definition: conv_generated.h:86
Conv2dOpBuilder Builder
Definition: conv_generated.h:590
const ::flatbuffers::Vector< int32_t > * kernel_size() const
Definition: conv_generated.h:368
bool Verify(::flatbuffers::Verifier &verifier) const
Definition: conv_generated.h:119
::flatbuffers::Optional< bool > enable_subblock_padding() const
Definition: conv_generated.h:116
tt::target::TensorLayout input_tensor_layout() const
Definition: conv_generated.h:347
::flatbuffers::Optional< bool > transpose_shards() const
Definition: conv_generated.h:95
uint32_t groups() const
Definition: conv_generated.h:383
const ::flatbuffers::String * weights_format() const
Definition: conv_generated.h:350
::flatbuffers::Optional< bool > reshard_if_not_optimal() const
Definition: conv_generated.h:83
::flatbuffers::Optional< bool > enable_weights_double_buffer() const
Definition: conv_generated.h:110
uint32_t out_channels() const
Definition: conv_generated.h:356
const tt::target::ttnn::CoreRangeSet * core_grid() const
Definition: conv_generated.h:92
::flatbuffers::Optional< uint32_t > act_block_h_override() const
Definition: conv_generated.h:77
::flatbuffers::Optional< tt::target::DataType > weights_dtype() const
Definition: conv_generated.h:65
static FLATBUFFERS_CONSTEXPR_CPP11 const char * GetFullyQualifiedName()
Definition: conv_generated.h:38
::flatbuffers::Optional< bool > deallocate_activation() const
Definition: conv_generated.h:71
::flatbuffers::Optional< bool > reallocate_halo_output() const
Definition: conv_generated.h:74
const tt::target::DeviceRef * device() const
Definition: conv_generated.h:386
PrepareConv2dWeightsOpBuilder Builder
Definition: conv_generated.h:313
::flatbuffers::Optional< bool > always_preprocess_weights() const
Definition: conv_generated.h:104
::flatbuffers::Optional< uint32_t > act_block_w_div() const
Definition: conv_generated.h:80
Definition: conv_generated.h:426
void add_input_height(uint32_t input_height)
Definition: conv_generated.h:454
void add_dilation(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> dilation)
Definition: conv_generated.h:469
void add_input_tensor_layout(tt::target::TensorLayout input_tensor_layout)
Definition: conv_generated.h:439
PrepareConv2dWeightsOp Table
Definition: conv_generated.h:427
PrepareConv2dWeightsOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: conv_generated.h:484
::flatbuffers::FlatBufferBuilder & fbb_
Definition: conv_generated.h:428
void add_has_bias(bool has_bias)
Definition: conv_generated.h:472
void add_in_channels(uint32_t in_channels)
Definition: conv_generated.h:445
void add_padding(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> padding)
Definition: conv_generated.h:466
void add_out(::flatbuffers::Offset< tt::target::ttnn::TensorRef > out)
Definition: conv_generated.h:433
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: conv_generated.h:478
void add_weights_format(::flatbuffers::Offset<::flatbuffers::String > weights_format)
Definition: conv_generated.h:442
void add_conv2d_config(::flatbuffers::Offset< tt::target::ttnn::Conv2dConfig > conv2d_config)
Definition: conv_generated.h:481
void add_out_channels(uint32_t out_channels)
Definition: conv_generated.h:448
void add_input_memory_config(::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > input_memory_config)
Definition: conv_generated.h:436
void add_kernel_size(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> kernel_size)
Definition: conv_generated.h:460
void add_stride(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> stride)
Definition: conv_generated.h:463
void add_groups(uint32_t groups)
Definition: conv_generated.h:475
void add_batch_size(uint32_t batch_size)
Definition: conv_generated.h:451
void add_weight_tensor(::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight_tensor)
Definition: conv_generated.h:430
::flatbuffers::Offset< PrepareConv2dWeightsOp > Finish()
Definition: conv_generated.h:488
::flatbuffers::uoffset_t start_
Definition: conv_generated.h:429
void add_input_width(uint32_t input_width)
Definition: conv_generated.h:457
Definition: conv_generated.h:537
static constexpr auto Create
Definition: conv_generated.h:539
PrepareConv2dWeightsOp type
Definition: conv_generated.h:538