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");
25 struct Conv2dConfigBuilder;
28 struct PrepareConv2dWeightsOpBuilder;
31 struct Conv2dOpBuilder;
34 struct ConvTranspose2dOpBuilder;
36 struct Conv2dConfig FLATBUFFERS_FINAL_CLASS :
private ::flatbuffers::Table {
40 return "tt.target.ttnn.Conv2dConfig";
42 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
61 VT_ENABLE_SUBBLOCK_PADDING = 40
63 ::flatbuffers::Optional<tt::target::DataType>
dtype()
const {
64 return GetOptional<uint16_t, tt::target::DataType>(
VT_DTYPE);
70 return GetPointer<const ::flatbuffers::String *>(
VT_ACTIVATION);
90 ::flatbuffers::Optional<tt::target::ttnn::TensorMemoryLayout>
shard_layout()
const {
91 return GetOptional<uint16_t, tt::target::ttnn::TensorMemoryLayout>(
VT_SHARD_LAYOUT);
93 const tt::target::ttnn::CoreRangeSet *
core_grid()
const {
94 return GetPointer<const tt::target::ttnn::CoreRangeSet *>(
VT_CORE_GRID);
99 ::flatbuffers::Optional<tt::target::TensorLayout>
output_layout()
const {
118 return GetOptional<uint8_t, bool>(VT_ENABLE_SUBBLOCK_PADDING);
120 bool Verify(::flatbuffers::Verifier &verifier)
const {
121 return VerifyTableStart(verifier) &&
122 VerifyField<uint16_t>(verifier,
VT_DTYPE, 2) &&
125 verifier.VerifyString(activation()) &&
134 verifier.VerifyTable(core_grid()) &&
142 VerifyField<uint8_t>(verifier, VT_ENABLE_SUBBLOCK_PADDING, 1) &&
149 ::flatbuffers::FlatBufferBuilder &
fbb_;
181 void add_core_grid(::flatbuffers::Offset<tt::target::ttnn::CoreRangeSet> core_grid) {
206 fbb_.AddElement<uint8_t>(Conv2dConfig::VT_ENABLE_SUBBLOCK_PADDING,
static_cast<uint8_t
>(enable_subblock_padding));
212 ::flatbuffers::Offset<Conv2dConfig>
Finish() {
214 auto o = ::flatbuffers::Offset<Conv2dConfig>(end);
220 ::flatbuffers::FlatBufferBuilder &_fbb,
221 ::flatbuffers::Optional<tt::target::DataType> dtype = ::flatbuffers::nullopt,
222 ::flatbuffers::Optional<tt::target::DataType> weights_dtype = ::flatbuffers::nullopt,
223 ::flatbuffers::Offset<::flatbuffers::String> activation = 0,
224 ::flatbuffers::Optional<bool> deallocate_activation = ::flatbuffers::nullopt,
225 ::flatbuffers::Optional<bool> reallocate_halo_output = ::flatbuffers::nullopt,
226 ::flatbuffers::Optional<uint32_t> act_block_h_override = ::flatbuffers::nullopt,
227 ::flatbuffers::Optional<uint32_t> act_block_w_div = ::flatbuffers::nullopt,
228 ::flatbuffers::Optional<bool> reshard_if_not_optimal = ::flatbuffers::nullopt,
229 ::flatbuffers::Optional<bool> override_sharding_config = ::flatbuffers::nullopt,
230 ::flatbuffers::Optional<tt::target::ttnn::TensorMemoryLayout> shard_layout = ::flatbuffers::nullopt,
231 ::flatbuffers::Offset<tt::target::ttnn::CoreRangeSet> core_grid = 0,
232 ::flatbuffers::Optional<bool> transpose_shards = ::flatbuffers::nullopt,
233 ::flatbuffers::Optional<tt::target::TensorLayout> output_layout = ::flatbuffers::nullopt,
234 ::flatbuffers::Optional<bool> preprocess_weights_on_device = ::flatbuffers::nullopt,
235 ::flatbuffers::Optional<bool> always_preprocess_weights = ::flatbuffers::nullopt,
236 ::flatbuffers::Optional<bool> enable_act_double_buffer = ::flatbuffers::nullopt,
237 ::flatbuffers::Optional<bool> enable_weights_double_buffer = ::flatbuffers::nullopt,
238 ::flatbuffers::Optional<bool> enable_split_reader = ::flatbuffers::nullopt,
239 ::flatbuffers::Optional<bool> enable_subblock_padding = ::flatbuffers::nullopt) {
248 if(dtype) { builder_.
add_dtype(*dtype); }
269 ::flatbuffers::FlatBufferBuilder &_fbb,
270 ::flatbuffers::Optional<tt::target::DataType> dtype = ::flatbuffers::nullopt,
271 ::flatbuffers::Optional<tt::target::DataType> weights_dtype = ::flatbuffers::nullopt,
272 const char *activation =
nullptr,
273 ::flatbuffers::Optional<bool> deallocate_activation = ::flatbuffers::nullopt,
274 ::flatbuffers::Optional<bool> reallocate_halo_output = ::flatbuffers::nullopt,
275 ::flatbuffers::Optional<uint32_t> act_block_h_override = ::flatbuffers::nullopt,
276 ::flatbuffers::Optional<uint32_t> act_block_w_div = ::flatbuffers::nullopt,
277 ::flatbuffers::Optional<bool> reshard_if_not_optimal = ::flatbuffers::nullopt,
278 ::flatbuffers::Optional<bool> override_sharding_config = ::flatbuffers::nullopt,
279 ::flatbuffers::Optional<tt::target::ttnn::TensorMemoryLayout> shard_layout = ::flatbuffers::nullopt,
280 ::flatbuffers::Offset<tt::target::ttnn::CoreRangeSet> core_grid = 0,
281 ::flatbuffers::Optional<bool> transpose_shards = ::flatbuffers::nullopt,
282 ::flatbuffers::Optional<tt::target::TensorLayout> output_layout = ::flatbuffers::nullopt,
283 ::flatbuffers::Optional<bool> preprocess_weights_on_device = ::flatbuffers::nullopt,
284 ::flatbuffers::Optional<bool> always_preprocess_weights = ::flatbuffers::nullopt,
285 ::flatbuffers::Optional<bool> enable_act_double_buffer = ::flatbuffers::nullopt,
286 ::flatbuffers::Optional<bool> enable_weights_double_buffer = ::flatbuffers::nullopt,
287 ::flatbuffers::Optional<bool> enable_split_reader = ::flatbuffers::nullopt,
288 ::flatbuffers::Optional<bool> enable_subblock_padding = ::flatbuffers::nullopt) {
289 auto activation__ = activation ? _fbb.CreateString(activation) : 0;
295 deallocate_activation,
296 reallocate_halo_output,
297 act_block_h_override,
299 reshard_if_not_optimal,
300 override_sharding_config,
305 preprocess_weights_on_device,
306 always_preprocess_weights,
307 enable_act_double_buffer,
308 enable_weights_double_buffer,
310 enable_subblock_padding);
317 return "tt.target.ttnn.PrepareConv2dWeightsOp";
319 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
342 const tt::target::ttnn::TensorRef *
out()
const {
343 return GetPointer<const tt::target::ttnn::TensorRef *>(
VT_OUT);
370 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_KERNEL_SIZE);
372 const ::flatbuffers::Vector<int32_t> *
stride()
const {
373 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_STRIDE);
375 const ::flatbuffers::Vector<int32_t> *
padding()
const {
376 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_PADDING);
378 const ::flatbuffers::Vector<int32_t> *
dilation()
const {
379 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_DILATION);
387 const tt::target::DeviceRef *
device()
const {
388 return GetPointer<const tt::target::DeviceRef *>(
VT_DEVICE);
391 return GetPointer<const tt::target::ttnn::Conv2dConfig *>(
VT_CONV2D_CONFIG);
393 bool Verify(::flatbuffers::Verifier &verifier)
const {
394 return VerifyTableStart(verifier) &&
396 verifier.VerifyTable(weight_tensor()) &&
397 VerifyOffset(verifier,
VT_OUT) &&
398 verifier.VerifyTable(out()) &&
400 verifier.VerifyTable(input_memory_config()) &&
403 verifier.VerifyString(weights_format()) &&
410 verifier.VerifyVector(kernel_size()) &&
412 verifier.VerifyVector(stride()) &&
414 verifier.VerifyVector(padding()) &&
416 verifier.VerifyVector(dilation()) &&
418 VerifyField<uint32_t>(verifier,
VT_GROUPS, 4) &&
420 verifier.VerifyTable(device()) &&
422 verifier.VerifyTable(conv2d_config()) &&
429 ::flatbuffers::FlatBufferBuilder &
fbb_;
434 void add_out(::flatbuffers::Offset<tt::target::ttnn::TensorRef> out) {
461 void add_kernel_size(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> kernel_size) {
464 void add_stride(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> stride) {
467 void add_padding(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> padding) {
470 void add_dilation(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> dilation) {
479 void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
489 ::flatbuffers::Offset<PrepareConv2dWeightsOp>
Finish() {
491 auto o = ::flatbuffers::Offset<PrepareConv2dWeightsOp>(end);
497 ::flatbuffers::FlatBufferBuilder &_fbb,
498 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> weight_tensor = 0,
499 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
500 ::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> input_memory_config = 0,
502 ::flatbuffers::Offset<::flatbuffers::String> weights_format = 0,
503 uint32_t in_channels = 0,
504 uint32_t out_channels = 0,
505 uint32_t batch_size = 0,
506 uint32_t input_height = 0,
507 uint32_t input_width = 0,
508 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> kernel_size = 0,
509 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> stride = 0,
510 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> padding = 0,
511 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> dilation = 0,
512 bool has_bias =
false,
514 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
515 ::flatbuffers::Offset<tt::target::ttnn::Conv2dConfig> conv2d_config = 0) {
544 ::flatbuffers::FlatBufferBuilder &_fbb,
545 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> weight_tensor = 0,
546 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
547 ::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> input_memory_config = 0,
549 const char *weights_format =
nullptr,
550 uint32_t in_channels = 0,
551 uint32_t out_channels = 0,
552 uint32_t batch_size = 0,
553 uint32_t input_height = 0,
554 uint32_t input_width = 0,
555 const std::vector<int32_t> *kernel_size =
nullptr,
556 const std::vector<int32_t> *stride =
nullptr,
557 const std::vector<int32_t> *padding =
nullptr,
558 const std::vector<int32_t> *dilation =
nullptr,
559 bool has_bias =
false,
561 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
562 ::flatbuffers::Offset<tt::target::ttnn::Conv2dConfig> conv2d_config = 0) {
563 auto weights_format__ = weights_format ? _fbb.CreateString(weights_format) : 0;
564 auto kernel_size__ = kernel_size ? _fbb.CreateVector<int32_t>(*kernel_size) : 0;
565 auto stride__ = stride ? _fbb.CreateVector<int32_t>(*stride) : 0;
566 auto padding__ = padding ? _fbb.CreateVector<int32_t>(*padding) : 0;
567 auto dilation__ = dilation ? _fbb.CreateVector<int32_t>(*dilation) : 0;
594 return "tt.target.ttnn.Conv2dOp";
596 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
613 VT_COMPUTE_CONFIG = 36
615 const tt::target::ttnn::TensorRef *
input()
const {
616 return GetPointer<const tt::target::ttnn::TensorRef *>(
VT_INPUT);
618 const tt::target::ttnn::TensorRef *
weight()
const {
619 return GetPointer<const tt::target::ttnn::TensorRef *>(
VT_WEIGHT);
621 const tt::target::ttnn::TensorRef *
bias()
const {
622 return GetPointer<const tt::target::ttnn::TensorRef *>(
VT_BIAS);
624 const tt::target::ttnn::TensorRef *
out()
const {
625 return GetPointer<const tt::target::ttnn::TensorRef *>(
VT_OUT);
627 const tt::target::DeviceRef *
device()
const {
628 return GetPointer<const tt::target::DeviceRef *>(
VT_DEVICE);
646 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_KERNEL_SIZE);
648 const ::flatbuffers::Vector<int32_t> *
stride()
const {
649 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_STRIDE);
651 const ::flatbuffers::Vector<int32_t> *
padding()
const {
652 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_PADDING);
654 const ::flatbuffers::Vector<int32_t> *
dilation()
const {
655 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_DILATION);
661 return GetPointer<const tt::target::ttnn::Conv2dConfig *>(
VT_CONV2D_CONFIG);
664 return GetPointer<const tt::target::ttnn::DeviceComputeKernelConfig *>(VT_COMPUTE_CONFIG);
666 bool Verify(::flatbuffers::Verifier &verifier)
const {
667 return VerifyTableStart(verifier) &&
669 verifier.VerifyTable(input()) &&
671 verifier.VerifyTable(weight()) &&
672 VerifyOffset(verifier,
VT_BIAS) &&
673 verifier.VerifyTable(bias()) &&
674 VerifyOffset(verifier,
VT_OUT) &&
675 verifier.VerifyTable(out()) &&
677 verifier.VerifyTable(device()) &&
684 verifier.VerifyVector(kernel_size()) &&
686 verifier.VerifyVector(stride()) &&
688 verifier.VerifyVector(padding()) &&
690 verifier.VerifyVector(dilation()) &&
691 VerifyField<uint32_t>(verifier,
VT_GROUPS, 4) &&
693 verifier.VerifyTable(conv2d_config()) &&
694 VerifyOffset(verifier, VT_COMPUTE_CONFIG) &&
695 verifier.VerifyTable(compute_config()) &&
702 ::flatbuffers::FlatBufferBuilder &
fbb_;
704 void add_input(::flatbuffers::Offset<tt::target::ttnn::TensorRef> input) {
707 void add_weight(::flatbuffers::Offset<tt::target::ttnn::TensorRef> weight) {
710 void add_bias(::flatbuffers::Offset<tt::target::ttnn::TensorRef> bias) {
713 void add_out(::flatbuffers::Offset<tt::target::ttnn::TensorRef> out) {
716 void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
734 void add_kernel_size(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> kernel_size) {
737 void add_stride(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> stride) {
740 void add_padding(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> padding) {
743 void add_dilation(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> dilation) {
752 void add_compute_config(::flatbuffers::Offset<tt::target::ttnn::DeviceComputeKernelConfig> compute_config) {
753 fbb_.AddOffset(Conv2dOp::VT_COMPUTE_CONFIG, compute_config);
759 ::flatbuffers::Offset<Conv2dOp>
Finish() {
761 auto o = ::flatbuffers::Offset<Conv2dOp>(end);
767 ::flatbuffers::FlatBufferBuilder &_fbb,
768 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> input = 0,
769 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> weight = 0,
770 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> bias = 0,
771 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
772 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
773 uint32_t in_channels = 0,
774 uint32_t out_channels = 0,
775 uint32_t batch_size = 0,
776 uint32_t input_height = 0,
777 uint32_t input_width = 0,
778 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> kernel_size = 0,
779 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> stride = 0,
780 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> padding = 0,
781 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> dilation = 0,
783 ::flatbuffers::Offset<tt::target::ttnn::Conv2dConfig> conv2d_config = 0,
784 ::flatbuffers::Offset<tt::target::ttnn::DeviceComputeKernelConfig> compute_config = 0) {
812 ::flatbuffers::FlatBufferBuilder &_fbb,
813 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> input = 0,
814 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> weight = 0,
815 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> bias = 0,
816 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
817 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
818 uint32_t in_channels = 0,
819 uint32_t out_channels = 0,
820 uint32_t batch_size = 0,
821 uint32_t input_height = 0,
822 uint32_t input_width = 0,
823 const std::vector<int32_t> *kernel_size =
nullptr,
824 const std::vector<int32_t> *stride =
nullptr,
825 const std::vector<int32_t> *padding =
nullptr,
826 const std::vector<int32_t> *dilation =
nullptr,
828 ::flatbuffers::Offset<tt::target::ttnn::Conv2dConfig> conv2d_config = 0,
829 ::flatbuffers::Offset<tt::target::ttnn::DeviceComputeKernelConfig> compute_config = 0) {
830 auto kernel_size__ = kernel_size ? _fbb.CreateVector<int32_t>(*kernel_size) : 0;
831 auto stride__ = stride ? _fbb.CreateVector<int32_t>(*stride) : 0;
832 auto padding__ = padding ? _fbb.CreateVector<int32_t>(*padding) : 0;
833 auto dilation__ = dilation ? _fbb.CreateVector<int32_t>(*dilation) : 0;
859 return "tt.target.ttnn.ConvTranspose2dOp";
861 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
880 const tt::target::ttnn::TensorRef *
input()
const {
881 return GetPointer<const tt::target::ttnn::TensorRef *>(
VT_INPUT);
883 const tt::target::ttnn::TensorRef *
weight()
const {
884 return GetPointer<const tt::target::ttnn::TensorRef *>(
VT_WEIGHT);
886 const tt::target::ttnn::TensorRef *
bias()
const {
887 return GetPointer<const tt::target::ttnn::TensorRef *>(
VT_BIAS);
889 const tt::target::ttnn::TensorRef *
out()
const {
890 return GetPointer<const tt::target::ttnn::TensorRef *>(
VT_OUT);
892 const tt::target::DeviceRef *
device()
const {
893 return GetPointer<const tt::target::DeviceRef *>(
VT_DEVICE);
911 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_KERNEL_SIZE);
913 const ::flatbuffers::Vector<int32_t> *
stride()
const {
914 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_STRIDE);
916 const ::flatbuffers::Vector<int32_t> *
padding()
const {
917 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_PADDING);
922 const ::flatbuffers::Vector<int32_t> *
dilation()
const {
923 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_DILATION);
929 return GetPointer<const tt::target::ttnn::MemoryConfig *>(
VT_MEMORY_CONFIG);
931 bool Verify(::flatbuffers::Verifier &verifier)
const {
932 return VerifyTableStart(verifier) &&
934 verifier.VerifyTable(input()) &&
936 verifier.VerifyTable(weight()) &&
937 VerifyOffset(verifier,
VT_BIAS) &&
938 verifier.VerifyTable(bias()) &&
939 VerifyOffset(verifier,
VT_OUT) &&
940 verifier.VerifyTable(out()) &&
942 verifier.VerifyTable(device()) &&
949 verifier.VerifyVector(kernel_size()) &&
951 verifier.VerifyVector(stride()) &&
953 verifier.VerifyVector(padding()) &&
955 verifier.VerifyVector(output_padding()) &&
957 verifier.VerifyVector(dilation()) &&
958 VerifyField<uint32_t>(verifier,
VT_GROUPS, 4) &&
960 verifier.VerifyTable(memory_config()) &&
967 ::flatbuffers::FlatBufferBuilder &
fbb_;
969 void add_input(::flatbuffers::Offset<tt::target::ttnn::TensorRef> input) {
972 void add_weight(::flatbuffers::Offset<tt::target::ttnn::TensorRef> weight) {
975 void add_bias(::flatbuffers::Offset<tt::target::ttnn::TensorRef> bias) {
978 void add_out(::flatbuffers::Offset<tt::target::ttnn::TensorRef> out) {
981 void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
999 void add_kernel_size(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> kernel_size) {
1002 void add_stride(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> stride) {
1005 void add_padding(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> padding) {
1011 void add_dilation(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> dilation) {
1024 ::flatbuffers::Offset<ConvTranspose2dOp>
Finish() {
1026 auto o = ::flatbuffers::Offset<ConvTranspose2dOp>(end);
1032 ::flatbuffers::FlatBufferBuilder &_fbb,
1033 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> input = 0,
1034 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> weight = 0,
1035 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> bias = 0,
1036 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
1037 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
1038 uint32_t in_channels = 0,
1039 uint32_t out_channels = 0,
1040 uint32_t batch_size = 0,
1041 uint32_t input_height = 0,
1042 uint32_t input_width = 0,
1043 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> kernel_size = 0,
1044 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> stride = 0,
1045 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> padding = 0,
1046 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> output_padding = 0,
1047 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> dilation = 0,
1048 uint32_t groups = 0,
1049 ::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> memory_config = 0) {
1068 return builder_.
Finish();
1077 ::flatbuffers::FlatBufferBuilder &_fbb,
1078 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> input = 0,
1079 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> weight = 0,
1080 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> bias = 0,
1081 ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
1082 ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
1083 uint32_t in_channels = 0,
1084 uint32_t out_channels = 0,
1085 uint32_t batch_size = 0,
1086 uint32_t input_height = 0,
1087 uint32_t input_width = 0,
1088 const std::vector<int32_t> *kernel_size =
nullptr,
1089 const std::vector<int32_t> *stride =
nullptr,
1090 const std::vector<int32_t> *padding =
nullptr,
1091 const std::vector<int32_t> *output_padding =
nullptr,
1092 const std::vector<int32_t> *dilation =
nullptr,
1093 uint32_t groups = 0,
1094 ::flatbuffers::Offset<tt::target::ttnn::MemoryConfig> memory_config = 0) {
1095 auto kernel_size__ = kernel_size ? _fbb.CreateVector<int32_t>(*kernel_size) : 0;
1096 auto stride__ = stride ? _fbb.CreateVector<int32_t>(*stride) : 0;
1097 auto padding__ = padding ? _fbb.CreateVector<int32_t>(*padding) : 0;
1098 auto output_padding__ = output_padding ? _fbb.CreateVector<int32_t>(*output_padding) : 0;
1099 auto dilation__ = dilation ? _fbb.CreateVector<int32_t>(*dilation) : 0;
VT_PREPROCESS_WEIGHTS_ON_DEVICE
Definition: conv_generated.h:56
VT_INPUT
Definition: conv_generated.h:597
VT_DTYPE
Definition: conv_generated.h:43
VT_REALLOCATE_HALO_OUTPUT
Definition: conv_generated.h:47
VT_DILATION
Definition: conv_generated.h:333
VT_WEIGHT_TENSOR
Definition: conv_generated.h:320
VT_ACT_BLOCK_W_DIV
Definition: conv_generated.h:49
VT_PADDING
Definition: conv_generated.h:332
VT_WEIGHT
Definition: conv_generated.h:598
VT_ENABLE_WEIGHTS_DOUBLE_BUFFER
Definition: conv_generated.h:59
VT_OVERRIDE_SHARDING_CONFIG
Definition: conv_generated.h:51
VT_STRIDE
Definition: conv_generated.h:331
VT_KERNEL_SIZE
Definition: conv_generated.h:330
VT_BATCH_SIZE
Definition: conv_generated.h:327
VT_CORE_GRID
Definition: conv_generated.h:53
VT_GROUPS
Definition: conv_generated.h:335
VT_WEIGHTS_FORMAT
Definition: conv_generated.h:324
VT_ENABLE_SPLIT_READER
Definition: conv_generated.h:60
VT_DEALLOCATE_ACTIVATION
Definition: conv_generated.h:46
VT_TRANSPOSE_SHARDS
Definition: conv_generated.h:54
VT_SHARD_LAYOUT
Definition: conv_generated.h:52
VT_CONV2D_CONFIG
Definition: conv_generated.h:612
VT_OUTPUT_LAYOUT
Definition: conv_generated.h:55
VT_WEIGHTS_DTYPE
Definition: conv_generated.h:44
VT_INPUT_HEIGHT
Definition: conv_generated.h:328
VT_INPUT_MEMORY_CONFIG
Definition: conv_generated.h:322
VT_ENABLE_ACT_DOUBLE_BUFFER
Definition: conv_generated.h:58
VT_IN_CHANNELS
Definition: conv_generated.h:325
VT_DEVICE
Definition: conv_generated.h:336
VT_INPUT_WIDTH
Definition: conv_generated.h:329
VT_ACT_BLOCK_H_OVERRIDE
Definition: conv_generated.h:48
VT_BIAS
Definition: conv_generated.h:599
VT_OUT
Definition: conv_generated.h:321
VT_RESHARD_IF_NOT_OPTIMAL
Definition: conv_generated.h:50
VT_OUTPUT_PADDING
Definition: conv_generated.h:875
VT_ACTIVATION
Definition: conv_generated.h:45
VT_OUT_CHANNELS
Definition: conv_generated.h:326
VT_ALWAYS_PREPROCESS_WEIGHTS
Definition: conv_generated.h:57
VT_INPUT_TENSOR_LAYOUT
Definition: conv_generated.h:323
VT_HAS_BIAS
Definition: conv_generated.h:334
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:219
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, ::flatbuffers::Offset< tt::target::ttnn::DeviceComputeKernelConfig > compute_config=0)
Definition: conv_generated.h:766
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:1031
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, ::flatbuffers::Offset< tt::target::ttnn::DeviceComputeKernelConfig > compute_config=0)
Definition: conv_generated.h:811
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:496
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:1076
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:543
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:268
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:147
void add_enable_subblock_padding(bool enable_subblock_padding)
Definition: conv_generated.h:205
::flatbuffers::FlatBufferBuilder & fbb_
Definition: conv_generated.h:149
void add_activation(::flatbuffers::Offset<::flatbuffers::String > activation)
Definition: conv_generated.h:157
void add_enable_split_reader(bool enable_split_reader)
Definition: conv_generated.h:202
void add_reallocate_halo_output(bool reallocate_halo_output)
Definition: conv_generated.h:163
::flatbuffers::uoffset_t start_
Definition: conv_generated.h:150
void add_weights_dtype(tt::target::DataType weights_dtype)
Definition: conv_generated.h:154
void add_enable_act_double_buffer(bool enable_act_double_buffer)
Definition: conv_generated.h:196
void add_shard_layout(tt::target::ttnn::TensorMemoryLayout shard_layout)
Definition: conv_generated.h:178
void add_core_grid(::flatbuffers::Offset< tt::target::ttnn::CoreRangeSet > core_grid)
Definition: conv_generated.h:181
void add_transpose_shards(bool transpose_shards)
Definition: conv_generated.h:184
void add_override_sharding_config(bool override_sharding_config)
Definition: conv_generated.h:175
void add_preprocess_weights_on_device(bool preprocess_weights_on_device)
Definition: conv_generated.h:190
void add_enable_weights_double_buffer(bool enable_weights_double_buffer)
Definition: conv_generated.h:199
::flatbuffers::Offset< Conv2dConfig > Finish()
Definition: conv_generated.h:212
void add_reshard_if_not_optimal(bool reshard_if_not_optimal)
Definition: conv_generated.h:172
void add_dtype(tt::target::DataType dtype)
Definition: conv_generated.h:151
void add_act_block_h_override(uint32_t act_block_h_override)
Definition: conv_generated.h:166
void add_deallocate_activation(bool deallocate_activation)
Definition: conv_generated.h:160
Conv2dConfigBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: conv_generated.h:208
void add_always_preprocess_weights(bool always_preprocess_weights)
Definition: conv_generated.h:193
Conv2dConfig Table
Definition: conv_generated.h:148
void add_act_block_w_div(uint32_t act_block_w_div)
Definition: conv_generated.h:169
void add_output_layout(tt::target::TensorLayout output_layout)
Definition: conv_generated.h:187
Definition: conv_generated.h:263
static constexpr auto Create
Definition: conv_generated.h:265
Conv2dConfig type
Definition: conv_generated.h:264
Definition: conv_generated.h:700
void add_in_channels(uint32_t in_channels)
Definition: conv_generated.h:719
void add_conv2d_config(::flatbuffers::Offset< tt::target::ttnn::Conv2dConfig > conv2d_config)
Definition: conv_generated.h:749
void add_input_width(uint32_t input_width)
Definition: conv_generated.h:731
void add_dilation(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> dilation)
Definition: conv_generated.h:743
Conv2dOp Table
Definition: conv_generated.h:701
void add_out_channels(uint32_t out_channels)
Definition: conv_generated.h:722
void add_padding(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> padding)
Definition: conv_generated.h:740
void add_stride(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> stride)
Definition: conv_generated.h:737
::flatbuffers::Offset< Conv2dOp > Finish()
Definition: conv_generated.h:759
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: conv_generated.h:716
Conv2dOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: conv_generated.h:755
void add_compute_config(::flatbuffers::Offset< tt::target::ttnn::DeviceComputeKernelConfig > compute_config)
Definition: conv_generated.h:752
void add_groups(uint32_t groups)
Definition: conv_generated.h:746
void add_out(::flatbuffers::Offset< tt::target::ttnn::TensorRef > out)
Definition: conv_generated.h:713
void add_weight(::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight)
Definition: conv_generated.h:707
void add_kernel_size(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> kernel_size)
Definition: conv_generated.h:734
void add_input(::flatbuffers::Offset< tt::target::ttnn::TensorRef > input)
Definition: conv_generated.h:704
::flatbuffers::uoffset_t start_
Definition: conv_generated.h:703
void add_batch_size(uint32_t batch_size)
Definition: conv_generated.h:725
void add_input_height(uint32_t input_height)
Definition: conv_generated.h:728
void add_bias(::flatbuffers::Offset< tt::target::ttnn::TensorRef > bias)
Definition: conv_generated.h:710
::flatbuffers::FlatBufferBuilder & fbb_
Definition: conv_generated.h:702
Definition: conv_generated.h:806
static constexpr auto Create
Definition: conv_generated.h:808
Conv2dOp type
Definition: conv_generated.h:807
Definition: conv_generated.h:965
void add_input(::flatbuffers::Offset< tt::target::ttnn::TensorRef > input)
Definition: conv_generated.h:969
void add_groups(uint32_t groups)
Definition: conv_generated.h:1014
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: conv_generated.h:981
void add_out(::flatbuffers::Offset< tt::target::ttnn::TensorRef > out)
Definition: conv_generated.h:978
ConvTranspose2dOp Table
Definition: conv_generated.h:966
void add_input_height(uint32_t input_height)
Definition: conv_generated.h:993
void add_weight(::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight)
Definition: conv_generated.h:972
::flatbuffers::Offset< ConvTranspose2dOp > Finish()
Definition: conv_generated.h:1024
ConvTranspose2dOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: conv_generated.h:1020
::flatbuffers::uoffset_t start_
Definition: conv_generated.h:968
void add_in_channels(uint32_t in_channels)
Definition: conv_generated.h:984
void add_memory_config(::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memory_config)
Definition: conv_generated.h:1017
void add_stride(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> stride)
Definition: conv_generated.h:1002
void add_batch_size(uint32_t batch_size)
Definition: conv_generated.h:990
void add_out_channels(uint32_t out_channels)
Definition: conv_generated.h:987
void add_bias(::flatbuffers::Offset< tt::target::ttnn::TensorRef > bias)
Definition: conv_generated.h:975
void add_kernel_size(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> kernel_size)
Definition: conv_generated.h:999
::flatbuffers::FlatBufferBuilder & fbb_
Definition: conv_generated.h:967
void add_padding(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> padding)
Definition: conv_generated.h:1005
void add_dilation(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> dilation)
Definition: conv_generated.h:1011
void add_output_padding(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> output_padding)
Definition: conv_generated.h:1008
void add_input_width(uint32_t input_width)
Definition: conv_generated.h:996
Definition: conv_generated.h:1071
ConvTranspose2dOp type
Definition: conv_generated.h:1072
static constexpr auto Create
Definition: conv_generated.h:1073
const tt::target::ttnn::TensorRef * bias() const
Definition: conv_generated.h:621
const tt::target::ttnn::TensorRef * weight_tensor() const
Definition: conv_generated.h:339
uint32_t in_channels() const
Definition: conv_generated.h:354
::flatbuffers::Optional< bool > preprocess_weights_on_device() const
Definition: conv_generated.h:102
uint32_t batch_size() const
Definition: conv_generated.h:360
const tt::target::ttnn::MemoryConfig * input_memory_config() const
Definition: conv_generated.h:345
ConvTranspose2dOpBuilder Builder
Definition: conv_generated.h:856
::flatbuffers::Optional< bool > enable_split_reader() const
Definition: conv_generated.h:114
uint32_t input_width() const
Definition: conv_generated.h:366
::flatbuffers::Optional< bool > enable_act_double_buffer() const
Definition: conv_generated.h:108
::flatbuffers::Optional< tt::target::ttnn::TensorMemoryLayout > shard_layout() const
Definition: conv_generated.h:90
const tt::target::ttnn::TensorRef * out() const
Definition: conv_generated.h:342
const tt::target::ttnn::MemoryConfig * memory_config() const
Definition: conv_generated.h:928
const tt::target::ttnn::Conv2dConfig * conv2d_config() const
Definition: conv_generated.h:390
::flatbuffers::Optional< tt::target::TensorLayout > output_layout() const
Definition: conv_generated.h:99
bool has_bias() const
Definition: conv_generated.h:381
const ::flatbuffers::Vector< int32_t > * dilation() const
Definition: conv_generated.h:378
const tt::target::ttnn::TensorRef * input() const
Definition: conv_generated.h:615
const ::flatbuffers::String * activation() const
Definition: conv_generated.h:69
uint32_t input_height() const
Definition: conv_generated.h:363
const ::flatbuffers::Vector< int32_t > * stride() const
Definition: conv_generated.h:372
::flatbuffers::Optional< tt::target::DataType > dtype() const
Definition: conv_generated.h:63
const ::flatbuffers::Vector< int32_t > * padding() const
Definition: conv_generated.h:375
const tt::target::ttnn::TensorRef * weight() const
Definition: conv_generated.h:618
Conv2dConfigBuilder Builder
Definition: conv_generated.h:37
const ::flatbuffers::Vector< int32_t > * output_padding() const
Definition: conv_generated.h:919
::flatbuffers::Optional< bool > override_sharding_config() const
Definition: conv_generated.h:87
Conv2dOpBuilder Builder
Definition: conv_generated.h:591
const ::flatbuffers::Vector< int32_t > * kernel_size() const
Definition: conv_generated.h:369
bool Verify(::flatbuffers::Verifier &verifier) const
Definition: conv_generated.h:120
::flatbuffers::Optional< bool > enable_subblock_padding() const
Definition: conv_generated.h:117
tt::target::TensorLayout input_tensor_layout() const
Definition: conv_generated.h:348
::flatbuffers::Optional< bool > transpose_shards() const
Definition: conv_generated.h:96
uint32_t groups() const
Definition: conv_generated.h:384
const tt::target::ttnn::DeviceComputeKernelConfig * compute_config() const
Definition: conv_generated.h:663
const ::flatbuffers::String * weights_format() const
Definition: conv_generated.h:351
::flatbuffers::Optional< bool > reshard_if_not_optimal() const
Definition: conv_generated.h:84
::flatbuffers::Optional< bool > enable_weights_double_buffer() const
Definition: conv_generated.h:111
uint32_t out_channels() const
Definition: conv_generated.h:357
const tt::target::ttnn::CoreRangeSet * core_grid() const
Definition: conv_generated.h:93
::flatbuffers::Optional< uint32_t > act_block_h_override() const
Definition: conv_generated.h:78
::flatbuffers::Optional< tt::target::DataType > weights_dtype() const
Definition: conv_generated.h:66
static FLATBUFFERS_CONSTEXPR_CPP11 const char * GetFullyQualifiedName()
Definition: conv_generated.h:39
::flatbuffers::Optional< bool > deallocate_activation() const
Definition: conv_generated.h:72
::flatbuffers::Optional< bool > reallocate_halo_output() const
Definition: conv_generated.h:75
const tt::target::DeviceRef * device() const
Definition: conv_generated.h:387
PrepareConv2dWeightsOpBuilder Builder
Definition: conv_generated.h:314
::flatbuffers::Optional< bool > always_preprocess_weights() const
Definition: conv_generated.h:105
::flatbuffers::Optional< uint32_t > act_block_w_div() const
Definition: conv_generated.h:81
Definition: conv_generated.h:427
void add_input_height(uint32_t input_height)
Definition: conv_generated.h:455
void add_dilation(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> dilation)
Definition: conv_generated.h:470
void add_input_tensor_layout(tt::target::TensorLayout input_tensor_layout)
Definition: conv_generated.h:440
PrepareConv2dWeightsOp Table
Definition: conv_generated.h:428
PrepareConv2dWeightsOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
Definition: conv_generated.h:485
::flatbuffers::FlatBufferBuilder & fbb_
Definition: conv_generated.h:429
void add_has_bias(bool has_bias)
Definition: conv_generated.h:473
void add_in_channels(uint32_t in_channels)
Definition: conv_generated.h:446
void add_padding(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> padding)
Definition: conv_generated.h:467
void add_out(::flatbuffers::Offset< tt::target::ttnn::TensorRef > out)
Definition: conv_generated.h:434
void add_device(::flatbuffers::Offset< tt::target::DeviceRef > device)
Definition: conv_generated.h:479
void add_weights_format(::flatbuffers::Offset<::flatbuffers::String > weights_format)
Definition: conv_generated.h:443
void add_conv2d_config(::flatbuffers::Offset< tt::target::ttnn::Conv2dConfig > conv2d_config)
Definition: conv_generated.h:482
void add_out_channels(uint32_t out_channels)
Definition: conv_generated.h:449
void add_input_memory_config(::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > input_memory_config)
Definition: conv_generated.h:437
void add_kernel_size(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> kernel_size)
Definition: conv_generated.h:461
void add_stride(::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> stride)
Definition: conv_generated.h:464
void add_groups(uint32_t groups)
Definition: conv_generated.h:476
void add_batch_size(uint32_t batch_size)
Definition: conv_generated.h:452
void add_weight_tensor(::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight_tensor)
Definition: conv_generated.h:431
::flatbuffers::Offset< PrepareConv2dWeightsOp > Finish()
Definition: conv_generated.h:489
::flatbuffers::uoffset_t start_
Definition: conv_generated.h:430
void add_input_width(uint32_t input_width)
Definition: conv_generated.h:458
Definition: conv_generated.h:538
static constexpr auto Create
Definition: conv_generated.h:540
PrepareConv2dWeightsOp type
Definition: conv_generated.h:539