tvm
Public Types | Public Member Functions | List of all members
tvm::meta_schedule::TuneContext Class Reference

Managed reference to TuneContextNode. More...

#include <tune_context.h>

Inheritance diagram for tvm::meta_schedule::TuneContext:
Collaboration diagram for tvm::meta_schedule::TuneContext:

Public Types

using TRandState = support::LinearCongruentialEngine::TRandState
 

Public Member Functions

 TuneContext (ObjectPtr< TuneContextNode > data)
 Constructor from ObjectPtr<TuneContextNode>. More...
 
 TuneContext (ffi::Optional< IRModule > mod, ffi::Optional< Target > target, ffi::Optional< SpaceGenerator > space_generator, ffi::Optional< SearchStrategy > search_strategy, ffi::Optional< ffi::String > task_name, int num_threads, TRandState rand_state, ffi::Function logger)
 Constructor. More...
 
 TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (TuneContext, ObjectRef, TuneContextNode)
 

Detailed Description

Managed reference to TuneContextNode.

See also
TuneContextNode

Member Typedef Documentation

◆ TRandState

Constructor & Destructor Documentation

◆ TuneContext() [1/2]

tvm::meta_schedule::TuneContext::TuneContext ( ObjectPtr< TuneContextNode data)
inlineexplicit

Constructor from ObjectPtr<TuneContextNode>.

Parameters
dataThe object pointer.

◆ TuneContext() [2/2]

tvm::meta_schedule::TuneContext::TuneContext ( ffi::Optional< IRModule mod,
ffi::Optional< Target target,
ffi::Optional< SpaceGenerator space_generator,
ffi::Optional< SearchStrategy search_strategy,
ffi::Optional< ffi::String >  task_name,
int  num_threads,
TRandState  rand_state,
ffi::Function  logger 
)
explicit

Constructor.

Parameters
modThe workload to be tuned.
targetThe target to be tuned for.
space_generatorThe design space generator.
search_strategyThe search strategy.
task_nameThe name of the tuning task.
num_threadsThe number of threads to be used.
rand_stateThe random state.
loggerThe tuning task's logging function.

Member Function Documentation

◆ TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE()

tvm::meta_schedule::TuneContext::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE ( TuneContext  ,
ObjectRef  ,
TuneContextNode   
)

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