24 #ifndef TVM_RELAX_ATTRS_IMAGE_H_
25 #define TVM_RELAX_ATTRS_IMAGE_H_
34 ffi::Array<FloatImm>
roi;
46 refl::ObjectDef<Resize2DAttrs>()
48 "Region of Interest for coordinate transformation mode 'tf_crop_and_resize'")
50 "Dimension ordering of input data. Can be 'NCHW', 'NHWC', etc."
51 "'N', 'C', 'H', 'W' stands for batch, channel, height, and width"
52 "dimensions respectively. Resize is applied on the 'H' and"
55 "Specify the mode to use for scaling."
56 "nearest_neighbor - Nearest Neighbor"
57 "linear - Bilinear Interpolation"
58 "cubic - Bicubic Interpolation")
60 "Describes how to transform the coordinate in the resized tensor"
61 "to the coordinate in the original tensor."
62 "Refer to the ONNX Resize operator specification for details"
63 "Available options are half_pixel, align_corners and asymmetric")
65 "indicates how to find the \"nearest\" pixel in nearest_neighbor method"
66 "Available options are round, floor, and ceil.")
68 "Spline Coefficient for Bicubic Interpolation")
70 "Flag to exclude exterior of the image during bicubic interpolation")
72 "Value to return when roi is outside of the image")
75 "The dtype of the output tensor. It it is not specified, the output will have the same "
76 "dtype as input if not specified.");
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 used in image resize2d operator.
Definition: image.h:33
ffi::Array< FloatImm > roi
Definition: image.h:34
DataType out_dtype
Definition: image.h:42
double extrapolation_value
Definition: image.h:41
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.Resize2DAttrs", Resize2DAttrs, BaseAttrsNode)
ffi::String method
Definition: image.h:36
ffi::String layout
Definition: image.h:35
static void RegisterReflection()
Definition: image.h:44
int cubic_exclude
Definition: image.h:40
double cubic_alpha
Definition: image.h:39
ffi::String rounding_method
Definition: image.h:38
ffi::String coordinate_transformation_mode
Definition: image.h:37