24 #ifndef TVM_RELAX_ATTRS_OP_H_
25 #define TVM_RELAX_ATTRS_OP_H_
41 "The name of the te gradient function associated with this call_tir_with_grad node.");
43 .describe(
"The keyword arguments passed to the te gradient function.");
54 "Indices that describe which input corresponds to which output. If the `i`th member "
55 "has the value `k` >= 0, then that means that input `k` should be used to store the "
56 "`i`th output. If an element has the value -1, that means a new tensor should be "
57 "allocated for that output.");
68 "Indices that describe which input corresponds to which output. If the `i`th member "
69 "has the value `k` >= 0, then that means that input `k` should be used to store the "
70 "`i`th output. If an element has the value -1, that means the output will be newly "
#define TVM_ATTR_FIELD(FieldName)
Declare an attribute field.
Definition: attrs.h:76
The base class of the all the Use "curiously recurring template pattern".
Definition: attrs.h:870
Managed reference to VDeviceNode.
Definition: global_info.h:95
Array, container representing a contiguous sequence of ObjectRefs.
Definition: array.h:289
Map container of NodeRef->NodeRef in DSL graph. Map implements copy on write semantics,...
Definition: map.h:1271
Reference to string objects.
Definition: string.h:97
GlobalInfo are globally static object that are referred by the IR itself.
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:36
Attributes used in call_inplace_packed.
Definition: op.h:62
TVM_DECLARE_ATTRS(CallInplacePackedAttrs, "relax.attrs.CallInplacePackedAttrs")
Definition: op.h:65
Array< Integer > inplace_indices
Definition: op.h:63
Attributes used in call_tir_inplace.
Definition: op.h:48
Array< Integer > inplace_indices
Definition: op.h:49
TVM_DECLARE_ATTRS(CallTIRInplaceAttrs, "relax.attrs.CallTIRInplaceAttrs")
Definition: op.h:51
Attributes used in call_tir_with_grad.
Definition: op.h:34
String te_grad_name
Definition: op.h:35
Map< String, ObjectRef > te_grad_kwargs
Definition: op.h:36
TVM_DECLARE_ATTRS(CallTIRWithGradAttrs, "relax.attrs.CallTIRWithGradAttrs")
Definition: op.h:38
Attributes used in hint_on_device.
Definition: op.h:84
int32_t dev_id
Definition: op.h:86
int32_t dev_type
Definition: op.h:85
TVM_DECLARE_ATTRS(HintOnDeviceAttrs, "relax.attrs.HintOnDeviceAttrs")
Definition: op.h:87
Attributes used in to_vdevice.
Definition: op.h:76
VDevice dst_vdevice
Definition: op.h:77
TVM_DECLARE_ATTRS(ToVDeviceAttrs, "relax.attrs.ToVDeviceAttrs")
Definition: op.h:78