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.");
64 refl::ObjectDef<CallTIRInplaceAttrs>().def_ro(
"inplace_indices",
84 refl::ObjectDef<CallInplacePackedAttrs>().def_ro(
"inplace_indices",
98 "The destination device where the data is copied to.");
110 refl::ObjectDef<HintOnDeviceAttrs>()
112 "The device type where the data is supposed to be executed.")
Adapter for AttrsNode with the new reflection API.
Definition: attrs.h:385
Base class of all attribute class.
Definition: attrs.h:102
Managed reference to VDeviceNode.
Definition: global_info.h:87
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:72
ffi::Array< Integer > inplace_indices
Indices that describe which input corresponds to which output.
Definition: op.h:80
static void RegisterReflection()
Definition: op.h:82
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.CallInplacePackedAttrs", CallInplacePackedAttrs, BaseAttrsNode)
Attributes used in call_tir_inplace.
Definition: op.h:52
static void RegisterReflection()
Definition: op.h:62
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.CallTIRInplaceAttrs", CallTIRInplaceAttrs, BaseAttrsNode)
ffi::Array< Integer > inplace_indices
Indices that describe which input corresponds to which output.
Definition: op.h:60
Attributes used in call_tir_with_grad.
Definition: op.h:34
ffi::String te_grad_name
Definition: op.h:35
static void RegisterReflection()
Definition: op.h:38
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.CallTIRWithGradAttrs", CallTIRWithGradAttrs, BaseAttrsNode)
ffi::Map< ffi::String, Any > te_grad_kwargs
Definition: op.h:36
Attributes used in hint_on_device.
Definition: op.h:104
int32_t device_type
Definition: op.h:105
static void RegisterReflection()
Definition: op.h:108
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.HintOnDeviceAttrs", HintOnDeviceAttrs, BaseAttrsNode)
int32_t index
Definition: op.h:106
Attributes used in to_vdevice.
Definition: op.h:92
static void RegisterReflection()
Definition: op.h:95
VDevice dst_vdevice
Definition: op.h:93
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.attrs.ToVDeviceAttrs", ToVDeviceAttrs, BaseAttrsNode)