24 #ifndef TVM_RELAX_ATTRS_SEARCH_H_
25 #define TVM_RELAX_ATTRS_SEARCH_H_
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 "
48 static constexpr
const char*
_type_key =
"relax.attrs.ArgmaxArgminAttrs";
59 refl::ObjectDef<BucketizeAttrs>()
61 "Indicate the output datatype, int32 if True, int64 otherwise.")
63 "Determines the behavior for values in boundaries");
66 static constexpr
const char*
_type_key =
"relax.attrs.BucketizeAttrs";
Adapter for AttrsNode with the new reflection API.
Definition: attrs.h:384
Base class of all attribute class.
Definition: attrs.h:103
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
Optional< int64_t > axis
Definition: search.h:34
static void RegisterReflection()
Definition: search.h:37
static constexpr const char * _type_key
Definition: search.h:48
TVM_FFI_DECLARE_FINAL_OBJECT_INFO(ArgmaxArgminAttrs, BaseAttrsNode)
Attributes for bucketize operator.
Definition: search.h:53
TVM_FFI_DECLARE_FINAL_OBJECT_INFO(BucketizeAttrs, BaseAttrsNode)
static void RegisterReflection()
Definition: search.h:57
bool right
Definition: search.h:55
bool out_int32
Definition: search.h:54
static constexpr const char * _type_key
Definition: search.h:66