tvm
Public Member Functions | Static Public Member Functions | List of all members
tvm::meta_schedule::SearchStrategy Class Reference

Managed reference to SearchStrategyNode. More...

#include <search_strategy.h>

Inheritance diagram for tvm::meta_schedule::SearchStrategy:
Collaboration diagram for tvm::meta_schedule::SearchStrategy:

Public Member Functions

 TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS (SearchStrategy, ObjectRef, SearchStrategyNode)
 
- 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 Objectget () const
 
const Objectoperator-> () 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...
 

Static Public Member Functions

static SearchStrategy PySearchStrategy (PySearchStrategyNode::FInitializeWithTuneContext f_initialize_with_tune_context, PySearchStrategyNode::FPreTuning f_pre_tuning, PySearchStrategyNode::FPostTuning f_post_tuning, PySearchStrategyNode::FGenerateMeasureCandidates f_generate_measure_candidates, PySearchStrategyNode::FNotifyRunnerResults f_notify_runner_results)
 Create a search strategy with customized methods on the python-side. More...
 
static SearchStrategy ReplayTrace (int num_trials_per_iter, int num_trials_total)
 Constructor of replay trace search strategy. 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
Objectget_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< Objectdata_
 Internal pointer that backs the reference. More...
 

Detailed Description

Managed reference to SearchStrategyNode.

See also
SearchStrategyNode

Member Function Documentation

◆ PySearchStrategy()

static SearchStrategy tvm::meta_schedule::SearchStrategy::PySearchStrategy ( PySearchStrategyNode::FInitializeWithTuneContext  f_initialize_with_tune_context,
PySearchStrategyNode::FPreTuning  f_pre_tuning,
PySearchStrategyNode::FPostTuning  f_post_tuning,
PySearchStrategyNode::FGenerateMeasureCandidates  f_generate_measure_candidates,
PySearchStrategyNode::FNotifyRunnerResults  f_notify_runner_results 
)
static

Create a search strategy with customized methods on the python-side.

Parameters
f_initialize_with_tune_contextThe packed function of InitializeWithTuneContext.
f_pre_tuningThe packed function of PreTuning.
f_post_tuningThe packed function of PostTuning.
f_generate_measure_candidatesThe packed function of GenerateMeasureCandidates.
f_notify_runner_resultsThe packed function of NotifyRunnerResults.
Returns
The search strategy created.

◆ ReplayTrace()

static SearchStrategy tvm::meta_schedule::SearchStrategy::ReplayTrace ( int  num_trials_per_iter,
int  num_trials_total 
)
static

Constructor of replay trace search strategy.

Parameters
num_trials_per_iterThe number of trials per iteration, i.e., the batch size.
num_trials_totalThe total number of trials for trace replaying.

◆ TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS()

tvm::meta_schedule::SearchStrategy::TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS ( SearchStrategy  ,
ObjectRef  ,
SearchStrategyNode   
)

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