24 #ifndef TVM_RELAX_ATTRS_SEARCH_H_
25 #define TVM_RELAX_ATTRS_SEARCH_H_
34 ffi::Optional<int64_t>
axis;
39 refl::ObjectDef<ArgmaxArgminAttrs>()
41 "The axis along which to perform the argmin/argmax.")
43 "If this is set to `True`, the reduced axis is left in the result as dimension "
58 refl::ObjectDef<BucketizeAttrs>()
60 "Indicate the output datatype, int32 if True, int64 otherwise.")
62 "Determines the behavior for values in boundaries");
Adapter for AttrsNode with the new reflection API.
Definition: attrs.h:385
Base class of all attribute class.
Definition: attrs.h:102
Definition: repr_printer.h:91
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37
Attributes for search operators.
Definition: search.h:33
bool keepdims
Definition: search.h:35
static void RegisterReflection()
Definition: search.h:37
ffi::Optional< int64_t > axis
Definition: search.h:34
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.ArgmaxArgminAttrs", ArgmaxArgminAttrs, BaseAttrsNode)
Attributes for bucketize operator.
Definition: search.h:52
static void RegisterReflection()
Definition: search.h:56
bool right
Definition: search.h:54
bool out_int32
Definition: search.h:53
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.BucketizeAttrs", BucketizeAttrs, BaseAttrsNode)