tvm::meta_schedule::TuneContextNode Class Reference

The auto tuning context. More...

#include <tune_context.h>

Inheritance diagram for tvm::meta_schedule::TuneContextNode:

Collaboration diagram for tvm::meta_schedule::TuneContextNode:

Public Types
using	TRandState = support::LinearCongruentialEngine::TRandState

Public Member Functions
void	Initialize ()
	Initialize members that needs initialization with tune context. More...
TuneContext	Clone () const
	Clone the tune context. More...
	TVM_DECLARE_FINAL_OBJECT_INFO (TuneContextNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
Optional< IRModule >	mod
	The workload to be tuned. More...
Optional< Target >	target
	The target to be tuned for. More...
Optional< SpaceGenerator >	space_generator
	The design space generator. More...
Optional< SearchStrategy >	search_strategy
	The search strategy. More...
Optional< String >	task_name
	The name of the tuning task. More...
int	num_threads
	The number of threads to be used. More...
TRandState	rand_state
	The random state. More...
ffi::Function	logger
	The tuning task's logging function. t. More...

Static Public Attributes
static constexpr const char *	_type_key = "meta_schedule.TuneContext"

Detailed Description

The auto tuning context.

Member Typedef Documentation

TRandState

using tvm::meta_schedule::TuneContextNode::TRandState = support::LinearCongruentialEngine::TRandState

Member Function Documentation

Clone()

TuneContext tvm::meta_schedule::TuneContextNode::Clone

(

)

const

Clone the tune context.

Returns

The cloned tune context.

Initialize()

void tvm::meta_schedule::TuneContextNode::Initialize

(

)

Initialize members that needs initialization with tune context.

RegisterReflection()

static void tvm::meta_schedule::TuneContextNode::RegisterReflection ( )

inlinestatic

TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::meta_schedule::TuneContextNode::TVM_DECLARE_FINAL_OBJECT_INFO	(	TuneContextNode	,
		Object
	)

Member Data Documentation

_type_key

constexpr const char* tvm::meta_schedule::TuneContextNode::_type_key = "meta_schedule.TuneContext"

staticconstexpr

logger

ffi::Function tvm::meta_schedule::TuneContextNode::logger

The tuning task's logging function. t.

mod

Optional<IRModule> tvm::meta_schedule::TuneContextNode::mod

The workload to be tuned.

num_threads

int tvm::meta_schedule::TuneContextNode::num_threads

The number of threads to be used.

rand_state

TRandState tvm::meta_schedule::TuneContextNode::rand_state

The random state.

search_strategy

Optional<SearchStrategy> tvm::meta_schedule::TuneContextNode::search_strategy

The search strategy.

space_generator

Optional<SpaceGenerator> tvm::meta_schedule::TuneContextNode::space_generator

The design space generator.

target

Optional<Target> tvm::meta_schedule::TuneContextNode::target

The target to be tuned for.

task_name

Optional<String> tvm::meta_schedule::TuneContextNode::task_name

The name of the tuning task.

---

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

• include/tvm/meta_schedule/tune_context.h