24 #ifndef TVM_RELAX_ATTRS_DATATYPE_H_
25 #define TVM_RELAX_ATTRS_DATATYPE_H_
38 refl::ObjectDef<AstypeAttrs>().def_ro(
"dtype", &
AstypeAttrs::dtype,
"Target data type");
41 static constexpr
const char*
_type_key =
"relax.attrs.AstypeAttrs";
54 static constexpr
const char*
_type_key =
"relax.attrs.WrapParamAttrs";
Adapter for AttrsNode with the new reflection API.
Definition: attrs.h:384
Base class of all attribute class.
Definition: attrs.h:103
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 astype operator.
Definition: datatype.h:33
TVM_FFI_DECLARE_FINAL_OBJECT_INFO(AstypeAttrs, BaseAttrsNode)
DataType dtype
Definition: datatype.h:34
static void RegisterReflection()
Definition: datatype.h:36
static constexpr const char * _type_key
Definition: datatype.h:41
Attributes used in wrap_param operator.
Definition: datatype.h:46
DataType dtype
Definition: datatype.h:47
TVM_FFI_DECLARE_FINAL_OBJECT_INFO(WrapParamAttrs, BaseAttrsNode)
static void RegisterReflection()
Definition: datatype.h:49
static constexpr const char * _type_key
Definition: datatype.h:54