24 #ifndef TVM_RELAX_ATTRS_OP_H_
25 #define TVM_RELAX_ATTRS_OP_H_
40 refl::ObjectDef<CallTIRWithGradAttrs>()
43 "The name of the te gradient function associated with this call_tir_with_grad node.")
45 "The keyword arguments passed to the te gradient function.");
48 static constexpr
const char*
_type_key =
"relax.attrs.CallTIRWithGradAttrs";
65 refl::ObjectDef<CallTIRInplaceAttrs>().def_ro(
"inplace_indices",
69 static constexpr
const char*
_type_key =
"relax.attrs.CallTIRInplaceAttrs";
86 refl::ObjectDef<CallInplacePackedAttrs>().def_ro(
"inplace_indices",
90 static constexpr
const char*
_type_key =
"relax.attrs.CallInplacePackedAttrs";
101 "The destination device where the data is copied to.");
104 static constexpr
const char*
_type_key =
"relax.attrs.ToVDeviceAttrs";
115 refl::ObjectDef<HintOnDeviceAttrs>()
117 "The device type where the data is supposed to be executed.")
121 static constexpr
const char*
_type_key =
"relax.attrs.HintOnDeviceAttrs";
Adapter for AttrsNode with the new reflection API.
Definition: attrs.h:384
Base class of all attribute class.
Definition: attrs.h:103
Managed reference to VDeviceNode.
Definition: global_info.h:90
GlobalInfo are globally static object that are referred by the IR itself.
Definition: repr_printer.h:91
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37
Attributes used in call_inplace_packed.
Definition: op.h:74
TVM_FFI_DECLARE_FINAL_OBJECT_INFO(CallInplacePackedAttrs, BaseAttrsNode)
static constexpr const char * _type_key
Definition: op.h:90
Array< Integer > inplace_indices
Indices that describe which input corresponds to which output.
Definition: op.h:82
static void RegisterReflection()
Definition: op.h:84
Attributes used in call_tir_inplace.
Definition: op.h:53
TVM_FFI_DECLARE_FINAL_OBJECT_INFO(CallTIRInplaceAttrs, BaseAttrsNode)
static void RegisterReflection()
Definition: op.h:63
Array< Integer > inplace_indices
Indices that describe which input corresponds to which output.
Definition: op.h:61
static constexpr const char * _type_key
Definition: op.h:69
Attributes used in call_tir_with_grad.
Definition: op.h:34
String te_grad_name
Definition: op.h:35
Map< String, Any > te_grad_kwargs
Definition: op.h:36
static void RegisterReflection()
Definition: op.h:38
TVM_FFI_DECLARE_FINAL_OBJECT_INFO(CallTIRWithGradAttrs, BaseAttrsNode)
static constexpr const char * _type_key
Definition: op.h:48
Attributes used in hint_on_device.
Definition: op.h:109
int32_t dev_id
Definition: op.h:111
TVM_FFI_DECLARE_FINAL_OBJECT_INFO(HintOnDeviceAttrs, BaseAttrsNode)
int32_t dev_type
Definition: op.h:110
static void RegisterReflection()
Definition: op.h:113
static constexpr const char * _type_key
Definition: op.h:121
Attributes used in to_vdevice.
Definition: op.h:95
static void RegisterReflection()
Definition: op.h:98
static constexpr const char * _type_key
Definition: op.h:104
VDevice dst_vdevice
Definition: op.h:96
TVM_FFI_DECLARE_FINAL_OBJECT_INFO(ToVDeviceAttrs, BaseAttrsNode)