Managed reference to SplitStepNode. More...

#include <transform_step.h>

Inheritance diagram for tvm::auto_scheduler::SplitStep:

Collaboration diagram for tvm::auto_scheduler::SplitStep:

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)

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 , typename = std::enable_if_t<std::is_base_of_v<Object, ObjectType>>>
const ObjectType *	as () const
	Try to downcast the internal Object to a raw pointer of a corresponding type. More...

template<typename ObjectRefType , typename = std::enable_if_t<std::is_base_of_v<ObjectRef, ObjectRefType>>>
Optional< ObjectRefType >	as () const
	Try to downcast the ObjectRef to a Optional<T> of the requested 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...

Detailed Description

Managed reference to SplitStepNode.

See also: SplitStepNode

Constructor & Destructor Documentation

◆ SplitStep() [1/2]

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.

Parameters

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.

◆ SplitStep() [2/2]

tvm::auto_scheduler::SplitStep::SplitStep ( dmlc::JSONReader * reader )

explicit

The constructor used to read a step record from JSONReader and create the corresponding step.

Parameters

reader The input JSONReader.

Member Function Documentation

◆ TVM_DEFINE_OBJECT_REF_METHODS()

tvm::auto_scheduler::SplitStep::TVM_DEFINE_OBJECT_REF_METHODS	(	SplitStep	,
		Step	,
		SplitStepNode
	)

The documentation for this class was generated from the following file:

include/tvm/auto_scheduler/transform_step.h