24 #ifndef TVM_RELAX_ATTRS_NN_H_
25 #define TVM_RELAX_ATTRS_NN_H_
45 refl::ObjectDef<Conv1DAttrs>()
48 "If padding is non-zero, then the input is implicitly zero-padded"
49 "Padding support both symmetric and asymmetric as"
50 "one int : same padding used on both sides"
51 "two int : padding width in the order of (left, right)")
53 "Specifies the dilation rate to use for dilated convolution.")
55 "Number of groups to split the input into for grouped convolution. The number of "
57 "output channels should be divisible by the number of groups.")
59 "Dimension ordering of input data. Can be 'NCW', 'NWC', etc."
60 "'N', 'C', 'W' stands for batch, channel, width"
61 "dimensions respectively. Convolution is applied on the 'W' dimensions.")
63 "Dimension ordering of weight. Can be 'OIW', 'IOW', etc."
64 "'O', 'I', 'W' stands for num_filter, input_channel, and width"
65 "dimensions respectively.")
67 "Dimension ordering of output. Can be 'NCW', 'NWC', etc."
68 "'N', 'C', 'W' stands for batch, channel, and width"
69 "dimensions respectively. Default to be same as input layout.")
71 "Output data type, set to explicit type under mixed precision setting");
89 refl::ObjectDef<Conv2DAttrs>()
92 "If padding is non-zero, then the input is implicitly zero-padded"
93 "Padding support both symmetric and asymmetric as"
94 "one int : same padding used on all sides"
95 "two int : bottom, right will use same padding as top, left"
96 "four int : padding width in the order of (top, left, bottom, right)")
98 "Specifies the dilation rate to use for dilated convolution.")
100 "Number of groups to split the input into for grouped convolution. The number of "
102 "output channels should be divisible by the number of groups.")
104 "Dimension ordering of input data. Can be 'NCHW', 'NHWC', etc."
105 "'N', 'C', 'H', 'W' stands for batch, channel, height, and width"
106 "dimensions respectively. Convolution is applied on the 'H' and"
109 "Dimension ordering of weight. Can be 'OIHW', 'OIHW16o16i', etc."
110 "'O', 'I', 'H', 'W' stands for num_filter, input_channel, height, and width"
111 "dimensions respectively.")
113 "Dimension ordering of output. Can be 'NCHW', 'NHWC', etc."
114 "'N', 'C', 'H', 'W' stands for batch, channel, height, and width"
115 "dimensions respectively. Default to be same as input layout.")
117 "Output data type, set to explicit type under mixed precision setting");
135 refl::ObjectDef<Conv3DAttrs>()
139 "If padding is non-zero, then the input is implicitly zero-padded"
140 "Padding support both symmetric and asymmetric as"
141 "one int : same padding used on all sides"
142 "two int : bottom, right will use same padding as top, left"
143 "four int : padding width in the order of (forward, back, top, left, bottom, right)")
145 "Specifies the dilation rate to use for dilated convolution.")
147 "Number of groups to split the input into for grouped convolution. The number of "
149 "output channels should be divisible by the number of groups.")
151 "Dimension ordering of input data. Can be 'NCDHW', 'NDHWC', etc."
152 "'N', 'C', 'D', 'H', 'W' stands for batch, channel, depth, height, and width"
153 "dimensions respectively. Convolution is applied on the 'D', 'H', and"
157 "Dimension ordering of weight. Can be 'OIDHW', 'OIDHW16o16i', etc."
158 "'O', 'I', 'D', 'H', 'W' stands for num_filter, input_channel, depth, height, and width"
159 "dimensions respectively.")
161 "Dimension ordering of output. Can be 'NCDHW', 'NDHWC', etc."
162 "'N', 'C', 'D', 'H', 'W' stands for batch, channel, depth, height, and width"
163 "dimensions respectively. Default to be same as input layout.")
165 "Output data type, set to explicit type under mixed precision setting");
184 refl::ObjectDef<Conv1DTransposeAttrs>()
186 "Specifies the strides of the convolution.")
188 "If padding is non-zero, then the input is implicitly zero-padded"
189 "Padding support both symmetric and asymmetric as"
190 "one int : same padding used on both sides"
191 "two int : padding width in the order of (left, right)")
193 "Used to disambiguate the output shape.")
195 "Specifies the dilation rate to use for dilated convolution.")
197 "Number of groups to split the input into for grouped convolution. The number of "
199 "output channels should be divisible by the number of groups.")
201 "Dimension ordering of input data. Can be 'NCW', 'NWC', etc."
202 "'N', 'C', 'W' stands for batch, channel, width"
203 "dimensions respectively. Convolution is applied on the 'W' dimensions.")
205 "Dimension ordering of weight. Can be 'OIW', 'IOW', etc."
206 "'O', 'I', 'W' stands for num_filter, input_channel, and width"
207 "dimensions respectively.")
209 "Dimension ordering of output. Can be 'NCW', 'NWC', etc."
210 "'N', 'C', 'W' stands for batch, channel, and width"
211 "dimensions respectively. Default to be same as input layout.")
213 "Output data type, set to explicit type under mixed precision setting");
233 refl::ObjectDef<Conv2DTransposeAttrs>()
235 "Specifies the strides of the convolution.")
237 "If padding is non-zero, then the input is implicitly zero-padded"
238 "Padding support both symmetric and asymmetric as"
239 "one int : same padding used on all sides"
240 "two int : bottom, right will use same padding as top, left"
241 "four int : padding width in the order of (top, left, bottom, right)")
243 "Used to disambiguate the output shape.")
245 "Specifies the dilation rate to use for dilated convolution.")
247 "Number of groups to split the input into for grouped convolution. The number of "
249 "output channels should be divisible by the number of groups.")
251 "Dimension ordering of input data. Can be 'NCHW', 'NHWC', etc."
252 "'N', 'C', 'H', 'W' stands for batch, channel, height, and width"
253 "dimensions respectively. Convolution is applied on the 'H' and"
256 "Dimension ordering of weight. Can be 'OIHW', 'OIHW16o16i', etc."
257 "'O', 'I', 'H', 'W' stands for num_filter, input_channel, height, and width"
258 "dimensions respectively.")
260 "Dimension ordering of output. Can be 'NCHW', 'NHWC', etc."
261 "'N', 'C', 'H', 'W' stands for batch, channel, height, and width"
262 "dimensions respectively. Default to be same as input layout.")
264 "Output data type, set to explicit type under mixed precision setting");
283 refl::ObjectDef<Pool1DAttrs>()
288 "If padding is non-zero, then the input is implicitly zero-padded"
289 "Padding support both symmetric and asymmetric as"
290 "one int : same padding used on all sides"
291 "two int : padding width in the order of (left, right)")
294 "A boolean indicating if use ceil or floor to compute the output shape. By using ceil, "
295 "every element in the input tensor will be covered by a sliding window.")
297 "When true, will include padding to compute the average")
299 "Dimension ordering of input data. Can be 'NCW', 'NWC', etc."
300 "'N', 'C', 'W' stands for batch, channel, and width"
301 "dimensions respectively. Pooling is applied on the 'W' dimensions.",
302 refl::DefaultValue(
"NCW"))
304 "Dimension ordering of output data. Can be 'NCW', 'NWC', etc."
305 "'N', 'C', 'W' stands for batch, channel, and width"
306 "dimensions respectively. Pooling is applied on the 'W' dimensions.");
324 refl::ObjectDef<Pool2DAttrs>()
329 "If padding is non-zero, then the input is implicitly zero-padded"
330 "Padding support both symmetric and asymmetric as"
331 "one int : same padding used on all sides"
332 "two int : bottom, right will use same padding as top, left"
333 "four int : padding width in the order of (top, left, bottom, right)")
336 "A boolean indicating if use ceil or floor to compute the output shape. By using ceil, "
337 "every element in the input tensor will be covered by a sliding window.")
339 "When true, will include padding to compute the average")
341 "Dimension ordering of input data. Can be 'NCHW', 'NHWC', etc."
342 "'N', 'C', 'H', 'W' stands for batch, channel, height, and width"
343 "dimensions respectively. Pooling is applied on the 'H' and"
346 "Dimension ordering of output data. Can be 'NCHW', 'NHWC', etc."
347 "'N', 'C', 'H', 'W' stands for batch, channel, height, and width"
348 "dimensions respectively. Pooling is applied on the 'H' and"
367 refl::ObjectDef<Pool3DAttrs>()
372 "If padding is non-zero, then the input is implicitly zero-padded"
373 "Padding support both symmetric and asymmetric as"
374 "one int : same padding used on all sides"
375 "three int : back, bottom, right will use same padding as front, top, left"
376 "four int : padding width in the order of (front, top, left, back, bottom, right)")
379 "A boolean indicating if use ceil or floor to compute the output shape. By using ceil, "
380 "every element in the input tensor will be covered by a sliding window.")
382 "When true, will include padding to compute the average")
384 "Dimension ordering of input data. Can be 'NCDHW', 'NDHWC', etc."
385 "'N', 'C', 'D', 'H', 'W' stands for batch, channel, depth, height, and width"
386 "dimensions respectively. Pooling is applied on the 'D', 'H' and"
389 "Dimension ordering of output data. Can be 'NCDHW', 'NDHWC', etc."
390 "'N', 'C', 'D', 'H', 'W' stands for batch, channel, depth, height, and width"
391 "dimensions respectively. Pooling is applied on the 'D', 'H' and"
405 refl::ObjectDef<AdaptivePool1DAttrs>()
408 "Dimension ordering of input data. Can be 'NCW', 'NWC', etc."
409 "'N', 'C', 'W' stands for batch, channel and width"
410 "dimensions respectively. Pooling is applied on the"
413 "Dimension ordering of output data. Can be 'NCW', 'NWC', etc."
414 "'N', 'C', 'W' stands for batch, channel and width"
415 "dimensions respectively. Pooling is applied on the"
430 refl::ObjectDef<AdaptivePool2DAttrs>()
433 "Dimension ordering of input data. Can be 'NCHW', 'NHWC', etc."
434 "'N', 'C', 'H', 'W' stands for batch, channel, height, and width"
435 "dimensions respectively. Pooling is applied on the 'H' and"
438 "Dimension ordering of output data. Can be 'NCHW', 'NHWC', etc."
439 "'N', 'C', 'H', 'W' stands for batch, channel, height, and width"
440 "dimensions respectively. Pooling is applied on the 'H' and"
455 refl::ObjectDef<AdaptivePool3DAttrs>()
458 "Dimension ordering of input data. Can be 'NCDHW', 'NDHWC', etc."
459 "'N', 'C', 'D', 'H', 'W' stands for batch, channel, depth, height, and width"
460 "dimensions respectively. Pooling is applied on 'D', 'H' and"
463 "Dimension ordering of output data. Can be 'NCDHW', 'NDHWC', etc."
464 "'N', 'C', 'D', 'H', 'W' stands for batch, channel, depth, height, and width"
465 "dimensions respectively. Pooling is applied on 'D', 'H' and"
479 "The axis to sum over when computing softmax.");
491 "The slope of the negative part.");
503 refl::ObjectDef<SoftplusAttrs>()
505 "Scaling factor controlling the sharpness of the Softplus transition.")
507 "Value determining when to use linear approximation for numerical stability.");
519 "The axis along which the alpha values are applied.");
535 refl::ObjectDef<BatchNormAttrs>()
538 "Small float added to variance to avoid dividing by zero")
540 "Indicating if the beta offset will be added to the normalized tensor.")
542 "Indicating if the gamma scale will be multiplied.")
544 "The value used for the moving_mean and moving_var update.")
546 "Whether we are training (i.e., not in eval mode).");
560 refl::ObjectDef<LayerNormAttrs>()
562 "The axes that along which the normalization is applied.")
564 "Small float added to variance to avoid dividing by zero")
566 "Indicating if the beta offset will be added to the normalized tensor.")
568 "Indicating if the gamma scale will be multiplied.");
584 refl::ObjectDef<GroupNormAttrs>()
586 "The number of groups to separate the channels into.")
588 "The axis that represents the channel.")
591 "The axes that along which the normalization is applied (excluding the channel axis).")
593 "Small float added to variance to avoid dividing by zero")
595 "Indicating if the beta offset will be added to the normalized tensor.")
597 "Indicating if the gamma scale will be multiplied.");
612 refl::ObjectDef<InstanceNormAttrs>()
614 "The axis that represents the channel.")
616 "The axes that along which the normalization is applied.")
618 "Small float added to variance to avoid dividing by zero")
620 "Indicating if the beta offset will be added to the normalized tensor.")
622 "Indicating if the gamma scale will be multiplied.");
635 refl::ObjectDef<RMSNormAttrs>()
637 "The axes that along which the normalization is applied.")
639 "Small float added to variance to avoid dividing by zero");
651 refl::ObjectDef<NLLLossAttrs>()
653 "The reduction method to apply to the output. Can be"
654 "'none', 'mean' or 'sum'.",
655 refl::DefaultValue(
"mean"))
667 refl::ObjectDef<DropoutAttrs>().def_ro(
669 "Fraction of the input that gets dropped out during training time");
682 refl::ObjectDef<AttentionAttrs>()
685 "The custom scale applied before the softmax. The default value is 1 / sqrt(head_dim).")
687 "The type of the causal mask, i.e. 'TopLeft' and 'BottomRight'.")
689 "The size of the window for sliding-window attention.");
702 refl::ObjectDef<PadAttrs>()
704 "Number of values padded to the edges of each axis, "
705 "in the format of (before_1, after_1, ..., before_N, after_N)")
707 refl::DefaultValue(0.0))
709 "Padding type to use. \"constant\" pads with constant_value, "
710 "\"edge\" pads using the edge values of the input array, "
711 "\"reflect\" pads by reflecting values with respect to the edges.",
712 refl::DefaultValue(
"constant"));
723 refl::ObjectDef<PixelShuffleAttrs>().def_ro(
"upscale_factor",
725 "Scale factor for spatial upsampling.");
Adapter for AttrsNode with the new reflection API.
Definition: attrs.h:385
Base class of all attribute class.
Definition: attrs.h:102
Runtime primitive data type.
Definition: data_type.h:47
Definition: repr_printer.h:91
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37
Attributes for 1d adaptive pool operator.
Definition: nn.h:398
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.AdaptivePool1DAttrs", AdaptivePool1DAttrs, BaseAttrsNode)
ffi::String out_layout
Definition: nn.h:401
ffi::String layout
Definition: nn.h:400
ffi::Optional< ffi::Array< IntImm > > output_size
Definition: nn.h:399
static void RegisterReflection()
Definition: nn.h:403
Attributes for 2d adaptive pool operator.
Definition: nn.h:423
static void RegisterReflection()
Definition: nn.h:428
ffi::Optional< ffi::Array< IntImm > > output_size
Definition: nn.h:424
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.AdaptivePool2DAttrs", AdaptivePool2DAttrs, BaseAttrsNode)
ffi::String out_layout
Definition: nn.h:426
ffi::String layout
Definition: nn.h:425
Attributes for 3d adaptive pool operator.
Definition: nn.h:448
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.AdaptivePool3DAttrs", AdaptivePool3DAttrs, BaseAttrsNode)
ffi::Optional< ffi::Array< IntImm > > output_size
Definition: nn.h:449
static void RegisterReflection()
Definition: nn.h:453
ffi::String layout
Definition: nn.h:450
ffi::String out_layout
Definition: nn.h:451
Attributes used in Attention operator.
Definition: nn.h:675
ffi::Optional< IntImm > window_size
Definition: nn.h:678
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.AttentionAttrs", AttentionAttrs, BaseAttrsNode)
ffi::Optional< FloatImm > scale
Definition: nn.h:676
ffi::Optional< ffi::String > causal_mask
Definition: nn.h:677
static void RegisterReflection()
Definition: nn.h:680
Attributes used in batch_norm operator.
Definition: nn.h:525
bool training
Definition: nn.h:531
bool scale
Definition: nn.h:529
static void RegisterReflection()
Definition: nn.h:533
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.BatchNormAttrs", BatchNormAttrs, BaseAttrsNode)
double epsilon
Definition: nn.h:527
int axis
Definition: nn.h:526
double momentum
Definition: nn.h:530
bool center
Definition: nn.h:528
Attributes used in Conv1d operator.
Definition: nn.h:33
ffi::String out_layout
Definition: nn.h:40
int groups
Definition: nn.h:37
ffi::Array< IntImm > padding
Definition: nn.h:35
ffi::Array< IntImm > dilation
Definition: nn.h:36
ffi::String data_layout
Definition: nn.h:38
static void RegisterReflection()
Definition: nn.h:43
DataType out_dtype
Definition: nn.h:41
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.Conv1DAttrs", Conv1DAttrs, BaseAttrsNode)
ffi::Array< IntImm > strides
Definition: nn.h:34
ffi::String kernel_layout
Definition: nn.h:39
Attributes used in Conv1DTranspose operator.
Definition: nn.h:171
ffi::String data_layout
Definition: nn.h:177
static void RegisterReflection()
Definition: nn.h:182
ffi::Array< IntImm > strides
Definition: nn.h:172
DataType out_dtype
Definition: nn.h:180
ffi::Array< IntImm > padding
Definition: nn.h:173
ffi::Array< IntImm > dilation
Definition: nn.h:175
ffi::Array< IntImm > output_padding
Definition: nn.h:174
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.Conv1DTransposeAttrs", Conv1DTransposeAttrs, BaseAttrsNode)
ffi::String out_layout
Definition: nn.h:179
int groups
Definition: nn.h:176
ffi::String kernel_layout
Definition: nn.h:178
Attributes used in Conv2d operator.
Definition: nn.h:77
ffi::String out_layout
Definition: nn.h:84
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.Conv2DAttrs", Conv2DAttrs, BaseAttrsNode)
DataType out_dtype
Definition: nn.h:85
ffi::Array< IntImm > padding
Definition: nn.h:79
ffi::String kernel_layout
Definition: nn.h:83
ffi::Array< IntImm > strides
Definition: nn.h:78
static void RegisterReflection()
Definition: nn.h:87
int groups
Definition: nn.h:81
ffi::String data_layout
Definition: nn.h:82
ffi::Array< IntImm > dilation
Definition: nn.h:80
Attributes used in Conv2d operator.
Definition: nn.h:220
ffi::Array< IntImm > strides
Definition: nn.h:221
ffi::String kernel_layout
Definition: nn.h:227
ffi::String data_layout
Definition: nn.h:226
ffi::Array< IntImm > padding
Definition: nn.h:222
ffi::Array< IntImm > output_padding
Definition: nn.h:223
int groups
Definition: nn.h:225
ffi::Array< IntImm > dilation
Definition: nn.h:224
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.Conv2DTransposeAttrs", Conv2DTransposeAttrs, BaseAttrsNode)
ffi::String out_layout
Definition: nn.h:228
static void RegisterReflection()
Definition: nn.h:231
DataType out_dtype
Definition: nn.h:229
Attributes used in Conv3d operator.
Definition: nn.h:123
ffi::String kernel_layout
Definition: nn.h:129
ffi::Array< IntImm > dilation
Definition: nn.h:126
ffi::String out_layout
Definition: nn.h:130
ffi::Array< IntImm > strides
Definition: nn.h:124
static void RegisterReflection()
Definition: nn.h:133
DataType out_dtype
Definition: nn.h:131
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.Conv3DAttrs", Conv3DAttrs, BaseAttrsNode)
ffi::String data_layout
Definition: nn.h:128
ffi::Array< IntImm > padding
Definition: nn.h:125
int groups
Definition: nn.h:127
Attributes used in dropout operator.
Definition: nn.h:662
double rate
Definition: nn.h:663
static void RegisterReflection()
Definition: nn.h:665
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.DropoutAttrs", DropoutAttrs, BaseAttrsNode)
Attributes used in group_norm operator.
Definition: nn.h:574
ffi::Array< Integer > axes
Definition: nn.h:577
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.GroupNormAttrs", GroupNormAttrs, BaseAttrsNode)
int num_groups
Definition: nn.h:575
int channel_axis
Definition: nn.h:576
double epsilon
Definition: nn.h:578
bool center
Definition: nn.h:579
static void RegisterReflection()
Definition: nn.h:582
bool scale
Definition: nn.h:580
Attributes used in instance_norm operator.
Definition: nn.h:603
bool center
Definition: nn.h:607
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.InstanceNormAttrs", InstanceNormAttrs, BaseAttrsNode)
double epsilon
Definition: nn.h:606
bool scale
Definition: nn.h:608
ffi::Array< Integer > axes
Definition: nn.h:605
static void RegisterReflection()
Definition: nn.h:610
int channel_axis
Definition: nn.h:604
Attributes used in layer_norm operator.
Definition: nn.h:552
ffi::Array< Integer > axes
Definition: nn.h:553
bool scale
Definition: nn.h:556
static void RegisterReflection()
Definition: nn.h:558
bool center
Definition: nn.h:555
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.LayerNormAttrs", LayerNormAttrs, BaseAttrsNode)
double epsilon
Definition: nn.h:554
Attributes used in softmax operators.
Definition: nn.h:485
static void RegisterReflection()
Definition: nn.h:488
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.LeakyReluAttrs", LeakyReluAttrs, BaseAttrsNode)
double alpha
Definition: nn.h:486
Attributes used in nll_loss operator.
Definition: nn.h:645
static void RegisterReflection()
Definition: nn.h:649
ffi::String reduction
Definition: nn.h:646
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.NLLLossAttrs", NLLLossAttrs, BaseAttrsNode)
int ignore_index
Definition: nn.h:647
Attributes used in PReLU operator.
Definition: nn.h:513
static void RegisterReflection()
Definition: nn.h:516
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.PReluAttrs", PReluAttrs, BaseAttrsNode)
int axis
Definition: nn.h:514
Attributes used for the padding operator.
Definition: nn.h:695
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.PadAttrs", PadAttrs, BaseAttrsNode)
ffi::Array< Integer > pad_width
Definition: nn.h:696
tvm::ffi::String pad_mode
Definition: nn.h:698
double pad_value
Definition: nn.h:697
static void RegisterReflection()
Definition: nn.h:700
Attributes used for the pixel shuffle operator.
Definition: nn.h:718
int upscale_factor
Definition: nn.h:719
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.PixelShuffleAttrs", PixelShuffleAttrs, BaseAttrsNode)
static void RegisterReflection()
Definition: nn.h:721
Attributes used in max_pool1d and avg_pool1d operator.
Definition: nn.h:271
ffi::Array< IntImm > strides
Definition: nn.h:273
ffi::Array< IntImm > padding
Definition: nn.h:274
static void RegisterReflection()
Definition: nn.h:281
ffi::String layout
Definition: nn.h:278
ffi::Array< IntImm > dilation
Definition: nn.h:275
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.Pool1DAttrs", Pool1DAttrs, BaseAttrsNode)
bool count_include_pad
Definition: nn.h:277
ffi::Array< IntImm > pool_size
Definition: nn.h:272
ffi::String out_layout
Definition: nn.h:279
bool ceil_mode
Definition: nn.h:276
Attributes used in max_pool2d and avg_pool2d operator.
Definition: nn.h:312
static void RegisterReflection()
Definition: nn.h:322
bool count_include_pad
Definition: nn.h:318
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.Pool2DAttrs", Pool2DAttrs, BaseAttrsNode)
ffi::String out_layout
Definition: nn.h:320
ffi::Array< IntImm > padding
Definition: nn.h:315
ffi::Array< IntImm > pool_size
Definition: nn.h:313
bool ceil_mode
Definition: nn.h:317
ffi::String layout
Definition: nn.h:319
ffi::Array< IntImm > dilation
Definition: nn.h:316
ffi::Array< IntImm > strides
Definition: nn.h:314
Attributes used in max_pool3d and avg_pool3d operator.
Definition: nn.h:355
bool ceil_mode
Definition: nn.h:360
ffi::String out_layout
Definition: nn.h:363
ffi::Array< IntImm > padding
Definition: nn.h:358
ffi::String layout
Definition: nn.h:362
static void RegisterReflection()
Definition: nn.h:365
bool count_include_pad
Definition: nn.h:361
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.Pool3DAttrs", Pool3DAttrs, BaseAttrsNode)
ffi::Array< IntImm > strides
Definition: nn.h:357
ffi::Array< IntImm > pool_size
Definition: nn.h:356
ffi::Array< IntImm > dilation
Definition: nn.h:359
Attributes used in rms_norm operator.
Definition: nn.h:629
ffi::Array< Integer > axes
Definition: nn.h:630
double epsilon
Definition: nn.h:631
static void RegisterReflection()
Definition: nn.h:633
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.RMSNormAttrs", RMSNormAttrs, BaseAttrsNode)
Attributes used in softmax operators.
Definition: nn.h:473
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.SoftmaxAttrs", SoftmaxAttrs, BaseAttrsNode)
int axis
Definition: nn.h:474
static void RegisterReflection()
Definition: nn.h:476
Attributes used in softplus operators.
Definition: nn.h:497
double threshold
Definition: nn.h:499
double beta
Definition: nn.h:498
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.SoftplusAttrs", SoftplusAttrs, BaseAttrsNode)
static void RegisterReflection()
Definition: nn.h:501