tvm
|
Managed reference to SearchStrategyNode. More...
#include <search_strategy.h>
Public Member Functions | |
TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS (SearchStrategy, ObjectRef, SearchStrategyNode) | |
![]() | |
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... | |
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 | |
![]() | |
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 SearchStrategyNode.
|
static |
Create a search strategy with customized methods on the python-side.
f_initialize_with_tune_context | The packed function of InitializeWithTuneContext . |
f_pre_tuning | The packed function of PreTuning . |
f_post_tuning | The packed function of PostTuning . |
f_generate_measure_candidates | The packed function of GenerateMeasureCandidates . |
f_notify_runner_results | The packed function of NotifyRunnerResults . |
|
static |
Constructor of replay trace search strategy.
num_trials_per_iter | The number of trials per iteration, i.e., the batch size. |
num_trials_total | The total number of trials for trace replaying. |
tvm::meta_schedule::SearchStrategy::TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS | ( | SearchStrategy | , |
ObjectRef | , | ||
SearchStrategyNode | |||
) |