Managed reference to SearchTaskNode. More...

Inheritance diagram for tvm::auto_scheduler::SearchTask:

Collaboration diagram for tvm::auto_scheduler::SearchTask:

Public Member Functions
	SearchTask (ComputeDAG compute_dag, String workload_key, Target target, Target target_host, Optional< HardwareParams > hardware_params, LayoutRewriteOption layout_rewrite_option, Array< String > task_input_names, String desc="")
	The constructor. More...

	TVM_DEFINE_OBJECT_REF_METHODS (SearchTask, ObjectRef, SearchTaskNode)

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 SearchTaskNode.

Constructor & Destructor Documentation

tvm::auto_scheduler::SearchTask::SearchTask	(	ComputeDAG	compute_dag,
		String	workload_key,
		Target	target,
		Target	target_host,
		Optional< HardwareParams >	hardware_params,
		LayoutRewriteOption	layout_rewrite_option,
		Array< String >	task_input_names,
		String	desc = `""`
	)

The constructor.

Parameters

compute_dag	The ComputeDAG for the compute declaration.
workload_key	The workload key for the compute declaration.
target	The target device of this search task.
target_host	The target host device of this search task.
hardware_params	Hardware parameters used in this search task.
layout_rewrite_option	The layout rewrite option used for measuring programs.
task_input_names	Names of some user defined input data used in program measuring.
desc	The description string of this task.

tvm::auto_scheduler::SearchTask::TVM_DEFINE_OBJECT_REF_METHODS	(	SearchTask	,
		ObjectRef	,
		SearchTaskNode
	)

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