tvm
|
Managed reference to SplitStepNode. More...
#include <transform_step.h>
Public Member Functions | |
SplitStep (int stage_id, int iter_id, Optional< PrimExpr > extent, const Array< Optional< Integer >> &lengths, bool inner_to_outer) | |
The constructor. More... | |
SplitStep (dmlc::JSONReader *reader) | |
The constructor used to read a step record from JSONReader and create the corresponding step. More... | |
TVM_DEFINE_OBJECT_REF_METHODS (SplitStep, Step, SplitStepNode) | |
![]() | |
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) | |
![]() | |
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 | |
![]() | |
using | ContainerType = Object |
type indicate the container type. More... | |
![]() | |
static constexpr bool | _type_is_nullable = true |
![]() | |
Object * | get_mutable () const |
![]() | |
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... | |
![]() | |
ObjectPtr< Object > | data_ |
Internal pointer that backs the reference. More... | |
Managed reference to SplitStepNode.
tvm::auto_scheduler::SplitStep::SplitStep | ( | int | stage_id, |
int | iter_id, | ||
Optional< PrimExpr > | extent, | ||
const Array< Optional< Integer >> & | lengths, | ||
bool | inner_to_outer | ||
) |
The constructor.
stage_id | The index of the stage to be split. |
iter_id | The index of the iterator to be split. |
extent | The extent length of the axis to split. |
lengths | The multiple split factors. Can be None to be filled by search policy. |
inner_to_outer | The split direction. |
|
explicit |
The constructor used to read a step record from JSONReader and create the corresponding step.
reader | The input JSONReader. |
tvm::auto_scheduler::SplitStep::TVM_DEFINE_OBJECT_REF_METHODS | ( | SplitStep | , |
Step | , | ||
SplitStepNode | |||
) |