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 (Optional< IRModule > mod, Optional< Target > target, Optional< SpaceGenerator > space_generator, Optional< SearchStrategy > search_strategy, Optional< String > task_name, int num_threads, TRandState rand_state, ffi::Function logger)
 Constructor. More...
 
 TVM_DEFINE_MUTABLE_NOTNULLABLE_OBJECT_REF_METHODS (TuneContext, ObjectRef, TuneContextNode)
 

Detailed Description

Managed reference to TuneContextNode.

See also
TuneContextNode

Member Typedef Documentation

◆ TRandState

Constructor & Destructor Documentation

◆ TuneContext()

tvm::meta_schedule::TuneContext::TuneContext ( Optional< IRModule mod,
Optional< Target target,
Optional< SpaceGenerator space_generator,
Optional< SearchStrategy search_strategy,
Optional< 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_DEFINE_MUTABLE_NOTNULLABLE_OBJECT_REF_METHODS()

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

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