Managed reference to TaskSchedulerNode.
More...
#include <task_scheduler.h>
◆ GradientBased()
Create a task scheduler that fetches tasks in a gradient based fashion.
- Parameters
-
logger | The tuning task's logging function. |
alpha | The parameter alpha to control gradient computation. |
window_size | The parameter to control backward window size. |
seed | The random seed. |
- Returns
- The task scheduler created.
◆ PyTaskScheduler()
Create a task scheduler with customized methods on the python-side.
- Parameters
-
logger | The tuning task's logging function. |
f_next_task_id | The packed function of NextTaskId . |
f_join_running_task | The packed function of JoinRunningTask . |
f_tune | The packed function of Tune . |
- Returns
- The task scheduler created.
◆ RoundRobin()
static TaskScheduler tvm::meta_schedule::TaskScheduler::RoundRobin |
( |
ffi::Function |
logger | ) |
|
|
static |
Create a task scheduler that fetches tasks in a round-robin fashion.
- Parameters
-
logger | The tuning task's logging function. |
- Returns
- The task scheduler created.
◆ TVM_DEFINE_MUTABLE_NOTNULLABLE_OBJECT_REF_METHODS()
The documentation for this class was generated from the following file: