tvm
|
Base class of all attribute class. More...
#include <attrs.h>
Public Types | |
using | TVMArgs = runtime::TVMArgs |
using | TVMRetValue = runtime::TVMRetValue |
![]() | |
typedef void(* | FDeleter) (Object *self) |
Object deleter. More... | |
using | RefCounterType = std::atomic< int32_t > |
Public Member Functions | |
virtual | ~BaseAttrsNode () |
virtual destructor More... | |
virtual void | VisitAttrs (AttrVisitor *v) |
template<typename... Args> | |
void | InitBySeq (Args &&... args) |
Initialize the attributes by sequence of arguments. More... | |
void | PrintDocString (std::ostream &os) const |
Print readible docstring to ostream, add newline. More... | |
virtual void | VisitNonDefaultAttrs (AttrVisitor *v)=0 |
Visit attributes that do not equal the default value. More... | |
virtual Array< AttrFieldInfo > | ListFieldInfo () const =0 |
Get the field information. More... | |
virtual void | InitByPackedArgs (const TVMArgs &kwargs, bool allow_unknown=false)=0 |
Initialize the attributes by arguments. More... | |
TVM_DECLARE_BASE_OBJECT_INFO (BaseAttrsNode, Object) | |
![]() | |
uint32_t | type_index () const |
std::string | GetTypeKey () const |
size_t | GetTypeKeyHash () const |
template<typename TargetType > | |
bool | IsInstance () const |
bool | unique () const |
Object () | |
Object (const Object &other) | |
Object (Object &&other) | |
Object & | operator= (const Object &other) |
Object & | operator= (Object &&other) |
Static Public Attributes | |
static constexpr const bool | _type_has_method_sequal_reduce = true |
static constexpr const bool | _type_has_method_shash_reduce = true |
static constexpr const char * | _type_key = "Attrs" |
![]() | |
static constexpr const char * | _type_key = "runtime.Object" |
static constexpr bool | _type_final = false |
static constexpr uint32_t | _type_child_slots = 0 |
static constexpr bool | _type_child_slots_can_overflow = true |
static constexpr bool | _type_has_method_visit_attrs = true |
static constexpr bool | _type_has_method_sequal_reduce = false |
static constexpr bool | _type_has_method_shash_reduce = false |
static constexpr uint32_t | _type_index = TypeIndex::kDynamic |
Additional Inherited Members | |
![]() | |
static std::string | TypeIndex2Key (uint32_t tindex) |
Get the type key of the corresponding index from runtime. More... | |
static size_t | TypeIndex2KeyHash (uint32_t tindex) |
Get the type key hash of the corresponding index from runtime. More... | |
static uint32_t | TypeKey2Index (const std::string &key) |
Get the type index of the corresponding key from runtime. More... | |
static uint32_t | _GetOrAllocRuntimeTypeIndex () |
static uint32_t | RuntimeTypeIndex () |
![]() | |
void | IncRef () |
developer function, increases reference counter. More... | |
void | DecRef () |
developer function, decrease reference counter. More... | |
![]() | |
static uint32_t | GetOrAllocRuntimeTypeIndex (const std::string &key, uint32_t static_tindex, uint32_t parent_tindex, uint32_t type_child_slots, bool type_child_slots_can_overflow) |
Get the type index using type key. More... | |
![]() | |
uint32_t | type_index_ {0} |
Type index(tag) that indicates the type of the object. More... | |
RefCounterType | ref_counter_ {0} |
The internal reference counter. More... | |
FDeleter | deleter_ = nullptr |
deleter of this object to enable customized allocation. If the deleter is nullptr, no deletion will be performed. The creator of the object must always set the deleter field properly. More... | |
Base class of all attribute class.
|
inlinevirtual |
virtual destructor
|
pure virtual |
Initialize the attributes by arguments.
kwargs | The key value pairs for initialization. [key0, value0, key1, value1, ..., key_n, value_n] |
allow_unknown | Whether allow additional unknown fields. |
Implemented in tvm::AttrsNode< DerivedType >, tvm::AttrsNode< ScatterCollectiveAttrs >, tvm::AttrsNode< HintOnDeviceAttrs >, tvm::AttrsNode< TakeAttrs >, tvm::AttrsNode< ScatterElementsAttrs >, tvm::AttrsNode< GatherElementsAttrs >, tvm::AttrsNode< EinsumAttrs >, tvm::AttrsNode< DistributionAttrs >, tvm::AttrsNode< CallInplacePackedAttrs >, tvm::AttrsNode< AttentionAttrs >, tvm::AttrsNode< AllReduceAttrs >, tvm::AttrsNode< AdaptivePool2DAttrs >, tvm::AttrsNode< ScatterNDAttrs >, tvm::AttrsNode< InitAttrs >, tvm::AttrsNode< FlipAttrs >, tvm::AttrsNode< AllGatherAttrs >, tvm::AttrsNode< AdaptivePool3DAttrs >, tvm::AttrsNode< ToVDeviceAttrs >, tvm::AttrsNode< ExpandDimsAttrs >, tvm::AttrsNode< ConcatAttrs >, tvm::AttrsNode< TriluAttrs >, tvm::AttrsNode< SoftmaxAttrs >, tvm::AttrsNode< AdaptivePool1DAttrs >, tvm::AttrsNode< MatmulAttrs >, tvm::AttrsNode< ArgmaxArgminAttrs >, tvm::AttrsNode< Resize2DAttrs >, tvm::AttrsNode< Pool3DAttrs >, tvm::AttrsNode< LayerNormAttrs >, tvm::AttrsNode< SqueezeAttrs >, tvm::AttrsNode< Pool2DAttrs >, tvm::AttrsNode< OneHotAttrs >, tvm::AttrsNode< NLLLossAttrs >, tvm::AttrsNode< BatchNormAttrs >, tvm::AttrsNode< RepeatAttrs >, tvm::AttrsNode< Pool1DAttrs >, tvm::AttrsNode< ScanopAttrs >, tvm::AttrsNode< LeakyReluAttrs >, tvm::AttrsNode< CallTIRWithGradAttrs >, tvm::AttrsNode< CallTIRInplaceAttrs >, tvm::AttrsNode< AstypeAttrs >, tvm::AttrsNode< PadAttrs >, tvm::AttrsNode< DropoutAttrs >, tvm::AttrsNode< TopKAttrs >, tvm::AttrsNode< MultinomialFromUniformAttrs >, tvm::AttrsNode< Conv3DAttrs >, tvm::AttrsNode< VirtualDeviceNode >, tvm::AttrsNode< WrapParamAttrs >, tvm::AttrsNode< TileAttrs >, tvm::AttrsNode< StridedSliceAttrs >, tvm::AttrsNode< SplitAttrs >, tvm::AttrsNode< RMSNormAttrs >, tvm::AttrsNode< QuantizeAttrs >, tvm::AttrsNode< Conv2DAttrs >, tvm::AttrsNode< SortAttrs >, tvm::AttrsNode< Conv1DAttrs >, tvm::AttrsNode< ArgsortAttrs >, tvm::AttrsNode< StatisticalAttrs >, tvm::AttrsNode< LayoutTransformAttrs >, tvm::AttrsNode< Conv1DTransposeAttrs >, tvm::AttrsNode< Conv2DTransposeAttrs >, tvm::AttrsNode< PermuteDimsAttrs >, tvm::AttrsNode< GroupNormAttrs >, tvm::AttrsNode< GatherNDAttrs >, and tvm::DictAttrsNode.
|
inline |
Initialize the attributes by sequence of arguments.
args | The positional arguments in the form [key0, value0, key1, value1, ..., key_n, value_n] |
|
pure virtual |
Get the field information.
Implemented in tvm::AttrsNode< DerivedType >, tvm::AttrsNode< ScatterCollectiveAttrs >, tvm::AttrsNode< HintOnDeviceAttrs >, tvm::AttrsNode< TakeAttrs >, tvm::AttrsNode< ScatterElementsAttrs >, tvm::AttrsNode< GatherElementsAttrs >, tvm::AttrsNode< EinsumAttrs >, tvm::AttrsNode< DistributionAttrs >, tvm::AttrsNode< CallInplacePackedAttrs >, tvm::AttrsNode< AttentionAttrs >, tvm::AttrsNode< AllReduceAttrs >, tvm::AttrsNode< AdaptivePool2DAttrs >, tvm::AttrsNode< ScatterNDAttrs >, tvm::AttrsNode< InitAttrs >, tvm::AttrsNode< FlipAttrs >, tvm::AttrsNode< AllGatherAttrs >, tvm::AttrsNode< AdaptivePool3DAttrs >, tvm::AttrsNode< ToVDeviceAttrs >, tvm::AttrsNode< ExpandDimsAttrs >, tvm::AttrsNode< ConcatAttrs >, tvm::AttrsNode< TriluAttrs >, tvm::AttrsNode< SoftmaxAttrs >, tvm::AttrsNode< AdaptivePool1DAttrs >, tvm::AttrsNode< MatmulAttrs >, tvm::AttrsNode< ArgmaxArgminAttrs >, tvm::AttrsNode< Resize2DAttrs >, tvm::AttrsNode< Pool3DAttrs >, tvm::AttrsNode< LayerNormAttrs >, tvm::AttrsNode< SqueezeAttrs >, tvm::AttrsNode< Pool2DAttrs >, tvm::AttrsNode< OneHotAttrs >, tvm::AttrsNode< NLLLossAttrs >, tvm::AttrsNode< BatchNormAttrs >, tvm::AttrsNode< RepeatAttrs >, tvm::AttrsNode< Pool1DAttrs >, tvm::AttrsNode< ScanopAttrs >, tvm::AttrsNode< LeakyReluAttrs >, tvm::AttrsNode< CallTIRWithGradAttrs >, tvm::AttrsNode< CallTIRInplaceAttrs >, tvm::AttrsNode< AstypeAttrs >, tvm::AttrsNode< PadAttrs >, tvm::AttrsNode< DropoutAttrs >, tvm::AttrsNode< TopKAttrs >, tvm::AttrsNode< MultinomialFromUniformAttrs >, tvm::AttrsNode< Conv3DAttrs >, tvm::AttrsNode< VirtualDeviceNode >, tvm::AttrsNode< WrapParamAttrs >, tvm::AttrsNode< TileAttrs >, tvm::AttrsNode< StridedSliceAttrs >, tvm::AttrsNode< SplitAttrs >, tvm::AttrsNode< RMSNormAttrs >, tvm::AttrsNode< QuantizeAttrs >, tvm::AttrsNode< Conv2DAttrs >, tvm::AttrsNode< SortAttrs >, tvm::AttrsNode< Conv1DAttrs >, tvm::AttrsNode< ArgsortAttrs >, tvm::AttrsNode< StatisticalAttrs >, tvm::AttrsNode< LayoutTransformAttrs >, tvm::AttrsNode< Conv1DTransposeAttrs >, tvm::AttrsNode< Conv2DTransposeAttrs >, tvm::AttrsNode< PermuteDimsAttrs >, tvm::AttrsNode< GroupNormAttrs >, tvm::AttrsNode< GatherNDAttrs >, and tvm::DictAttrsNode.
|
inline |
Print readible docstring to ostream, add newline.
os | the stream to print the docstring to. |
tvm::BaseAttrsNode::TVM_DECLARE_BASE_OBJECT_INFO | ( | BaseAttrsNode | , |
Object | |||
) |
|
inlinevirtual |
Reimplemented in tvm::DictAttrsNode, tvm::AttrsNode< DerivedType >, tvm::AttrsNode< ScatterCollectiveAttrs >, tvm::AttrsNode< HintOnDeviceAttrs >, tvm::AttrsNode< TakeAttrs >, tvm::AttrsNode< ScatterElementsAttrs >, tvm::AttrsNode< GatherElementsAttrs >, tvm::AttrsNode< EinsumAttrs >, tvm::AttrsNode< DistributionAttrs >, tvm::AttrsNode< CallInplacePackedAttrs >, tvm::AttrsNode< AttentionAttrs >, tvm::AttrsNode< AllReduceAttrs >, tvm::AttrsNode< AdaptivePool2DAttrs >, tvm::AttrsNode< ScatterNDAttrs >, tvm::AttrsNode< InitAttrs >, tvm::AttrsNode< FlipAttrs >, tvm::AttrsNode< AllGatherAttrs >, tvm::AttrsNode< AdaptivePool3DAttrs >, tvm::AttrsNode< ToVDeviceAttrs >, tvm::AttrsNode< ExpandDimsAttrs >, tvm::AttrsNode< ConcatAttrs >, tvm::AttrsNode< TriluAttrs >, tvm::AttrsNode< SoftmaxAttrs >, tvm::AttrsNode< AdaptivePool1DAttrs >, tvm::AttrsNode< MatmulAttrs >, tvm::AttrsNode< ArgmaxArgminAttrs >, tvm::AttrsNode< Resize2DAttrs >, tvm::AttrsNode< Pool3DAttrs >, tvm::AttrsNode< LayerNormAttrs >, tvm::AttrsNode< SqueezeAttrs >, tvm::AttrsNode< Pool2DAttrs >, tvm::AttrsNode< OneHotAttrs >, tvm::AttrsNode< NLLLossAttrs >, tvm::AttrsNode< BatchNormAttrs >, tvm::AttrsNode< RepeatAttrs >, tvm::AttrsNode< Pool1DAttrs >, tvm::AttrsNode< ScanopAttrs >, tvm::AttrsNode< LeakyReluAttrs >, tvm::AttrsNode< CallTIRWithGradAttrs >, tvm::AttrsNode< CallTIRInplaceAttrs >, tvm::AttrsNode< AstypeAttrs >, tvm::AttrsNode< PadAttrs >, tvm::AttrsNode< DropoutAttrs >, tvm::AttrsNode< TopKAttrs >, tvm::AttrsNode< MultinomialFromUniformAttrs >, tvm::AttrsNode< Conv3DAttrs >, tvm::AttrsNode< VirtualDeviceNode >, tvm::AttrsNode< WrapParamAttrs >, tvm::AttrsNode< TileAttrs >, tvm::AttrsNode< StridedSliceAttrs >, tvm::AttrsNode< SplitAttrs >, tvm::AttrsNode< RMSNormAttrs >, tvm::AttrsNode< QuantizeAttrs >, tvm::AttrsNode< Conv2DAttrs >, tvm::AttrsNode< SortAttrs >, tvm::AttrsNode< Conv1DAttrs >, tvm::AttrsNode< ArgsortAttrs >, tvm::AttrsNode< StatisticalAttrs >, tvm::AttrsNode< LayoutTransformAttrs >, tvm::AttrsNode< Conv1DTransposeAttrs >, tvm::AttrsNode< Conv2DTransposeAttrs >, tvm::AttrsNode< PermuteDimsAttrs >, tvm::AttrsNode< GroupNormAttrs >, and tvm::AttrsNode< GatherNDAttrs >.
|
pure virtual |
Visit attributes that do not equal the default value.
v | The visitor |
Implemented in tvm::DictAttrsNode, tvm::AttrsNode< DerivedType >, tvm::AttrsNode< ScatterCollectiveAttrs >, tvm::AttrsNode< HintOnDeviceAttrs >, tvm::AttrsNode< TakeAttrs >, tvm::AttrsNode< ScatterElementsAttrs >, tvm::AttrsNode< GatherElementsAttrs >, tvm::AttrsNode< EinsumAttrs >, tvm::AttrsNode< DistributionAttrs >, tvm::AttrsNode< CallInplacePackedAttrs >, tvm::AttrsNode< AttentionAttrs >, tvm::AttrsNode< AllReduceAttrs >, tvm::AttrsNode< AdaptivePool2DAttrs >, tvm::AttrsNode< ScatterNDAttrs >, tvm::AttrsNode< InitAttrs >, tvm::AttrsNode< FlipAttrs >, tvm::AttrsNode< AllGatherAttrs >, tvm::AttrsNode< AdaptivePool3DAttrs >, tvm::AttrsNode< ToVDeviceAttrs >, tvm::AttrsNode< ExpandDimsAttrs >, tvm::AttrsNode< ConcatAttrs >, tvm::AttrsNode< TriluAttrs >, tvm::AttrsNode< SoftmaxAttrs >, tvm::AttrsNode< AdaptivePool1DAttrs >, tvm::AttrsNode< MatmulAttrs >, tvm::AttrsNode< ArgmaxArgminAttrs >, tvm::AttrsNode< Resize2DAttrs >, tvm::AttrsNode< Pool3DAttrs >, tvm::AttrsNode< LayerNormAttrs >, tvm::AttrsNode< SqueezeAttrs >, tvm::AttrsNode< Pool2DAttrs >, tvm::AttrsNode< OneHotAttrs >, tvm::AttrsNode< NLLLossAttrs >, tvm::AttrsNode< BatchNormAttrs >, tvm::AttrsNode< RepeatAttrs >, tvm::AttrsNode< Pool1DAttrs >, tvm::AttrsNode< ScanopAttrs >, tvm::AttrsNode< LeakyReluAttrs >, tvm::AttrsNode< CallTIRWithGradAttrs >, tvm::AttrsNode< CallTIRInplaceAttrs >, tvm::AttrsNode< AstypeAttrs >, tvm::AttrsNode< PadAttrs >, tvm::AttrsNode< DropoutAttrs >, tvm::AttrsNode< TopKAttrs >, tvm::AttrsNode< MultinomialFromUniformAttrs >, tvm::AttrsNode< Conv3DAttrs >, tvm::AttrsNode< VirtualDeviceNode >, tvm::AttrsNode< WrapParamAttrs >, tvm::AttrsNode< TileAttrs >, tvm::AttrsNode< StridedSliceAttrs >, tvm::AttrsNode< SplitAttrs >, tvm::AttrsNode< RMSNormAttrs >, tvm::AttrsNode< QuantizeAttrs >, tvm::AttrsNode< Conv2DAttrs >, tvm::AttrsNode< SortAttrs >, tvm::AttrsNode< Conv1DAttrs >, tvm::AttrsNode< ArgsortAttrs >, tvm::AttrsNode< StatisticalAttrs >, tvm::AttrsNode< LayoutTransformAttrs >, tvm::AttrsNode< Conv1DTransposeAttrs >, tvm::AttrsNode< Conv2DTransposeAttrs >, tvm::AttrsNode< PermuteDimsAttrs >, tvm::AttrsNode< GroupNormAttrs >, and tvm::AttrsNode< GatherNDAttrs >.
|
staticconstexpr |
|
staticconstexpr |
|
staticconstexpr |