24 #ifndef TVM_RELAX_ATTRS_INDEX_H_
25 #define TVM_RELAX_ATTRS_INDEX_H_
39 refl::ObjectDef<TakeAttrs>()
40 .def_ro(
"axis", &
TakeAttrs::axis,
"The axis over which to select values.")
41 .def_ro(
"mode", &
TakeAttrs::mode,
"The mode for handling out-of-bounds indices.",
42 refl::DefaultValue(
"fast"));
45 static constexpr
const char*
_type_key =
"relax.attrs.TakeAttrs";
55 refl::ObjectDef<StridedSliceAttrs>().def_ro(
57 "Whether to assume the indices are in bound. If it is set to false, "
58 "out of bound indices will be clipped to the bound.",
59 refl::DefaultValue(
true));
62 static constexpr
const char*
_type_key =
"relax.attrs.StridedSliceAttrs";
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 used in strided_slice operator.
Definition: index.h:50
TVM_FFI_DECLARE_FINAL_OBJECT_INFO(StridedSliceAttrs, BaseAttrsNode)
bool assume_inbound
Definition: index.h:51
static constexpr const char * _type_key
Definition: index.h:62
static void RegisterReflection()
Definition: index.h:53
Attributes used in take operator.
Definition: index.h:33
TVM_FFI_DECLARE_FINAL_OBJECT_INFO(TakeAttrs, BaseAttrsNode)
String mode
Definition: index.h:35
static constexpr const char * _type_key
Definition: index.h:45
Optional< int64_t > axis
Definition: index.h:34
static void RegisterReflection()
Definition: index.h:37