|
tvm
|
Managed reference to AnnotationStepNode. More...
#include <transform_step.h>
Public Member Functions | |
| AnnotationStep (int stage_id, int iter_id, IteratorAnnotation ann) | |
| The constructor. More... | |
| AnnotationStep (dmlc::JSONReader *reader) | |
| The constructor used to read a step record from JSONReader and create the corresponding step. More... | |
| TVM_DEFINE_OBJECT_REF_METHODS (AnnotationStep, Step, AnnotationStepNode) | |
Public Member Functions inherited from tvm::auto_scheduler::Step | |
| StepNode * | CopyOnWrite () |
| CopyOnWrite function for Step. This works almost the same as a normal ObjectRef.CopyOnWrite(), but can dispatch to different steps. More... | |
| TVM_DEFINE_OBJECT_REF_METHODS (Step, ObjectRef, StepNode) | |
Public Member Functions inherited from tvm::runtime::ObjectRef | |
| ObjectRef ()=default | |
| default constructor More... | |
| ObjectRef (ObjectPtr< Object > data) | |
| Constructor from existing object ptr. More... | |
| bool | same_as (const ObjectRef &other) const |
| Comparator. More... | |
| bool | operator== (const ObjectRef &other) const |
| Comparator. More... | |
| bool | operator!= (const ObjectRef &other) const |
| Comparator. More... | |
| bool | operator< (const ObjectRef &other) const |
| Comparator. More... | |
| bool | defined () const |
| const Object * | get () const |
| const Object * | operator-> () const |
| bool | unique () const |
| int | use_count () const |
| template<typename ObjectType > | |
| const ObjectType * | as () const |
| Try to downcast the internal Object to a raw pointer of a corresponding type. More... | |
Additional Inherited Members | |
Public Types inherited from tvm::runtime::ObjectRef | |
| using | ContainerType = Object |
| type indicate the container type. More... | |
Static Public Attributes inherited from tvm::runtime::ObjectRef | |
| static constexpr bool | _type_is_nullable = true |
Protected Member Functions inherited from tvm::runtime::ObjectRef | |
| Object * | get_mutable () const |
Static Protected Member Functions inherited from tvm::runtime::ObjectRef | |
| template<typename T > | |
| static T | DowncastNoCheck (ObjectRef ref) |
| Internal helper function downcast a ref without check. More... | |
| static void | FFIClearAfterMove (ObjectRef *ref) |
| Clear the object ref data field without DecRef after we successfully moved the field. More... | |
| template<typename ObjectType > | |
| static ObjectPtr< ObjectType > | GetDataPtr (const ObjectRef &ref) |
| Internal helper function get data_ as ObjectPtr of ObjectType. More... | |
Protected Attributes inherited from tvm::runtime::ObjectRef | |
| ObjectPtr< Object > | data_ |
| Internal pointer that backs the reference. More... | |
Managed reference to AnnotationStepNode.
| tvm::auto_scheduler::AnnotationStep::AnnotationStep | ( | int | stage_id, |
| int | iter_id, | ||
| IteratorAnnotation | ann | ||
| ) |
The constructor.
| stage_id | The index of the stage to add annotation. |
| iter_id | The index of the iterator to add annotation. |
| ann | The annotation type of this step. |
|
explicit |
The constructor used to read a step record from JSONReader and create the corresponding step.
| reader | The input JSONReader. |
| tvm::auto_scheduler::AnnotationStep::TVM_DEFINE_OBJECT_REF_METHODS | ( | AnnotationStep | , |
| Step | , | ||
| AnnotationStepNode | |||
| ) |
1.8.13