24 #ifndef TVM_RELAX_ATTRS_STATISTICAL_H_
25 #define TVM_RELAX_ATTRS_STATISTICAL_H_
34 ffi::Optional<ffi::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 "
53 ffi::Optional<int64_t>
axis;
59 refl::ObjectDef<ScanopAttrs>()
61 "The axis along which to perform the scan computation."
62 "The default (None) is to compute over the flattened array.")
64 "The output data type."
65 "If dtype is not specified, it defaults to the dtype of input data.")
67 refl::DefaultValue(
Bool(
false)));
Adapter for AttrsNode with the new reflection API.
Definition: attrs.h:385
Base class of all attribute class.
Definition: attrs.h:102
Boolean constant.
Definition: expr.h:565
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:52
ffi::Optional< int64_t > axis
Definition: statistical.h:53
DataType dtype
Definition: statistical.h:54
Bool exclusive
Definition: statistical.h:55
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.ScanopAttrs", ScanopAttrs, BaseAttrsNode)
static void RegisterReflection()
Definition: statistical.h:57
Attributes for statistical operators.
Definition: statistical.h:33
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.StatisticalAttrs", StatisticalAttrs, BaseAttrsNode)
static void RegisterReflection()
Definition: statistical.h:37
ffi::Optional< ffi::Array< Integer > > axis
Definition: statistical.h:34
bool keepdims
Definition: statistical.h:35