24 #ifndef TVM_RELAX_ATTRS_STATISTICAL_H_
25 #define TVM_RELAX_ATTRS_STATISTICAL_H_
34 Optional<Array<Integer>>
axis;
39 refl::ObjectDef<StatisticalAttrs>()
41 "The axis or axes along which to perform the reduction.")
43 "If this is set to `True`, the reduced axes are left in the result as dimension "
48 static constexpr
const char*
_type_key =
"relax.attrs.StatisticalAttrs";
60 refl::ObjectDef<ScanopAttrs>()
62 "The axis along which to perform the scan computation."
63 "The default (None) is to compute over the flattened array.")
65 "The output data type."
66 "If dtype is not specified, it defaults to the dtype of input data.")
68 refl::DefaultValue(
Bool(
false)));
71 static constexpr
const char*
_type_key =
"relax.attrs.ScanopAttrs";
Adapter for AttrsNode with the new reflection API.
Definition: attrs.h:384
Base class of all attribute class.
Definition: attrs.h:103
Boolean constant.
Definition: expr.h:577
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 scan operators like cumsum, cumprod.
Definition: statistical.h:53
static constexpr const char * _type_key
Definition: statistical.h:71
DataType dtype
Definition: statistical.h:55
Bool exclusive
Definition: statistical.h:56
TVM_FFI_DECLARE_FINAL_OBJECT_INFO(ScanopAttrs, BaseAttrsNode)
Optional< int64_t > axis
Definition: statistical.h:54
static void RegisterReflection()
Definition: statistical.h:58
Attributes for statistical operators.
Definition: statistical.h:33
static void RegisterReflection()
Definition: statistical.h:37
static constexpr const char * _type_key
Definition: statistical.h:48
Optional< Array< Integer > > axis
Definition: statistical.h:34
bool keepdims
Definition: statistical.h:35
TVM_FFI_DECLARE_FINAL_OBJECT_INFO(StatisticalAttrs, BaseAttrsNode)